# Piloting Structure-Based Drug Design via Modality-Specific Optimal Schedule

arXiv: https://arxiv.org/abs/2505.07286 · HTML: https://mlsanity.com/paper/2505.07286/

## Key facts

- Cited by 6 papers in the map (data through 14 Oct 2026).
- Builds on 16 of its 33 references that are in the map.
- First posted on arXiv 12 May 2025.

## Builds on (most cited first, 16 of 16)

- [Variational Diffusion Models](https://mlsanity.com/paper/2107.00630/) (2021)
- [DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking](https://mlsanity.com/paper/2210.01776/) (2022)
- [Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Design](https://mlsanity.com/paper/2402.04997/) (2024)
- [Continuous diffusion for categorical data](https://mlsanity.com/paper/2211.15089/) (2022)
- [Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pockets](https://mlsanity.com/paper/2205.07249/) (2022)
- [3D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Prediction](https://mlsanity.com/paper/2303.03543/) (2023)
- [A 3D Generative Model for Structure-Based Drug Design](https://mlsanity.com/paper/2203.10446/) (2022)
- [Generating 3D Molecules for Target Protein Binding](https://mlsanity.com/paper/2204.09410/) (2022)
- [DecompDiff: Diffusion Models with Decomposed Priors for Structure-Based Drug Design](https://mlsanity.com/paper/2403.07902/) (2024)
- [MolDiff: Addressing the Atom-Bond Inconsistency Problem in 3D Molecule Diffusion Generation](https://mlsanity.com/paper/2305.07508/) (2023)
- [Bayesian Flow Networks](https://mlsanity.com/paper/2308.07037/) (2023)
- [Equivariant Flow Matching with Hybrid Probability Transport](https://mlsanity.com/paper/2312.07168/) (2023)
- [MolCRAFT: Structure-Based Drug Design in Continuous Parameter Space](https://mlsanity.com/paper/2404.12141/) (2024)
- [Structure-based drug design with geometric deep learning](https://mlsanity.com/paper/2210.11250/) (2022)
- [Functional-Group-Based Diffusion for Pocket-Specific Molecule Generation and Elaboration](https://mlsanity.com/paper/2306.13769/) (2023)
- [Unified Generative Modeling of 3D Molecules via Bayesian Flow Networks](https://mlsanity.com/paper/2403.15441/) (2024)

## Built on by (most cited first, 6 of 6)

- [Kinetic Langevin Diffusion for Crystalline Materials Generation](https://mlsanity.com/paper/2507.03602/) (2025)
- [MolPIF: A Parameter Interpolation Flow Model for Molecule Generation](https://mlsanity.com/paper/2507.13762/) (2025)
- [SiDGen: Structure-informed Diffusion for Generative modeling of Ligands for Proteins](https://mlsanity.com/paper/2511.09529/) (2025)
- [EvoEGF-Mol: Evolving Exponential Geodesic Flow for Structure-based Drug Design](https://mlsanity.com/paper/2601.22466/) (2026)
- [Probe Before You Edit: Probing-Guided Molecular Optimization for LLM Agents in Structure-Based Drug Design](https://mlsanity.com/paper/2606.00555/) (2026)
- [Demystifying Multimodal Biomolecular Co-design With Intrinsic Geodesic Coupling](https://mlsanity.com/paper/2606.01628/) (2026)

## Abstract (arXiv)

Structure-Based Drug Design (SBDD) is crucial for identifying bioactive molecules. Recent deep generative models are faced with challenges in geometric structure modeling. A major bottleneck lies in the twisted probability path of multi-modalities -- continuous 3D positions and discrete 2D topologies -- which jointly determine molecular geometries. By establishing the fact that noise schedules decide the Variational Lower Bound (VLB) for the twisted probability path, we propose VLB-Optimal Scheduling (VOS) strategy in this under-explored area, which optimizes VLB as a path integral for SBDD. Our model effectively enhances molecular geometries and interaction modeling, achieving state-of-the-art PoseBusters passing rate of 95.9% on CrossDock, more than 10% improvement upon strong baselines, while maintaining high affinities and robust intramolecular validity evaluated on held-out test set. Code is available at https://github.com/AlgoMole/MolCRAFT.
