Improving generalisability of 3D binding affinity models in low data regimes

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Buhmann, Julia, Haddadin, Ward, Pravda, Lukáš, Bilsland, Alan, Triendl, Hagen
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912036314677248
author Buhmann, Julia
Haddadin, Ward
Pravda, Lukáš
Bilsland, Alan
Triendl, Hagen
author_facet Buhmann, Julia
Haddadin, Ward
Pravda, Lukáš
Bilsland, Alan
Triendl, Hagen
contents Predicting protein-ligand binding affinity is an essential part of computer-aided drug design. However, generalisable and performant global binding affinity models remain elusive, particularly in low data regimes. Despite the evolution of model architectures, current benchmarks are not well-suited to probe the generalisability of 3D binding affinity models. Furthermore, 3D global architectures such as GNNs have not lived up to performance expectations. To investigate these issues, we introduce a novel split of the PDBBind dataset, minimizing similarity leakage between train and test sets and allowing for a fair and direct comparison between various model architectures. On this low similarity split, we demonstrate that, in general, 3D global models are superior to protein-specific local models in low data regimes. We also demonstrate that the performance of GNNs benefits from three novel contributions: supervised pre-training via quantum mechanical data, unsupervised pre-training via small molecule diffusion, and explicitly modeling hydrogen atoms in the input graph. We believe that this work introduces promising new approaches to unlock the potential of GNN architectures for binding affinity modelling.
format Preprint
id arxiv_https___arxiv_org_abs_2409_12995
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Improving generalisability of 3D binding affinity models in low data regimes
Buhmann, Julia
Haddadin, Ward
Pravda, Lukáš
Bilsland, Alan
Triendl, Hagen
Machine Learning
Artificial Intelligence
Predicting protein-ligand binding affinity is an essential part of computer-aided drug design. However, generalisable and performant global binding affinity models remain elusive, particularly in low data regimes. Despite the evolution of model architectures, current benchmarks are not well-suited to probe the generalisability of 3D binding affinity models. Furthermore, 3D global architectures such as GNNs have not lived up to performance expectations. To investigate these issues, we introduce a novel split of the PDBBind dataset, minimizing similarity leakage between train and test sets and allowing for a fair and direct comparison between various model architectures. On this low similarity split, we demonstrate that, in general, 3D global models are superior to protein-specific local models in low data regimes. We also demonstrate that the performance of GNNs benefits from three novel contributions: supervised pre-training via quantum mechanical data, unsupervised pre-training via small molecule diffusion, and explicitly modeling hydrogen atoms in the input graph. We believe that this work introduces promising new approaches to unlock the potential of GNN architectures for binding affinity modelling.
title Improving generalisability of 3D binding affinity models in low data regimes
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2409.12995