Multi-task Bioassay Pre-training for Protein-ligand Binding Affinity Prediction
Not on the map. Why?
Abstract (arXiv)
Protein-ligand binding affinity (PLBA) prediction is the fundamental task in drug discovery. Recently, various deep learning-based models predict binding affinity by incorporating the three-dimensional structure of protein-ligand complexes as input and achieving astounding progress. However, due to the scarcity of high-quality training data, the generalization ability of current models is still limited. In addition, different bioassays use varying affinity measurement labels (i.e., IC50, Ki, Kd), and different experimental conditions inevitably introduce systematic noise, which poses a significant challenge to constructing high-precision affinity prediction models. To address these issues, we (1) propose Multi-task Bioassay Pre-training (MBP), a pre-training framework for structure-based PLBA prediction; (2) construct a pre-training dataset called ChEMBL-Dock with more than 300k experimentally measured affinity labels and about 2.8M docked three-dimensional structures. By introducing multi-task pre-training to treat the prediction of different affinity labels as different tasks and classifying relative rankings between samples from the same bioassay, MBP learns robust and transferrable structural knowledge from our new ChEMBL-Dock dataset with varied and noisy labels. Experiments substantiate the capability of MBP as a general framework that can improve and be tailored to mainstream structure-based PLBA prediction tasks. To the best of our knowledge, MBP is the first affinity pre-training model and shows great potential for future development.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 32 of its 108 references that are in the map. First posted on arXiv 8 Jun 2023.
Builds on
32 of 108 references.
References · 32
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2018How Powerful are Graph Neural Networks?cited by 4,355
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2019BioBERT: a pre-trained biomedical language representation model for biomedical text miningcited by 1,935
- 2018Representation Learning on Graphs with Jumping Knowledge Networkscited by 875
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
- 2019Analyzing Learned Molecular Representations for Property Predictioncited by 373
- 2022DiffDock: Diffusion Steps, Twists, and Turns for Molecular Dockingcited by 307
20 more
- 2021ChemRL-GEM: Geometry Enhanced Molecular Representation Learning for Property Predictioncited by 204
- 2022Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pocketscited by 182
- 2020ProtTrans: Towards Cracking the Language of Life's Code Through Self-Supervised Deep Learning and High Performance Computingcited by 162
- 2021Motif-based Graph Self-Supervised Learning for Molecular Property Predictioncited by 159
- 2022EquiBind: Geometric Deep Learning for Drug Binding Structure Predictioncited by 143
- 2020Molecule Attention Transformercited by 92
- 2021Independent SE(3)-Equivariant Models for End-to-End Rigid Protein Dockingcited by 81
- 2022One Transformer Can Understand Both 2D & 3D Molecular Datacited by 77
- 2021Structure-aware Interactive Graph Neural Networks for the Prediction of Protein-Ligand Binding Affinitycited by 60
- 2020Robust Optimization as Data Augmentation for Large-scale Graphscited by 58
- 2022Geometrically Equivariant Graph Neural Networks: A Surveycited by 55
- 2017Development and evaluation of a deep learning model for protein-ligand binding affinity predictioncited by 50
- 2022Unified 2D and 3D Pre-Training of Molecular Representationscited by 45
- 2020Improved Protein-ligand Binding Affinity Prediction with Structure-Based Deep Fusion Inferencecited by 35
- 2016Accurate De Novo Prediction of Protein Contact Map by Ultra-Deep Learning Modelcited by 31
- 2020PIGNet: A physics-informed deep learning model toward generalized drug-target interaction predictionscited by 31
- 2019Pairwise Learning to Rank by Neural Networks Revisited: Reconstruction, Theoretical Analysis and Practical Performancecited by 13
- 2022Efficient and Accurate Physics-aware Multiplex Graph Neural Networks for 3D Small Molecules and Macromolecule Complexescited by 6
- 2022Predicting Protein-Ligand Binding Affinity via Joint Global-Local Interaction Modelingcited by 5
- 2022MP2: A Momentum Contrast Approach for Recommendation with Pointwise and Pairwise Learningcited by 3
Built on by
Citing papers per year: 2023: 2; 2024: 0; 2025: 3; 2026: 2.
7 papers.
- 2023Geometric Deep Learning for Structure-Based Drug Design: A Surveycited by 2
- 2025Fast and Accurate Blind Flexible Dockingcited by 2
- 2025Drug-Target Interaction/Affinity Prediction: Deep Learning Models and Advances Reviewcited by 2
- 2023Embracing assay heterogeneity with neural processes for markedly improved bioactivity predictionscited by 1
- 2025A Generalist Cross-Domain Molecular Learning Framework for Structure-Based Drug Discoverycited by 0
- 2026Rigidity-Aware Geometric Pretraining for Protein Design and Conformational Ensemblescited by 0
- 2026HBGSA: Hydrogen Bond Graph with Self-Attention for Drug-Target Binding Affinity Predictioncited by 0
- 2026HBGSA: Hydrogen Bond Graph with Self-Attention for Drug-Target Binding Affinity Prediction25 Apr 2026
- 2026Rigidity-Aware Geometric Pretraining for Protein Design and Conformational Ensembles2 Mar 2026
- 2025A Generalist Cross-Domain Molecular Learning Framework for Structure-Based Drug Discovery6 Mar 2025
- 2025Drug-Target Interaction/Affinity Prediction: Deep Learning Models and Advances Review21 Feb 2025
- 2025Fast and Accurate Blind Flexible Docking20 Feb 2025
- 2023Embracing assay heterogeneity with neural processes for markedly improved bioactivity predictions17 Aug 2023
- 2023Geometric Deep Learning for Structure-Based Drug Design: A Survey20 Jun 2023