Beyond Ensembles: Simulating All-Atom Protein Dynamics in a Learned Latent Space
Aditya Sengar, Jiying Zhang, Pierre Vandergheynst, Patrick Barth
Abstract
Simulating the long-timescale dynamics of biomolecules is a central challenge in computational science. While enhanced sampling methods can accelerate these simulations, they rely on pre-defined collective variables that are often difficult to identify, restricting their ability to model complex switching mechanisms between metastable states. A recent generative model, LD-FPG, demonstrated that this problem could be bypassed by learning to sample the static equilibrium ensemble as all-atom deformations from a reference structure, establishing a powerful method for all-atom ensemble generation. However, while this approach successfully captures a system's probable conformations, it does not model the temporal evolution between them. We introduce the Graph Latent Dynamics Propagator (GLDP), a modular component for simulating dynamics within the learned latent space of LD-FPG. We then compare three classes of propagators: (i) score-guided Langevin dynamics, (ii) Koopman-based linear operators, and (iii) autoregressive neural networks. Within a unified encoder–propagator–decoder framework, we evaluate long-horizon stability, backbone and side-chain ensemble fidelity, and temporal kinetics via TICA. Benchmarks on systems ranging from small peptides to mixed-topology proteins and large GPCRs reveal that autoregressive neural networks deliver the most robust long rollouts and coherent physical timescales; score-guided Langevin best recovers side-chain thermodynamics when the score is well learned; and Koopman provides an interpretable, lightweight baseline that tends to damp fluctuations. These results clarify the trade-offs among propagators and offer practical guidance for latent-space simulators of all-atom protein 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 e362976a-856f-4dea-9b58-17689ebd22d7Builds on12
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- GeoDiff: A Geometric Diffusion Model for Molecular Conformation GenerationMinkai Xu, Lantao Yu, Yang Song, Chence Shi et al.ICLR 2022 · 695 citations
- Neural SDEs as Infinite-Dimensional GANsPatrick Kidger, James Foster, Xuechen Li, Terry J. LyonsICML 2021 · 214 citations
- Forecasting Sequential Data Using Consistent Koopman AutoencodersOmri Azencot, N. Benjamin Erichson, Vanessa Lin, Michael W. MahoneyICML 2020 · 203 citations
Related papers
- Generative Modeling of Full-Atom Protein Conformations using Latent Diffusion on Graph EmbeddingsAditya Sengar, Ali Hariri, Daniel Probst, Patrick Barth et al.NeurIPS 2025 · 7 citations
- ProAR: Probabilistic Autoregressive Modeling for Molecular DynamicsKaiwen Cheng, Yutian Liu, Zhiwei Nie, Mujie Lin et al.AAAI 2026
- Exploit Your Latents: Coarse-Grained Protein Backmapping with Latent Diffusion ModelsRongchao Zhang, Yu Huang, Yiwei Lou, Yi Xin et al.AAAI 2025 · 7 citations
- Amortized Sampling with Transferable Normalizing FlowsCharlie B. Tan, Majdi Hassan, Leon Klein, Saifuddin Syed et al.NeurIPS 2025 · 21 citations
- BioMD: All-atom Generative Model for Biomolecular Dynamics SimulationBin Feng, Jiying Zhang, Xinni Zhang, Zijing Liu et al.ICLR 2026 · 10 citations
