An application of convolutional neural networks with salient features for relation classification

• Main Authors: Zolzaya Dashdorj, Min Song
• Format: Article
• Language: English
• Published: BMC 2019-05-01
• Series: BMC Bioinformatics
• Subjects: Convolutional neural networks; Biomedical data analysis; Relation classification; Hyperparameter optimization; Deep learning
• Online Access: http://link.springer.com/article/10.1186/s12859-019-2808-3

Abstract Background Due to the advent of deep learning, the increasing number of studies in the biomedical domain has attracted much interest in feature extraction and classification tasks. In this research, we seek the best combination of feature set and hyperparameter setting of deep learning algorithms for relation classification. To this end, we incorporate an entity and relation extraction tool, PKDE4J to extract biomedical features (i.e., biomedical entities, relations) for the relation classification. We compared the chosen Convolutional Neural Networks (CNN) based classification model with the most widely used learning algorithms. Results Our CNN based classification model outperforms the most widely used supervised algorithms. We achieved a significant performance on binary classification with a weighted macro-average F1-score: 94.79% using pre-extracted relevant feature combinations. For multi-class classification, the weighted macro-average F1-score is estimated around 86.95%. Conclusions Our results suggest that our proposed CNN based model using the not only single feature as the raw text of the sentences of biomedical literature, but also coupling with multiple and highlighted features extracted from the biomedical sentences could improve the classification performance significantly. We offer hyperparameter tuning and optimization approaches for our proposed model to obtain optimal hyperparameters of the models with the best performance.