CCL-DTI: contributing the contrastive loss in drug–target interaction prediction

Alireza Dehghan,Karim Abbasi,Parvin Razzaghi,Hossein Banadkuki,Sajjad Gharaghani
DOI: https://doi.org/10.1186/s12859-024-05671-3
IF: 3.307
2024-02-01
BMC Bioinformatics
Abstract:The Drug–Target Interaction (DTI) prediction uses a drug molecule and a protein sequence as inputs to predict the binding affinity value. In recent years, deep learning-based models have gotten more attention. These methods have two modules: the feature extraction module and the task prediction module. In most deep learning-based approaches, a simple task prediction loss (i.e., categorical cross entropy for the classification task and mean squared error for the regression task) is used to learn the model. In machine learning, contrastive-based loss functions are developed to learn more discriminative feature space. In a deep learning-based model, extracting more discriminative feature space leads to performance improvement for the task prediction module.
biochemical research methods,biotechnology & applied microbiology,mathematical & computational biology
What problem does this paper attempt to address?
The problem this paper attempts to address is: how to improve the performance of drug-target interaction (DTI) prediction models by using different contrastive loss functions. Specifically, the authors propose a new method that consists of two stages: 1. **Feature Extraction Stage**: Utilizing multimodal knowledge as input, including drug molecules, protein sequences, protein-protein interaction networks, and drug-drug interaction networks. Features are extracted using 1D convolutional neural networks and Node2vec networks, and these different modalities of knowledge are fused using a bidirectional attention mechanism. 2. **Loss Function Stage**: Combining task prediction loss and contrastive loss functions to learn a more discriminative feature space. The authors consider four types of contrastive loss functions: maximum margin contrastive loss, triplet loss, multi-class N-pair loss, and NT-Xent loss. Through experiments on four well-known datasets (Wang et al. dataset, Luo dataset, Davis dataset, and KIBA dataset), the authors validate the impact of different contrastive loss functions on model performance and demonstrate that this method significantly outperforms existing state-of-the-art methods on multiple evaluation metrics. This indicates that using appropriate contrastive loss functions can enhance the feature representation capability of the model, thereby improving the accuracy of DTI prediction.