IBEX: Information-Bottleneck-EXplored Coarse-to-Fine Molecular Generation under Limited Data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xu, Dong, Yang, Zhangfan, Yao, Jenna Xinyi, Song, Shuangbao, Zhu, Zexuan, Ji, Junkai
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909736993030144
author Xu, Dong
Yang, Zhangfan
Yao, Jenna Xinyi
Song, Shuangbao
Zhu, Zexuan
Ji, Junkai
author_facet Xu, Dong
Yang, Zhangfan
Yao, Jenna Xinyi
Song, Shuangbao
Zhu, Zexuan
Ji, Junkai
contents Three-dimensional generative models increasingly drive structure-based drug discovery, yet it remains constrained by the scarce publicly available protein-ligand complexes. Under such data scarcity, almost all existing pipelines struggle to learn transferable geometric priors and consequently overfit to training-set biases. As such, we present IBEX, an Information-Bottleneck-EXplored coarse-to-fine pipeline to tackle the chronic shortage of protein-ligand complex data in structure-based drug design. Specifically, we use PAC-Bayesian information-bottleneck theory to quantify the information density of each sample. This analysis reveals how different masking strategies affect generalization and indicates that, compared with conventional de novo generation, the constrained Scaffold Hopping task endows the model with greater effective capacity and improved transfer performance. IBEX retains the original TargetDiff architecture and hyperparameters for training to generate molecules compatible with the binding pocket; it then applies an L-BFGS optimization step to finely refine each conformation by optimizing five physics-based terms and adjusting six translational and rotational degrees of freedom in under one second. With only these modifications, IBEX raises the zero-shot docking success rate on CBGBench CrossDocked2020-based from 53% to 64%, improves the mean Vina score from $-7.41 kcal mol^{-1}$ to $-8.07 kcal mol^{-1}$, and achieves the best median Vina energy in 57 of 100 pockets versus 3 for the original TargetDiff. IBEX also increases the QED by 25%, achieves state-of-the-art validity and diversity, and markedly reduces extrapolation error.
format Preprint
id arxiv_https___arxiv_org_abs_2508_10775
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle IBEX: Information-Bottleneck-EXplored Coarse-to-Fine Molecular Generation under Limited Data
Xu, Dong
Yang, Zhangfan
Yao, Jenna Xinyi
Song, Shuangbao
Zhu, Zexuan
Ji, Junkai
Machine Learning
Biomolecules
Three-dimensional generative models increasingly drive structure-based drug discovery, yet it remains constrained by the scarce publicly available protein-ligand complexes. Under such data scarcity, almost all existing pipelines struggle to learn transferable geometric priors and consequently overfit to training-set biases. As such, we present IBEX, an Information-Bottleneck-EXplored coarse-to-fine pipeline to tackle the chronic shortage of protein-ligand complex data in structure-based drug design. Specifically, we use PAC-Bayesian information-bottleneck theory to quantify the information density of each sample. This analysis reveals how different masking strategies affect generalization and indicates that, compared with conventional de novo generation, the constrained Scaffold Hopping task endows the model with greater effective capacity and improved transfer performance. IBEX retains the original TargetDiff architecture and hyperparameters for training to generate molecules compatible with the binding pocket; it then applies an L-BFGS optimization step to finely refine each conformation by optimizing five physics-based terms and adjusting six translational and rotational degrees of freedom in under one second. With only these modifications, IBEX raises the zero-shot docking success rate on CBGBench CrossDocked2020-based from 53% to 64%, improves the mean Vina score from $-7.41 kcal mol^{-1}$ to $-8.07 kcal mol^{-1}$, and achieves the best median Vina energy in 57 of 100 pockets versus 3 for the original TargetDiff. IBEX also increases the QED by 25%, achieves state-of-the-art validity and diversity, and markedly reduces extrapolation error.
title IBEX: Information-Bottleneck-EXplored Coarse-to-Fine Molecular Generation under Limited Data
topic Machine Learning
Biomolecules
url https://arxiv.org/abs/2508.10775