# Learning Implicit Bias in Generative Spaces for Accelerating Protein Dynamics Emulation

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

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 22 of its 42 references that are in the map.
- First posted on arXiv 1 Jun 2026.

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

- [Denoising Diffusion Probabilistic Models](https://mlsanity.com/paper/2006.11239/) (2020)
- [Denoising Diffusion Implicit Models](https://mlsanity.com/paper/2010.02502/) (2020)
- [Score-Based Generative Modeling through Stochastic Differential Equations](https://mlsanity.com/paper/2011.13456/) (2020)
- [Diffusion Models Beat GANs on Image Synthesis](https://mlsanity.com/paper/2105.05233/) (2021)
- [Classifier-Free Diffusion Guidance](https://mlsanity.com/paper/2207.12598/) (2022)
- [Generative Modeling by Estimating Gradients of the Data Distribution](https://mlsanity.com/paper/1907.05600/) (2019)
- [SE(3) diffusion model with application to protein backbone generation](https://mlsanity.com/paper/2302.02277/) (2023)
- [SE(3)-Stochastic Flow Matching for Protein Backbone Generation](https://mlsanity.com/paper/2310.02391/) (2023)
- [AlphaFold Meets Flow Matching for Generating Protein Ensembles](https://mlsanity.com/paper/2402.04845/) (2024)
- [Two for One: Diffusion Models and Force Fields for Coarse-Grained Molecular Dynamics](https://mlsanity.com/paper/2302.00600/) (2023)
- [Generative Modeling of Molecular Dynamics Trajectories](https://mlsanity.com/paper/2409.17808/) (2024)
- [Protein Conformation Generation via Force-Guided SE(3) Diffusion Models](https://mlsanity.com/paper/2403.14088/) (2024)
- [Str2Str: A Score-based Framework for Zero-shot Protein Conformation Sampling](https://mlsanity.com/paper/2306.03117/) (2023)
- [DiffMD: A Geometric Diffusion Model for Molecular Dynamics Simulations](https://mlsanity.com/paper/2204.08672/) (2022)
- [Consistent Sampling and Simulation: Molecular Dynamics with Energy-Based Diffusion Models](https://mlsanity.com/paper/2506.17139/) (2025)
- [Enhanced Diffusion Sampling: Efficient Rare Event Sampling and Free Energy Calculation with Diffusion Models](https://mlsanity.com/paper/2602.16634/) (2026)
- [Dynamic PDB: A New Dataset and a SE(3) Model Extension by Integrating Dynamic Behaviors and Physical Properties in Protein Structures](https://mlsanity.com/paper/2408.12413/) (2024)
- [Enhancing Diffusion-Based Sampling with Molecular Collective Variables](https://mlsanity.com/paper/2510.11923/) (2025)
- [ConfRover: Simultaneous Modeling of Protein Conformation and Dynamics via Autoregression](https://mlsanity.com/paper/2505.17478/) (2025)
- [Unlocking hidden biomolecular conformational landscapes in diffusion models at inference time](https://mlsanity.com/paper/2512.03312/) (2025)
- [Scalable Spatio-Temporal SE(3) Diffusion for Long-Horizon Protein Dynamics](https://mlsanity.com/paper/2602.02128/) (2026)
- [AlphaFolding: 4D Diffusion for Dynamic Protein Structure Prediction with Reference and Motion Guidance](https://mlsanity.com/paper/2408.12419/) (2024)

## Abstract (arXiv)

Generative emulators of protein dynamics produce plausible trajectories at a fraction of the cost of molecular dynamics, but they inherit their training distribution and tend to revisit known states rather than reach rare ones under long-horizon extrapolation. Inspired by classical enhanced sampling, we introduce an implicit, history-dependent bias in the generative space of a pretrained emulator. Specifically, a history-aware score estimator augments the frozen emulator with a distance-weighted bias that steers reverse-time sampling away from previously generated structures, regularized by an environment-support term. To preserve structural validity at long horizons, a score-based refinement step re-projects drifted samples onto the data manifold using the frozen emulator. Our experiments demonstrate that the method (i) raises diversity by 35% on DynamicPDB-80; (ii) on 12 zero-shot Fast-Folding proteins, the learned bias alone reaches the unbiased emulator's coverage up to ~15× faster, and pairing it with refinement reaches the coverage up to ~37× faster while covering ~3× as many low-energy states. Code will be released soon.