Saved in:
Bibliographic Details
Main Authors: Dunn, Ian, Toft, Liv, Katz, Tyler, Gupta, Juhi, Shah, Riya, Hettiarachchi, Ramith, Koes, David R.
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.05080
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909943852957696
author Dunn, Ian
Toft, Liv
Katz, Tyler
Gupta, Juhi
Shah, Riya
Hettiarachchi, Ramith
Koes, David R.
author_facet Dunn, Ian
Toft, Liv
Katz, Tyler
Gupta, Juhi
Shah, Riya
Hettiarachchi, Ramith
Koes, David R.
contents Structure-based drug design (SBDD) focuses on designing small-molecule ligands that bind to specific protein pockets. Computational methods are integral in modern SBDD workflows and often make use of virtual screening methods via docking or pharmacophore search. Modern generative modeling approaches have focused on improving novel ligand discovery by enabling de novo design. In this work, we recognize that these tasks share a common structure and can therefore be represented as different instantiations of a consistent generative modeling framework. We propose a unified approach in OMTRA, a multi-modal flow matching model that flexibly performs many tasks relevant to SBDD, including some with no analogue in conventional workflows. Additionally, we curate a dataset of 500M 3D molecular conformers, complementing protein-ligand data and expanding the chemical diversity available for training. OMTRA obtains state of the art performance on pocket-conditioned de novo design and docking; however, the effects of large-scale pretraining and multi-task training are modest. All code, trained models, and dataset for reproducing this work are available at https://github.com/gnina/OMTRA
format Preprint
id arxiv_https___arxiv_org_abs_2512_05080
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle OMTRA: A Multi-Task Generative Model for Structure-Based Drug Design
Dunn, Ian
Toft, Liv
Katz, Tyler
Gupta, Juhi
Shah, Riya
Hettiarachchi, Ramith
Koes, David R.
Machine Learning
Structure-based drug design (SBDD) focuses on designing small-molecule ligands that bind to specific protein pockets. Computational methods are integral in modern SBDD workflows and often make use of virtual screening methods via docking or pharmacophore search. Modern generative modeling approaches have focused on improving novel ligand discovery by enabling de novo design. In this work, we recognize that these tasks share a common structure and can therefore be represented as different instantiations of a consistent generative modeling framework. We propose a unified approach in OMTRA, a multi-modal flow matching model that flexibly performs many tasks relevant to SBDD, including some with no analogue in conventional workflows. Additionally, we curate a dataset of 500M 3D molecular conformers, complementing protein-ligand data and expanding the chemical diversity available for training. OMTRA obtains state of the art performance on pocket-conditioned de novo design and docking; however, the effects of large-scale pretraining and multi-task training are modest. All code, trained models, and dataset for reproducing this work are available at https://github.com/gnina/OMTRA
title OMTRA: A Multi-Task Generative Model for Structure-Based Drug Design
topic Machine Learning
url https://arxiv.org/abs/2512.05080