Abstract
Motivation
The accurate prediction of drug–target interactions (DTI) is a crucial step in drug discovery, providing a foundation for identifying novel therapeutics. Traditional drug development is both costly and time-consuming, often spanning over a decade. Computational approaches help narrow the pool of compound candidates, offering significant starting points for experimental validation. In this study, we propose a Top-DTI framework for predicting DTI by integrating topological data analysis (TDA) with large language models (LLMs). Top-DTI leverages persistent homology to extract topological features from protein contact maps and drug molecular images. Simultaneously, protein and drug LLMs generate semantically rich embeddings that capture sequential and contextual information from protein sequences and drug SMILES strings. By combining these complementary features, Top-DTI enhances predictive performance and robustness.
Results
Experimental results on the public BioSNAP and Human DTI benchmark datasets demonstrate that the proposed Top-DTI model outperforms state-of-the-art approaches across multiple evaluation metrics, including AUROC, AUPRC, sensitivity, and specificity. Furthermore, the Top-DTI model achieves superior performance in the challenging cold-split scenario, where the test and validation sets contain drugs or targets absent from the training set. This setting simulates real-world scenarios and highlights the robustness of the model. Notably, incorporating topological features alongside LLM embeddings significantly improves predictive performance, underscoring the value of integrating structural and sequence-based representations.
Availability and implementation
The data and source code of Top-DTI are available at https://github.com/bozdaglab/Top_DTI under the Creative Commons Attribution NonCommercial 4.0 International Public License.
1 Introduction
Drug development is a costly and long-term process because of the expensive and labor-intensive nature of experimental assays. The approval process for a new drug in clinical practice generally takes 10–15 years, with associated costs ranging between $161 million and $4.54 billion (Hinkson et al. 2020; Schlander et al. 2021). The drug development process encounters significant failure rates due to safety concerns, lack of efficacy, and the constraints of traditional methods, such as high-throughput screening, which is challenged by the complex nature of evaluating extensive compound libraries (Romanelli et al. 2024). To overcome these problems, computational methods that improve and automate traditional methodologies have become an essential resource in drug discovery.
Drug target interaction (DTI) prediction is a crucial component of drug discovery research to identify target proteins of a drug molecule. Docking simulations, ligand-based methods, and machine learning approaches are the three fundamental types of computational techniques for DTI prediction. A docking-based approach uses 3D structures of target proteins and drug molecules with simulations to find binding sites. However, this method has limitations related to its laborious processes and difficulties in modeling complex proteins (Yamanishi et al. 2008). Ligand-based models, which involve the comparison of novel ligands to known protein ligands, exhibit poor results when the number of known ligands is limited. Ligand-based and docking simulation methods yield promising predictive results; however, their reliance on high-quality data for drug molecules and proteins restricts their applicability in DTI prediction tasks (Speck-Planche et al. 2012; Yang et al. 2021).
The methodologies used in machine learning for DTI prediction can be categorized into two groups: similarity- and feature vector-based approaches. Similarity-based methods are based on the assumption that similar drugs have a tendency to interact with similar targets. These methods employ a variety of similarity metrics based on, e.g. chemical, ligand, expression, and side-effect data to predict DTIs. Feature vector-based methodologies represent drugs and targets as feature vectors, which makes it easier to capture complex interactions and connections among distant molecular components. Drugs are usually represented using multiple techniques, including the Simplified Molecular Input Line Entry System (SMILES), molecular fingerprints, 2D structural representations, learned embeddings, and molecular graphs. Proteins are represented by sequence-, structural-, and network-based feature vectors. Network-based representations, like protein-protein interaction networks and knowledge graphs, combine various sources of data to enhance feature extraction. Machine learning models are trained using these representations to predict DTIs.
Deep learning techniques have significantly enhanced DTI prediction by automatically extracting high-dimensional features and modeling non-linear relationships between drugs and targets. These methods usually encode drug and protein structures separately, concatenate their learned representations, and then use them as input for a classifier. For instance, DeepDTA (Öztürk et al. 2018) uses convolutional neural networks (CNNs) to extract features from protein sequences and drug SMILES strings for the prediction of binding affinity. Similarly, DeepConv-DTI (Lee et al. 2019) employs a CNN model to predict DTI using protein sequences and Morgan fingerprints of drugs. Hybrid deep learning approaches that combine LSTM and CNN architectures, as well as models that integrate diverse features, such as protein sequence, structure, and drug chemical properties, have been used to improve the accuracy of DTI prediction (Zheng et al. 2018; Rifaioglu et al. 2021).
Graph Neural Networks (GNN) are also capable of accurately predicting DTI by learning intricate relationships of drugs and targets within networks. For example, GraphDTA (Nguyen et al. 2021) and MGNDTI (Peng et al. 2024) model drugs as molecular graphs and employ GNN for predicting drug–target affinity and DTI. The GSL-DTI model incorporates heterogeneous networks using meta-path-based graph convolution to learn drug and protein representations for DTI prediction (Zixuan et al. 2024).
Attention-based models have received great interest in this domain. DrugBAN (Bai et al. 2023) integrates a bilinear attention network to comprehend local interactions between drugs and targets. MolTrans (Huang et al. 2021) utilizes a self-attention mechanism to change structural embeddings. HyperAttentionDTI (Zhao et al. 2022) employs the attention mechanism on the feature matrices, assigning an attention vector to each amino acid. CoaDTI (Huang et al. 2022) models the interaction information from the drug and protein modalities using a co-attention mechanism.
Large language models (LLMs) have emerged as powerful tools in deep learning, utilizing vast amounts of unlabeled data through self-supervised learning. For downstream tasks, LLMs trained on protein sequences and drug SMILES string representations provide informative and contextually rich features. Protein sequence LLMs such as ProtT5 (Elnaggar et al. 2021) and ESM2 (Beal, 2015) models are trained using over millions of amino acid sequences. Similarly, drug-related LLMs such as MoLFormer (Ross et al. 2022), ChemBERTa (Ahmad et al. 2022), and ChemGPT (Frey et al. 2023) have shown significant effectiveness in capturing chemical characteristics. In recent years, pre-trained models have been employed for DTI prediction. The ConPLex model (Singh et al. 2023) utilizes a pre-trained protein language model to acquire protein representations using the contrastive learning method, aligning proteins and drugs in a common latent space. Kang et al. employed the pre-trained ChemBERTa model for chemical compounds and ProBERT for target proteins to predict DTIs (Kang et al. 2022). The DrugLAMP model (Luo et al. 2024) combines molecular graph and protein sequence features derived from protein language models for DTI prediction.
Despite advances in DTI prediction through similarity-based, graph-based, and deep learning approaches, these methods mainly rely on features such as chemical descriptors, protein sequences, and graph-based embeddings. However, they often overlook topological components and structural data, which are also critical to capture interactions between drugs and targets.
Topological Deep Learning (TDL) is an emerging research area that combines the concepts of Topological Data Analysis (TDA) with current machine learning methodologies (Demir et al. 2022). TDA utilizes algebraic topology to uncover the fundamental structure of high-dimensional datasets by examining their topological characteristics, such as connectivity, loops, and voids. Persistent homology, a fundamental technique in TDA, provides a robust and noise-resistant framework for capturing and analyzing multi-scale topological patterns in data. The collaboration between TDA and deep learning allows models to extract and leverage higher-order structural information that is frequently neglected by traditional methods (Coskunuzer and Akçora, 2024; Zia et al. 2024).
There has been a growing demand for utilizing TDA to tackle complex and high-dimensional problems across multiple domains. For example, in biomedical imaging, TDL has significantly improved histopathological cancer detection (Yadav et al. 2023) and biological image segmentation (Glatt and Liu, 2023). In genomics, TDL has facilitated the identification of intricate patterns within genetic data, providing insights into genetic diseases (Amézquita et al. 2023). In drug discovery, ToDD (Demir et al. 2022) uses multiparameter persistence homology, enhancing virtual screening performance by incorporating domain-specific chemical features.
In this study, we propose a novel computational framework called Top-DTI for DTI prediction by integrating embeddings learned from TDA and LLMs. These embeddings are dynamically fused and further refined within a GNN, using the connectivity of the DTI graph. Our model leverages two key input features for both drugs and targets: embeddings from pre-trained drug and protein LLMs, and topological features derived through TDA on drug molecular images and protein contact maps.
To effectively combine these complementary features, we designed a feature fusion module that dynamically integrates TDA and LLM embeddings by assessing their relative importance during training. This fusion enhances the model’s ability to utilize both sequence-based and topological information. The integrated embeddings were subsequently processed through a heterogeneous GNN, which models the relationships between drugs and targets. This approach enables the identification of complex interaction patterns while efficiently leveraging both network topology and the fused feature representations.
The proposed Top-DTI model was evaluated on publicly available BioSNAP (Zitnik et al. 2018) and Human (Liu et al. 2015; Chen et al. 2020) benchmark datasets. Experimental results demonstrate that Top-DTI consistently outperforms state-of-the-art DTI prediction models, highlighting its effectiveness and robustness. The main contributions of this work are as follows:
We present a DTI prediction framework that utilizes TDA through the cubical persistence features derived from 2D drug molecule images and protein contact maps.
The proposed framework uses MoLFormer and ProtT5 LLMs to extract 1D embeddings from drug SMILES strings and target protein sequences, respectively.
We design a feature fusion module that dynamically fuses LLM and TDA embeddings for drugs and targets during training.
We demonstrate the effectiveness of the Top-DTI model by conducting a comprehensive assessment on public benchmark datasets, obtaining superior results compared to the state-of-the-art methodologies.
We examine the robustness of the proposed model in a challenging cold-split scenario, using unseen drugs and targets in the test set, imitating real-world situations.
Top-DTI provides a robust and efficient framework for DTI prediction by integrating complementary feature representations, thereby advancing computational drug discovery.
2 Materials and methods
2.1 Datasets
To evaluate Top-DTI, we used public BioSNAP and Human benchmark datasets, along with their variants, including BioSNAP unseen drugs, BioSNAP unseen targets, and Human cold datasets, which were obtained from (Luo et al. 2024) and (Singh et al. 2023). The BioSNAP dataset is derived from the DrugBank database and includes genes that are targeted by drugs available in the US market. The dataset is balanced, having verified 13 830 positive interactions and 13 634 negative interactions randomly sampled from non-interacting drug–target pairs. The Human dataset contains 2633 positive interactions and 3364 highly accurate negative interactions obtained by an in silico screening process. The datasets were divided into training, validation, and test sets and included protein sequences, drug SMILES strings, and interaction information for all pairs of drugs and targets. To ensure a fair comparison with other methods, we used the same training and test splits as used in the prior work. The statistical details of the benchmark datasets are given in Table 1.
Table 1.
Statistics of benchmark datasets
| Dataset | Drugs | Targets | Train | Validation | Test |
|---|---|---|---|---|---|
| BioSNAP Random | 4505 | 2181 | 9684 / 9540 | 1398 / 1349 | 2748 / 2745 |
| BioSNAP Unseen Drugs | 4505 | 2181 | 9531 / 9607 | 1383 / 1353 | 2916 / 2674 |
| BioSNAP Unseen Targets | 4505 | 2181 | 9872 / 9489 | 1382 / 1386 | 2576 / 2759 |
| Human Random | 2726 | 2001 | 1846 / 2351 | 251 / 349 | 536 / 664 |
| Human Cold | 1813 | 1503 | 1815 / 1638 | 59 / 96 | 121 / 190 |
We present the number of unique drugs and targets in each dataset. The number pairs in the train, validation, and test columns are reported as positive/negative, where positive indicates known interactions and negative indicates non-interacting drug–target pairs.
We have created 2D representations of drug molecular structures and protein contact maps to capture the structural features of drugs and protein targets. The images of drug molecules were generated from their SMILES representations utilizing the RDKit library (Landrum, 2013). To create protein contact maps, we used a transformer-based contact prediction model (Rao et al. 2020). The contact maps were created using the self-attention maps of the transformer model.
2.2 The proposed method
The proposed framework, called Top-DTI for DTI prediction, integrates features derived from TDA and LLMs. Initially, TDA methods were utilized to derive topological features from molecular images of drugs and protein contact maps. Additionally, ProtT5 and MoLFormer LLMs were used to extract embeddings from protein sequences and drug SMILES strings, respectively. Then, the TDA and LLM embeddings were combined through a learnable fusion mechanism that dynamically balances the contributions of topological and sequence-based features. Afterwards, these integrated representations were fed to a heterogeneous GNN to learn relational information from the DTI network. Finally, embeddings learned from GNN were utilized to train a multilayer perceptron (MLP) classifier to predict DTIs. Figure 1 illustrates the Top-DTI framework.
Figure 1.
Top-DTI Model Architecture. (a and b) Drug and protein feature extraction includes vectorization of molecular and contact map images using Betti curves and persistence landscapes. MoLFormer and ProtT5 transformer models were employed to extract sequential embeddings from SMILES and protein sequences, respectively. Topological and sequential embeddings are combined using feature fusion modules. (c) The fused features were used as node features in a graph, where a Graph Neural Network (GNN) was applied to learn node embeddings. (d) The embeddings obtained from the GNN were concatenated and forwarded to a multilayer perceptron (MLP) to predict DTI.
2.2.1 Topological feature extraction from drug molecular images and protein contact maps
In this study, we utilized persistent homology (PH) to extract topological features from two-dimensional molecular images of drugs and protein contact maps. PH is a method in TDA that analyzes structural information to capture topological features such as connected components and loops. These features are essential for capturing hidden shape patterns. The process of PH can be outlined in three steps. The first step, ‘filtration’, involves the construction of a nested sequence of cubical complexes from the images by monitoring the evolution of topological structures. The second step, ‘persistence diagrams’, records the birth (appearance) and death (disappearance) of topological features during the filtration process. The third step, ‘feature vectorization’, converts persistence diagrams into feature vectors using methodologies such as Betti curves and persistence landscapes.
Filtration is defined based on pixel values for images. For a given image X with size , each pixel in the image has a value , where k represents the selected color channel (grayscale, red, green, or blue). To construct the filtration, we choose a series of thresholds , where N is the number of thresholds. At each threshold , a binary image is generated as:
| (1) |
where includes all pixel values less than or equal to for the color channel k. At lower thresholds, only low-pixel-value regions are activated. As the threshold increases, more pixels are added, resulting in the formation of loops and larger connected structures. The result is a nested sequence of binary images (cubical complexes),
| (2) |
where additional pixels are “activated” as the threshold (pixel values) increases (Coskunuzer and Akçora, 2024).
After the filtration, we compute persistence diagrams for dimensions and . Here, and represent connected components and loops, respectively. Each persistence diagram consists of points , where and represent the birth and death times of a topological feature during the filtration, respectively. Features farther from the diagonal have longer lifespans and are considered significant, while features closer to the diagonal are typically noise.
Figure 2 illustrates this process with an example of a molecular image and its corresponding persistence diagram. In this figure, connected components () are highlighted as green squares, and loops () are represented by red diamonds. The connected components monitor distinct regions or components in the molecular structure, such as bonds between atoms. Loops (holes) correspond to cyclic substructures, such as rings in a molecular framework. A loop farther from the diagonal indicates persistent, significant cyclic structures. For example, four red diamonds positioned away from the diagonal represent the four rings in a given sample molecular image. In contrast, features near the diagonal have shorter lifespans and are more likely to represent noise or less significant variations.
Figure 2.
A sample of molecular image (left) and its corresponding persistence diagram (right). The green squares () represent connected components, while the red diamonds () represent loops. Each point denotes the birth and death of a topological feature as it appears and disappears during the filtration process.
After creating the persistence diagrams, we converted these diagrams into feature vectors by using vectorization methods, such as Betti curves and persistence landscapes. Betti curves summarize the evolution of topological features across a filtration. For a given dimension d, the Betti number represents the count of d-dimensional features in a topological space. In our case, denotes the number of connected components, and represents the number of loops in a binary image. Betti vectors are defined as:
| (3) |
where d represents the dimension and k denotes the color channel. Using 50 thresholds (), and two dimensions (), we computed Betti vectors for each dimension ( and ). Therefore, we obtained a 100D vector for each color channel.
We utilized the persistence landscapes method alongside Betti curves to enhance the representation of structural patterns. Persistence landscapes directly use the lifespan of features. For each birth–death pair in a persistence diagram, a piecewise-linear function is defined as:
| (4) |
which creates a triangular shape over the interval , with a peak at . Persistence landscape (Bubenik et al. 2015) is defined as the sequence of functions , where:
| (5) |
In Equation (5), n is the total number of birth–death pairs. We calculated persistence landscapes for the and dimensions using the first layer of persistent landscapes (the largest value, ). To achieve computational efficiency, we partitioned the filtration parameter range into 100 sampling points, referred to as bins. The feature vector for each color channel ends up being 200, as we performed an analysis on two homology dimensions.
As a result, Betti curves generated a 100D vector, and persistence landscapes contributed a 200D vector for each color channel, resulting in a total size of 300 dimensions. By concatenating the feature vectors across all four channels (grayscale, red, green, and blue), we obtained a 1200D feature vector. These representations were used as topological features in the TOP-DTI model.
2.2.2 Sequence-based feature extraction using LLMs
We employed pre-trained LLMs to capture the sequence-based features of drugs and protein targets. We utilized the ProtT5 model for protein sequences and the MoLFormer architecture for drug representations. The ProtT5 model employs a transformer architecture featuring an encoder–decoder configuration and is trained on the UniRef50 database to predict masked amino acids with a masking ratio of 15%. We obtained 1024D protein target embeddings from the last hidden states of the model’s encoder. The resulting embeddings were averaged across sequence lengths to generate a fixed-size representation for each protein.
For drug molecules, we utilized MoLFormer, a transformer-based molecular LLM framework, to produce drug representations from chemical SMILES strings. MoLFormer was pre-trained on 1.1 billion unlabeled molecules sourced from the PubChem and ZINC databases (Ross et al. 2022). MoLFormer utilizes rotary position embeddings and an efficient linear attention mechanism to encode the spatial relationships of molecular structures. In the pre-training phase, 15% of the tokens in a SMILES string are masked for prediction. The pre-training task did not involve DTI prediction, thus, there is no risk of data leakage. We have obtained 768D embeddings for drug molecules.
After the extraction of LLM embeddings and topological features of drugs and target proteins, two fully connected layers with ReLU activations were used to reduce their dimensionality and project them into a common 512D latent space to reduce computational complexity. These embeddings were used as high-dimensional sequence-based representations in our proposed Top-DTI framework.
2.2.3 The integrated modeling of sequence and structural embeddings using dynamic feature fusion and GNN
In the Top-DTI framework, we implemented a feature fusion module for drugs and targets to efficiently fuse complementary information from sequence-based embeddings and topological features. During training, this module dynamically assigns weights to each feature type, facilitating the creation of enhanced embeddings. In particular, the sequence-based embeddings and topological features are concatenated, and the fully connected layer provides a weighting factor by sigmoid activation function (Equation (6)).
| (6) |
where is the sigmoid activation function, denotes the sequence embedding vector of a drug or protein, denotes the topological feature vector of a drug or protein, is a learnable weight matrix, and is a learnable bias vector. The sigmoid activation function ensures that the weight vector is in the range of [0, 1].
Using the weight vector , the fused embedding is computed as:
| (7) |
where represents the element-wise multiplication. In Equation (7), the complement is applied element-wise to adjust the contribution of topological features in the fusion process.
The DTI graphs were constructed utilizing known interactions from the BioSNAP and Human benchmark datasets. The fused features of drugs and targets were utilized as node features in a heterogeneous DTI graph. The nodes in this graph represent drugs and targets, while the edges indicate potential interactions. Negative interactions are derived from previous research, thereby maintaining consistency. A two-layer GNN with SAGEConv architecture was applied to learn enriched embeddings for drugs and targets (Hamilton et al. 2017). Batch normalization and dropout layers were implemented at each layer to enhance generalization and mitigate overfitting.
After generating embeddings for drugs and targets using GNN, these embeddings were concatenated and used to train an MLP to predict DTIs.
2.2.4 Model training and evaluation
We evaluated the performance of the Top-DTI model in two evaluation settings: random-split and cold-split. The random split refers to the traditional method of dividing data into training, validation, and test sets, where drugs and targets can appear in both the training and test sets with different interactions. The random-split setting, while useful for preliminary evaluation, may not fully reflect the challenges of predicting interactions for unseen drugs or targets. In contrast, the cold-split setting ensures that any drug or target in the test set is not present in the training set. The cold-split scenario mimics real-world scenarios requiring the model to predict interactions for unseen drugs and targets. Therefore, the model must generalize the interaction beyond its training samples.
In the random-split setting, the interactions in the BioSNAP and Human datasets were split into 70% for training, 10% for validation, and 20% for testing by the dataset curators (Table 1). We used the same splits as the previous studies to have a fair comparison with state-of-the-art (SOTA) methods (Zitnik et al. 2018; Liu et al. 2015; Chen et al. 2020).
In the cold-split setting, the Human dataset, prepared by Luo et al. (Bai et al. 2023), was designed to ensure that all drugs and proteins in the validation and test sets were excluded from the training set, thereby minimizing the risk of hidden data bias, as outlined in (Chen et al. 2020). ‘Unseen drugs’ and ‘unseen targets’ are variations of the BIOSNAP dataset in which drugs or targets in the test set are absent from any interactions in the training set. The unseen target dataset was created by picking 20% of the targets from the entire dataset, including all interactions related to these proteins in the test set, guaranteeing that there is no target overlap between the training and test sets. The same methodology was applied to generate the unseen drugs dataset. The remaining dataset was further divided, allocating 7/8 of the interactions for training and 1/8 for validation (Singh et al. 2023).
The proposed Top-DTI model was trained using a binary cross-entropy loss function and Adam optimizer to update parameters. In order to mitigate overfitting, validation loss was monitored at each epoch, and an early stopping criterion was applied to terminate training if validation loss did not reduce for five consecutive epochs.
We used the area under the receiver operating characteristic curve (AUROC), the area under the precision-recall curve (AUPRC), sensitivity, and specificity as evaluation metrics. The results were reported as average performance over five independent runs, each initialized with a different random seed. All model outputs, including detailed performance metrics, were provided in Jupyter Notebooks to facilitate reproducibility. The source code and documentation are available at https://github.com/bozdaglab/Top_DTI.
3 Results and discussion
To evaluate Top-DTI, we compared it against SOTA and baseline methods using the BioSNAP and Human benchmark datasets, using both traditional random-split and more challenging cold-split scenarios. We also performed an ablation study to examine the contribution of different components of Top-DTI. These analyses are summarized in the subsequent sections.
3.1 Comparison of Top-DTI with baseline and SOTA methods
We evaluated the performance of the Top-DTI model by comparing it with the baseline and SOTA methods reported in a recent study by Luo et al. (2024). The descriptions of these methods are given in the Supplementary Document.
In a standard random-split setting, the performance comparison of Top-DTI with SOTA and baseline methods on the BioSNAP dataset is presented in Table 2. Top-DTI demonstrated superior performance compared to existing SOTA methods, including Kang et al. (2022), DLM-DTI (Lee et al. 2024), and DrugLAMP, which mainly depend on LLM-based drug and target embeddings. Specifically, Top-DTI demonstrated an improvement above 2% in both AUROC and AUPRC compared to DrugLAMP, the previous SOTA method. The sensitivity and specificity scores are balanced, which shows that the TOP-DTI model precisely detects both interacting and non-interacting drug–target pairs.
Table 2.
Performance comparison of methods on the ‘BioSNAP’ dataset.
| Method | AUROC | AUPRC | Sens. | Spec. |
|---|---|---|---|---|
| SVM | 0.862 | 0.864 | 0.711 | 0.841 |
| RF | 0.860 | 0.886 | 0.823 | 0.786 |
| DeepConv-DTI | 0.886 | 0.890 | 0.760 | 0.851 |
| GraphDTA | 0.887 | 0.890 | 0.745 | 0.854 |
| MolTrans | 0.895 | 0.897 | 0.818 | 0.831 |
| Kang et al. | 0.910 | 0.900 | 0.862 | 0.847 |
| DrugBAN | 0.903 | 0.902 | 0.820 | 0.847 |
| DLM-DTI | 0.914 | 0.914 | 0.848 | 0.844 |
| DrugLAMP | 0.917 | 0.922 | 0.844 | 0.855 |
| Top-DTI | 0.939 | 0.941 | 0.866 | 0.857 |
Sens: sensitivity, Spec: specificity. The best and second-best results for each evaluation metric are shown in bold and underlined, respectively.
Table 3 presents the results of the performance comparison of Top-DTI with SOTA and baseline methods on the Human dataset in a random-split setting. Top-DTI shows superior performance compared to all approaches, obtaining the highest AUROC and AUPRC values.
Table 3.
Performance comparison of methods on the ‘Human’ dataset.
| Method | AUROC | AUPRC |
|---|---|---|
| SVM | 0.940 | 0.920 |
| RF | 0.952 | 0.953 |
| DeepConv-DTI | 0.980 | 0.981 |
| GraphDTA | 0.981 | 0.982 |
| MolTrans | 0.980 | 0.978 |
| DrugBAN | 0.982 | 0.980 |
| DrugLAMP | 0.985 | 0.983 |
| Top-DTI | 0.993 | 0.992 |
The best and second-best results for each evaluation metric are shown in bold and underlined, respectively.
Notably, all methods demonstrate high scores for this dataset, with AUROC and AUPRC values greater than 92%. Chen et al. indicated that there could be biases in the Human dataset due to single-class ligands and algorithmically generated negative samples (Chen et al. 2020). The superior predictive performance of the models may be linked to this biased information. The cold-split setting resolves this issue by guaranteeing that test drugs and targets are completely removed from the training set. This design strategy requires all models to generate predictions without prior knowledge of specific drugs or targets (Bai et al. 2023). Table 4 presents the results for the Human Cold dataset in the cold-split setting.
Table 4.
Performance comparison of methods on the ‘Human Cold’ dataset.
| Method | AUROC | AUPRC |
|---|---|---|
| SVM | 0.691 | 0.629 |
| RF | 0.732 | 0.670 |
| DeepConv-DTI | 0.820 | 0.780 |
| GraphDTA | 0.816 | 0.770 |
| MolTrans | 0.804 | 0.767 |
| DrugBAN | 0.850 | 0.794 |
| DrugLAMP | 0.860 | 0.814 |
| Top-DTI | 0.898 | 0.837 |
As expected, all model performances decreased; however, Top-DTI still outperformed all the methods. In particular, compared to the previous best method, Top-DTI showed an improvement of 4.4% in AUROC and 2.8% in AUPRC.
3.2 Ablation study
To evaluate the contributions of different components of Top-DTI, we performed ablation studies using the unseen drugs and unseen targets datasets. We compared the performance of Top-DTI with seven ablation models, each leveraging distinct combinations of topological and LLM features as follows:
Top_D + Top_T: Integrates only topological features for both drugs and targets.
Top_D + LLM_T: Integrates topological features of drugs with LLM embeddings of targets.
LLM_D + Top_T: Integrates LLM embeddings of drugs with topological features of targets.
LLM_D + LLM_T: Integrates only LLM embeddings for both drugs and targets.
Static Fusion ( ): Integrates LLM embeddings with topological features by assigning equal weight to both modalities with a fixed value of 0.5.
Betti Fusion: Integrates LLM embeddings with only Betti topological features using dynamic values.
PL Fusion: Integrates LLM embeddings with only Persistence Landscape topological features using dynamic values.
Dynamic Fusion (proposed method): Integrates both LLM embeddings and topological features using both Betti and persistence landscape features along with dynamic values.
The performance comparison of the Top-DTI model with different feature types for unseen drugs and unseen targets in the BioSNAP dataset is given in Table 5. The results demonstrate that the model that included only LLM or topological features demonstrated the worst performance, underscoring the limitations of single feature types. Additionally, the integration of drug topological features with LLM embeddings in the unseen drug dataset improves model performance. Furthermore, the Static Fusion model, with an equal weight of 0.5 assigned to each data modality, outperforms single-feature-type methods, highlighting the value of integrated feature approaches. Both Betti Fusion and PL Fusion also surpass single-feature-type models and demonstrate competitive performance with the Static Fusion model. Finally, the Top-DTI model, which dynamically weights LLM and topological features while utilizing both Betti and persistence landscape embeddings, achieves the highest AUROC and AUPRC for both unseen drugs and targets. These results suggest that the adaptive integration of topological features with LLM embeddings further improves performance due to the complementary benefit of both representations.
Table 5.
Performance of the Top-DTI model with ‘Feature Combinations’ on unseen drug and unseen target datasets.
| Features | Unseen Drug |
Unseen Target |
||
|---|---|---|---|---|
| AUROC | AUPRC | AUROC | AUPRC | |
| Top_D + Top_T | ||||
| Top_D + LLM_T | ||||
| LLM_D + Top_T | ||||
| LLM_D + LLM_T | ||||
| Static Fusion () | 0.906 ± 0.004 | 0.904 ± 0.006 | ||
| Betti Fusion | 0.922 ± 0.002 | |||
| PL Fusion | 0.908 ± 0.004 | |||
| Dynamic Fusion (Proposed) | 0.911 ± 0.003 | 0.924 ± 0.002 | 0.907 ± 0.003 | 0.904 ± 0.003 |
The best and second-best results for each evaluation metric was shown in bold and underlined, respectively.
To evaluate Top-DTI’s performance under limited data conditions, we trained multiple models with varying amounts of training data. We observed that as the proportion of training data decreased from 30% to 5%, the performance of all models declined (Supplementary Table S1). However, Top-DTI consistently outperformed baseline methods, demonstrating greater robustness in data-scarce settings.
To evaluate the biological relevance of Top-DTI predictions, we examined the top-ranked DTIs in the BioSNAP unseen drug test set that are not in the ground truth list (i.e. false-positive DTIs). We ranked DTIs by their confidence and selected the top 30 false-positive DTIs. We were able to find support in the literature for five of these DTIs, demonstrating the real-world applicability of Top-DTI in uncovering biologically relevant interactions (Supplementary Table S2).
Top-DTI, despite its multi-component architecture, maintains practical computational efficiency. The most resource-intensive component, PH, scales with a time complexity of approximately for 2D inputs, where P denotes the number of pixels. This implies that the computational load increases nearly quadratically with image size. In our setting, with molecular images and protein contact maps, PH remains tractable. All models were trained using a single NVIDIA A100 GPU (80 GB) and completed within four hours.
In the final step of our analysis, we examined the dynamic weighting of the feature fusion module to understand how the Top-DTI model aligns the topological and LLM-based embeddings in the unseen target dataset. For this, we calculated the mean of the dynamically assigned weights () for both drugs and targets at each epoch during training. The weight vector reflects the relative significance of drug and target feature types throughout the training process (Supplementary Figure S1).
The TOP-DTI model assigns greater weights to LLM-based embeddings, 72% for drugs and 64% for targets (see Supplementary for details). The ablation study also showed that using only LLM-based embeddings performed better than using only topological features (Table 5). These findings demonstrate how each feature type contributes to the TOP-DTI model with the semantic richness of sequential representations and the complementary role of topological features.
4 Conclusion
DTI prediction plays a critical role in the drug development process, as it can improve efficiency while reducing time and costs. In this study, we investigate the integration of TDA with LLMs for the prediction of DTI. We used TDA to extract topological features from 2D representations of drugs and protein targets and utilized LLMs to encode sequence-level information of both drugs and proteins. The proposed model, Top-DTI, significantly enhanced DTI prediction by integrating topological features with sequential embeddings using a dynamic feature fusion module. Future research could further improve DTI prediction by integrating multi-omic data, such as gene expression and proteomics, for targets and molecular features for drugs.
Supplementary Material
Contributor Information
Muhammed Talo, Department of Computer Science and Engineering, University of North Texas, Denton 76207, TX, United States; BioDiscovery Institute, University of North Texas, Denton, TX 76207, United States; Center for Computational Life Sciences, University of North Texas, Denton, TX 76207, United States.
Serdar Bozdag, Department of Computer Science and Engineering, University of North Texas, Denton 76207, TX, United States; BioDiscovery Institute, University of North Texas, Denton, TX 76207, United States; Center for Computational Life Sciences, University of North Texas, Denton, TX 76207, United States; Department of Mathematics, University of North Texas, Denton, TX 76207, United States.
Supplementary data
Supplementary data are available at Bioinformatics online.
Conflict of interest: None declared.
Funding
This work was supported by the National Institute of General Medical Sciences of the National Institutes of Health under Award No. R35GM133657 and the startup funds from the University of North Texas.
Data availability
All benchmark datasets used in this study are publicly available and provided in the datasets folder of the Top-DTI GitHub repository at https://github.com/bozdaglab/Top_DTI.
References
- Ahmad W, Simon E, Chithrananda S et al. Chemberta-2: towards chemical foundation models. ArXiv, 10.48550/arXiv.2209.01712, 2022, preprint: not peer reviewed. [DOI]
- Amézquita E, Nasrin F, Storey K et al. Genomics data analysis via spectral shape and topology. PLoS One 2023;18:e0284820. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Bai P, Miljković F, John B et al. Interpretable bilinear attention network with domain adaptation improves drug–target prediction. Nature Mach Intel 2023;5:126–136. [Google Scholar]
- Beal D. ESM 2.0: state of the art and future potential of experience sampling methods in organizational research. Annu Rev Organ Psychol Organ Behav 2015;2:383–407. [Google Scholar]
- Bubenik P. et al. Statistical topological data analysis using persistence landscapes. J Mach Learn Res 2015;16:77–102. [Google Scholar]
- Chen L, Tan X, Wang D et al. TransformerCPI: improving compound–protein interaction prediction by sequence-based deep learning with self-attention mechanism and label reversal experiments. Bioinformatics 2020;36:4406–4414. [DOI] [PubMed] [Google Scholar]
- Coskunuzer B, Akçora C. Topological methods in machine learning: a tutorial for practitioners. ArXiv, 10.48550/arXiv.2409.02901, 2024, preprint: not peer reviewed. [DOI]
- Demir A, Coskunuzer B, Gel Y et al. ToDD: topological compound fingerprinting in computer-aided drug discovery. Advances In Neural Information Processing Systems. 2022;35:27978–27993. [Google Scholar]
- Elnaggar A, Heinzinger M, Dallago C et al. Prottrans: toward understanding the language of life through self-supervised learning. IEEE Transact Pattern Anal Mach Intel 2021;44:7112–7127. [DOI] [PubMed] [Google Scholar]
- Frey N, Soklaski R, Axelrod S et al. Neural scaling of deep chemical models. Nature Mach Intel 2023;5:1297–1305. [Google Scholar]
- Glatt R, Liu S. Topological data analysis guided segment anything model prompt optimization for zero-shot segmentation in biological imaging. ArXiv, 10.48550/arXiv.2306.17400, 2023, preprint: not peer reviewed. [DOI]
- Hamilton W, Ying Z, Leskovec J. Inductive representation learning on large graphs. arXiv.1706.02216, 10.48550/arXiv.1706.02216, 2017, preprint: not peer reviewed. [DOI] [Google Scholar]
- Hinkson I, Madej B, Stahlberg E. Accelerating therapeutics for opportunities in medicine: a paradigm shift in drug discovery. Front Pharmacol 2020;11:770. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Huang L, Lin J, Liu R et al. CoaDTI: multi-modal co-attention based framework for drug–target interaction annotation. Brief Bioinformat 2022;23:bbac446. [DOI] [PubMed] [Google Scholar]
- Huang K, Xiao C, Glass L et al. MolTrans: molecular interaction transformer for drug–target interaction prediction. Bioinformatics 2021;37:830–836. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Kang H, Goo S, Lee H et al. Fine-tuning of BERT model to accurately predict drug–target interactions. Pharmaceutics 2022;14:1710. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Landrum G. Rdkit documentation. Release 2013;1:4. [Google Scholar]
- Lee J, Jun D, Song I et al. DLM-DTI: a dual language model for the prediction of drug-target interaction with hint-based learning. J Cheminfo 2024;16:14. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Lee I, Keum J, Nam H. DeepConv-DTI: prediction of drug-target interactions via deep learning with convolution on protein sequences. PLoS Comput Biol 2019;15:e1007129. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Liu H, Sun J, Guan J et al. Improving compound–protein interaction prediction by building up highly credible negative samples. Bioinformatics 2015;31:i221–i229. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Luo Z, Wu W, Sun Q et al. Accurate and transferable drug–target interaction prediction with DrugLAMP. Bioinformatics 2024;40:btae693. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Nguyen T, Le H, Quinn T et al. GraphDTA: predicting drug–target binding affinity with graph neural networks. Bioinformatics 2021;37:1140–1147. [DOI] [PubMed] [Google Scholar]
- Öztürk H, Özgür A, Ozkirimli E. DeepDTA: deep drug–target binding affinity prediction. Bioinformatics 2018;34:i821–i829. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Peng L, Liu X, Chen M et al. MGNDTI: a drug-target interaction prediction framework based on multimodal representation learning and the gating mechanism. J Chem Info Model 2024;64:6684–6698. [DOI] [PubMed] [Google Scholar]
- Rao R, Meier J, Sercu T et al. Transformer protein language models are unsupervised structure learners. Biorxiv 2020;2020–12. 10.1101/2020.12.15.422761 [DOI] [Google Scholar]
- Rifaioglu A, Cetin Atalay R, Cansen Kahraman D et al. MDeePred: novel multi-channel protein featurization for deep learning-based binding affinity prediction in drug discovery. Bioinformatics 2021;37:693–704. [DOI] [PubMed] [Google Scholar]
- Romanelli V, Cerchia C, Lavecchia A. Unlocking the potential of generative artificial intelligence in drug discovery. Appl Generat AI 2024;37–63. 10.1007/978-3-031-46238-2_3. [DOI] [Google Scholar]
- Ross J, Belgodere B, Chenthamarakshan V et al. Large-scale chemical language representations capture molecular structure and properties. Nature Mach Intel 2022;4:1256–1264. [Google Scholar]
- Schlander M, Hernandez-Villafuerte K, Cheng C et al. How much does it cost to research and develop a new drug? A systematic review and assessment. Pharmacoeconomics 2021;39:1243–1269. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Singh R, Sledzieski S, Bryson B et al. Contrastive learning in protein language space predicts interactions between drugs and protein targets. Proceed Nat Acad Sci 2023;120:e2220778120. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Speck-Planche A, Kleandrova V, Luan F et al. A ligand-based approach for the in silico discovery of multi-target inhibitors for proteins associated with HIV infection. Molecul BioSyst 2012;8:2188–2196. [DOI] [PubMed] [Google Scholar]
- Yadav A, Ahmed F, Daescu O et al. Histopathological cancer detection with topological signatures. 2023 IEEE International Conference On Bioinformatics And Biomedicine (BIBM) 2023;1610–1619. [Google Scholar]
- Yamanishi Y, Araki M, Gutteridge A et al. Prediction of drug–target interaction networks from the integration of chemical and genomic spaces. Bioinformatics 2008;24:i232–i240. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Yang Y, Zhu Z, Wang X et al. Ligand-based approach for predicting drug targets and for virtual screening against COVID-19. Briefing Bioinformat 2021;22:1053–1064. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Zhao Q, Zhao H, Zheng K et al. HyperAttentionDTI: improving drug–protein interaction prediction by sequence-based deep learning with attention mechanism. Bioinformatics 2022;38:655–662. [DOI] [PubMed] [Google Scholar]
- Zheng X, He S, Song X et al. DTI-RCNN: New efficient hybrid neural network model to predict drug–target interactions. Artificial Neural Networks And Machine Learning–ICANN 2018: 27th International Conference On Artificial Neural Networks, Rhodes, Greece, October 4-7, 2018, Proceedings, Part I 27. pp. 104–114 (2018).
- Zia A, Khamis A, Nichols J et al. Topological deep learning: a review of an emerging paradigm. Artificial Intelligence Review 2024;57:77. [Google Scholar]
- Zitnik M, Sosic R, Leskovec J. BioSNAP datasets: Stanford biomedical network dataset collection. 2018;5:1. http://snap.stanford.edu/biodata. (20 May 2025, date last accessed). [Google Scholar]
- Zixuan E, Qiao G, Wang G et al. GSL-DTI: graph structure learning network for drug-target interaction prediction. Methods 2024;223:136–145. [DOI] [PubMed] [Google Scholar]
Associated Data
This section collects any data citations, data availability statements, or supplementary materials included in this article.
Supplementary Materials
Data Availability Statement
All benchmark datasets used in this study are publicly available and provided in the datasets folder of the Top-DTI GitHub repository at https://github.com/bozdaglab/Top_DTI.


