Unified Biomolecular Trajectory Generation via Pretrained Variational Bridge
Ziyang Yu, Wenbing Huang, Yang Liu
Abstract
Molecular Dynamics (MD) simulations provide a fundamental tool for characterizing molecular behavior at full atomic resolution, but their applicability is severely constrained by the computational cost. To address this, a surge of deep generative models has recently emerged to learn dynamics at coarsened timesteps for efficient trajectory generation, yet they either generalize poorly across systems or, due to limited molecular diversity of trajectory data, fail to fully exploit structural information to improve generative fidelity. Here, we present the Pretrained Variational Bridge (PVB) in an encoder-decoder fashion, which maps the initial structure into a noised latent space and transports it toward stage-specific targets through augmented bridge matching. This unifies training on both single-structure and paired trajectory data, enabling consistent use of cross-domain structural knowledge across training stages. Moreover, for protein-ligand complexes, we further introduce a reinforcement learning-based optimization via adjoint matching that speeds progression toward the holo state, which supports efficient post-optimization of docking poses. Experiments on proteins and protein-ligand complexes demonstrate that PVB faithfully reproduces thermodynamic and kinetic observables from MD while delivering stable and efficient generative dynamics.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ebcab9d9-52d9-45b1-b6ec-0607bcf98cf3Builds on11
- DiffDock: Diffusion Steps, Twists, and Turns for Molecular DockingGabriele Corso, Hannes Stärk, Bowen Jing, Regina Barzilay et al.ICLR 2023 · 331 citations
- AlphaFold Meets Flow Matching for Generating Protein EnsemblesBowen Jing, Bonnie Berger, Tommi S. JaakkolaICML 2024 · 229 citations
- Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened DynamicsLeon Klein, Andrew Y. K. Foong, Tor Erlend Fjelde, Bruno Mlodozeniec et al.NeurIPS 2023 · 110 citations
- Generative Modeling of Molecular Dynamics TrajectoriesBowen Jing, Hannes Stärk, Tommi S. Jaakkola, Bonnie BergerNeurIPS 2024 · 97 citations
- Protein Conformation Generation via Force-Guided SE(3) Diffusion ModelsYan Wang, Lihao Wang, Yuning Shen, Yiqun Wang et al.ICML 2024 · 65 citations
Related papers
- Align Your Structures: Generating Trajectories with Structure Pretraining for Molecular DynamicsAniketh Iyengar, Jiaqi Han, Pengwei Sun, Mingjian Jiang et al.ICLR 2026 · 1 citation
- UniSim: A Unified Simulator for Time-Coarsened Dynamics of BiomoleculesZiyang Yu, Wenbing Huang, Yang LiuICML 2025
- LIMO: Latent Inceptionism for Targeted Molecule GenerationPeter Eckmann, Kunyang Sun, Bo Zhao, Mudong Feng et al.ICML 2022 · 60 citations
- ProAR: Probabilistic Autoregressive Modeling for Molecular DynamicsKaiwen Cheng, Yutian Liu, Zhiwei Nie, Mujie Lin et al.AAAI 2026
- Reinforced Genetic Algorithm for Structure-based Drug DesignTianfan Fu, Wenhao Gao, Connor W. Coley, Jimeng SunNeurIPS 2022 · 79 citations
