Beyond Ensembles: Simulating All-Atom Protein Dynamics in a Learned Latent Space
By: Aditya Sengar , Jiying Zhang , Pierre Vandergheynst and more
Potential Business Impact:
Models how tiny body parts move over time.
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. 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 functional free-energy landscapes. Autoregressive neural networks deliver the most robust long rollouts; 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.
Similar Papers
Thermodynamically Consistent Latent Dynamics Identification for Parametric Systems
Machine Learning (CS)
Makes computer models run much faster and smarter.
Exploring Molecule Generation Using Latent Space Graph Diffusion
Machine Learning (CS)
Creates new medicines by designing molecules.
Learning Stochastic Nonlinear Dynamics with Embedded Latent Transfer Operators
Machine Learning (CS)
Helps computers understand how things change over time.