TSEDTA: a transformer-based neural network with SMILES transformer and ESM2 embeddings for drug-target binding affinity prediction
Abstract
<h4>Motivation</h4>Drug-target binding affinity (DTA) prediction plays a vital role in drug repositioning. The emergence of large language models (LLMs) has introduced new perspectives for predicting DTA. Herein, we present TSEDTA, a Transformer-based neural network with SMILES Transformer and ESM2 embeddings for predicting DTA. It leverages pre-trained LLMs (SMILES Transformer and ESM2) to extract deep evolutionary representations from drug SMILES and protein sequences. The representations are directly fused with raw sequence embeddings and processed via dual Transformer encoders to capture complex local and global dependencies.<h4>Results</h4>The experiments demonstrate that TSEDTA outperforms ten advanced models on the Davis and KIBA datasets, and seven on the BindingDB dataset. Ablation studies show that incorporating LLM embeddings significantly improves the performance of TSEDTA. Furthermore, a practical case study demonstrates its real-world applicability. Ultimately, TSEDTA provides a highly accurate, robust tool for DTA prediction, offering new insights into the application of LLMs for DTA tasks.<h4>Availability</h4>The source code and data are available at: https://github.com/SunXu24Math/TSEDTA. The version of record is archived in Zenodo with the DOI: 10.5281/zenodo.19103249.