Binding Affinity Prediction with 3D Machine Learning: Training Data and Challenging External Testing

Jose Carlos Gómez-Tamayo,Gary Tresadern,Lili Cao,Mazen Ahmad

DOI: https://doi.org/10.26434/chemrxiv-2024-s5sbs

2024-02-01

Abstract:Protein-ligand binding affinity prediction is one of the major challenges in computational assisted drug discovery. An active area of research uses machine learning (ML) models trained on 3D structures of protein ligand complexes to predict binding modes, discriminate active and inactives, or predict affinity. Methodological advances in deep learning, and artificial intelligence along with increased experimental data (3D structures and bioactivities) has led to many studies using different architectures, representation, and features. Unfortunately, many models do not learn details of interactions or the underlying physics that drive protein-ligand affinity, but instead just memorize patterns in the available training data with poor generalizability and future use. In this work we incorporate “dense”, feature rich datasets that contain up to several thousand analogue molecules per drug discovery target. For the training set, PDBbind dataset is used with enrichment from 8 internal lead optimization (LO) datasets and inactive and decoy poses in a variety of combinations. A variety of different model architectures was used and the model performance was validated using the binding affinity for 12 internal LO and 6 ChEMBL external test sets. Results show a significant improvement in the performance and generalization power, especially for virtual screening and suggest promise for the future of ML protein-ligand affinity prediction with a greater emphasis on training using datasets that capture the rich details of the affinity landscape.

Chemistry

What problem does this paper attempt to address?

This paper focuses on the problem of predicting protein-ligand binding affinity, which is an important challenge in computational-aided drug discovery. Current research methods often utilize machine learning (ML) models, particularly deep learning (DL) models, but they often only memorize the patterns in the training data, lacking the learning of physical interactions and details that drive protein-ligand affinity, thus lacking generalization ability.

The paper proposes a new approach to improve training by using a "dense" feature-rich dataset containing thousands of similar molecules targeting each drug discovery target. The study uses the PDBbind database as a foundation and combines it with internal Lead Optimization (LO) datasets and the ChEMBL dataset to increase data diversity and coverage. Through this approach, they reduce data sparsity and improve model performance and generalization ability.

The paper also discusses how to use approximate co-crystal structures generated through molecular docking (such as maximum common substructure docking) and introduce ligand efficiency to prevent overfitting, while using virtual screening and external test sets to validate the predictive ability of the model. The results indicate that this approach has potential for predicting protein-ligand affinity in the future, especially when training with datasets that emphasize the capturing of rich details in the affinity landscape.