Unified Biomolecular Trajectory Generation via Pretrained Variational Bridge

ICLR 2026 Conference SubmissionAnonymous Authors
deep generative modelmolecular dynamicstrajectory generationaugmented bridge matchingadjoint matching
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.

Disclaimer
This report is AI-GENERATED using Large Language Models and WisPaper (A scholar search engine). It analyzes academic papers' tasks and contributions against retrieved prior work. While this system identifies POTENTIAL overlaps and novel directions, ITS COVERAGE IS NOT EXHAUSTIVE AND JUDGMENTS ARE APPROXIMATE. These results are intended to assist human reviewers and SHOULD NOT be relied upon as a definitive verdict on novelty.
NOTE that some papers exist in multiple, slightly different versions (e.g., with different titles or URLs). The system may retrieve several versions of the same underlying work. The current automated pipeline does not reliably align or distinguish these cases, so human reviewers will need to disambiguate them manually.
If you have any questions, please contact: mingzhang23@m.fudan.edu.cn

Overview

Overall Novelty Assessment

The paper introduces a Pretrained Variational Bridge (PVB) framework that unifies training on single-structure and paired trajectory data through augmented bridge matching, with reinforcement learning-based optimization for protein-ligand complexes. It resides in the 'Diffusion-Based Generative Models for Biomolecular Dynamics' leaf, which contains five papers including the original work. This represents a moderately populated research direction within the broader conformational sampling branch, suggesting active but not overcrowded exploration of diffusion-based approaches for molecular trajectory generation.

The taxonomy reveals neighboring leaves focused on autoencoder-based conformational ensembles, generative Markov state models, and comparative benchmarking studies. The paper's bridge matching approach connects to the diffusion paradigm shared by sibling works like Diffusion Protein Conformations and Dynamics Diffusion, yet diverges by explicitly unifying cross-domain structural knowledge across training stages. The scope note clarifies this leaf excludes VAE-based or GAN-based methods, positioning the work firmly within the diffusion modeling tradition while the reinforcement learning component introduces elements typically associated with optimization-focused branches.

Among twelve candidates examined across three contributions, no clearly refutable prior work was identified. The unified training framework examined one candidate with no refutations, the RL-based adjoint matching examined one candidate with no refutations, and the cross-domain trajectory generation demonstration examined ten candidates with no refutations. This limited search scope—twelve papers from semantic retrieval—suggests the specific combination of pretrained variational bridges with adjoint matching for protein-ligand systems may occupy a relatively unexplored niche, though the absence of refutations does not confirm comprehensive novelty given the constrained candidate pool.

Based on top-twelve semantic matches, the work appears to introduce distinctive methodological elements within an active research area. The analysis covers diffusion-based trajectory generation but does not exhaustively survey all enhanced sampling or force field approaches that might address similar acceleration goals through different paradigms. The taxonomy structure indicates this sits at the intersection of generative modeling and molecular dynamics, where rapid methodological evolution makes definitive novelty assessments challenging without broader literature coverage.

Taxonomy

Core-task Taxonomy Papers
50
3
Claimed Contributions
12
Contribution Candidate Papers Compared
0
Refutable Paper

Research Landscape Overview

Core task: accelerating molecular dynamics simulations through deep generative modeling. The field has organized itself into several major branches that reflect different strategies for leveraging deep learning to overcome the timescale limitations of traditional MD. One branch focuses on conformational sampling and trajectory generation, where models learn to produce realistic molecular configurations or entire dynamical pathways without exhaustive simulation. Another branch emphasizes enhanced sampling and free energy calculation, using neural networks to identify collective variables or reweight biased simulations. A third branch develops machine learning force fields that replace expensive quantum calculations with fast learned potentials, while additional branches address property prediction, drug discovery applications, and methodological reviews. Representative works such as Diffusion Protein Conformations[11] and Dynamics Diffusion[37] illustrate how diffusion-based architectures have become prominent tools for generating biomolecular ensembles, whereas efforts like Deep Boosted MD[15] and Unbiasing Enhanced Sampling[10] show how learning can guide or correct biased sampling schemes. Within the conformational sampling branch, diffusion-based generative models have emerged as a particularly active line of work, balancing the need for physical realism with computational efficiency. These methods must navigate trade-offs between sampling speed, conformational diversity, and adherence to underlying energy landscapes. Biomolecular Variational Bridge[0] sits squarely in this diffusion-focused cluster, sharing methodological DNA with Diffusion Protein Conformations[11] and Dynamics Diffusion[37], which similarly apply diffusion frameworks to generate protein or molecular trajectories. Compared to DynamicBind[1], which targets dynamic protein-ligand binding, Biomolecular Variational Bridge[0] appears to emphasize broader conformational exploration rather than binding-specific dynamics. Meanwhile, Membrane Partitioning Diffusion[26] applies diffusion models to membrane systems, highlighting how these generative techniques are being adapted across diverse biomolecular contexts. The central challenge remains ensuring that generated samples faithfully represent thermodynamic ensembles while achieving meaningful speedups over conventional simulation.

Claimed Contributions

Pretrained Variational Bridge (PVB) with unified training framework

The authors introduce PVB, which uses an encoder-decoder architecture combined with augmented bridge matching to create a unified objective for pretraining on single-structure data and finetuning on MD trajectory pairs. This design enables consistent exploitation of cross-domain structural knowledge across training stages.

1 retrieved paper
RL-based finetuning via adjoint matching for protein-ligand complexes

The authors develop a reinforcement learning finetuning method using adjoint matching and stochastic optimal control. This approach modulates the generative distribution with explicit reward functions to accelerate progression toward protein-ligand holo states, enabling efficient post-optimization of docking poses.

1 retrieved paper
Demonstration of stable and efficient cross-domain trajectory generation

The authors show that PVB faithfully reproduces thermodynamic and kinetic observables from molecular dynamics simulations across protein monomers and protein-ligand complexes, while providing more stable trajectory generation compared to baseline methods.

10 retrieved papers

Core Task Comparisons

Comparisons with papers in the same taxonomy category

Contribution Analysis

Detailed comparisons for each claimed contribution

Contribution

Pretrained Variational Bridge (PVB) with unified training framework

The authors introduce PVB, which uses an encoder-decoder architecture combined with augmented bridge matching to create a unified objective for pretraining on single-structure data and finetuning on MD trajectory pairs. This design enables consistent exploitation of cross-domain structural knowledge across training stages.

Contribution

RL-based finetuning via adjoint matching for protein-ligand complexes

The authors develop a reinforcement learning finetuning method using adjoint matching and stochastic optimal control. This approach modulates the generative distribution with explicit reward functions to accelerate progression toward protein-ligand holo states, enabling efficient post-optimization of docking poses.

Contribution

Demonstration of stable and efficient cross-domain trajectory generation

The authors show that PVB faithfully reproduces thermodynamic and kinetic observables from molecular dynamics simulations across protein monomers and protein-ligand complexes, while providing more stable trajectory generation compared to baseline methods.

Unified Biomolecular Trajectory Generation via Pretrained Variational Bridge | Novelty Validation