Equivariant Scalar Fields for Molecular Docking with Fast Fourier Transforms

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Jing, Bowen, Jaakkola, Tommi, Berger, Bonnie
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910584907235328
author Jing, Bowen
Jaakkola, Tommi
Berger, Bonnie
author_facet Jing, Bowen
Jaakkola, Tommi
Berger, Bonnie
contents Molecular docking is critical to structure-based virtual screening, yet the throughput of such workflows is limited by the expensive optimization of scoring functions involved in most docking algorithms. We explore how machine learning can accelerate this process by learning a scoring function with a functional form that allows for more rapid optimization. Specifically, we define the scoring function to be the cross-correlation of multi-channel ligand and protein scalar fields parameterized by equivariant graph neural networks, enabling rapid optimization over rigid-body degrees of freedom with fast Fourier transforms. The runtime of our approach can be amortized at several levels of abstraction, and is particularly favorable for virtual screening settings with a common binding pocket. We benchmark our scoring functions on two simplified docking-related tasks: decoy pose scoring and rigid conformer docking. Our method attains similar but faster performance on crystal structures compared to the widely-used Vina and Gnina scoring functions, and is more robust on computationally predicted structures. Code is available at https://github.com/bjing2016/scalar-fields.
format Preprint
id arxiv_https___arxiv_org_abs_2312_04323
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Equivariant Scalar Fields for Molecular Docking with Fast Fourier Transforms
Jing, Bowen
Jaakkola, Tommi
Berger, Bonnie
Biomolecules
Machine Learning
Molecular docking is critical to structure-based virtual screening, yet the throughput of such workflows is limited by the expensive optimization of scoring functions involved in most docking algorithms. We explore how machine learning can accelerate this process by learning a scoring function with a functional form that allows for more rapid optimization. Specifically, we define the scoring function to be the cross-correlation of multi-channel ligand and protein scalar fields parameterized by equivariant graph neural networks, enabling rapid optimization over rigid-body degrees of freedom with fast Fourier transforms. The runtime of our approach can be amortized at several levels of abstraction, and is particularly favorable for virtual screening settings with a common binding pocket. We benchmark our scoring functions on two simplified docking-related tasks: decoy pose scoring and rigid conformer docking. Our method attains similar but faster performance on crystal structures compared to the widely-used Vina and Gnina scoring functions, and is more robust on computationally predicted structures. Code is available at https://github.com/bjing2016/scalar-fields.
title Equivariant Scalar Fields for Molecular Docking with Fast Fourier Transforms
topic Biomolecules
Machine Learning
url https://arxiv.org/abs/2312.04323