Stochastic Optimal Control for Collective Variable Free Sampling of Molecular Transition Paths
Not on the map. Why?
Abstract (arXiv)
We consider the problem of sampling transition paths between two given metastable states of a molecular system, e.g. a folded and unfolded protein or products and reactants of a chemical reaction. Due to the existence of high energy barriers separating the states, these transition paths are unlikely to be sampled with standard Molecular Dynamics (MD) simulation. Traditional methods to augment MD with a bias potential to increase the probability of the transition rely on a dimensionality reduction step based on Collective Variables (CVs). Unfortunately, selecting appropriate CVs requires chemical intuition and traditional methods are therefore not always applicable to larger systems. Additionally, when incorrect CVs are used, the bias potential might not be minimal and bias the system along dimensions irrelevant to the transition. Showing a formal relation between the problem of sampling molecular transition paths, the Schr"odinger bridge problem and stochastic optimal control with neural network policies, we propose a machine learning method for sampling said transitions. Unlike previous non-machine learning approaches our method, named PIPS, does not depend on CVs. We show that our method successful generates low energy transitions for Alanine Dipeptide as well as the larger Polyproline and Chignolin proteins.
Key facts. Cited by 36 papers in the map (data through 7 Oct 2026). Builds on 12 of its 51 references that are in the map. First posted on arXiv 27 Jun 2022.
Builds on
12 of 51 references.
References · 12
- 2016Group Equivariant Convolutional Networkscited by 1,154
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
- 2020Machine Learning Force Fieldscited by 204
- 2021Learning Gradient Fields for Molecular Conformation Generationcited by 162
- 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
- 2014On the relation between optimal transport and Schr"odinger bridges: A stochastic control viewpointcited by 99
- 2015Adaptive importance sampling for control and inferencecited by 23
- 2021Reinforcement learning of rare diffusive dynamicscited by 14
- 2014Path Integral Control and State Dependent Feedbackcited by 11
- 2018Transferable neural networks for enhanced sampling of protein dynamicscited by 9
- 2020Multitask machine learning of collective variables for enhanced sampling of rare eventscited by 9
Built on by
Citing papers per year: 2023: 3; 2024: 11; 2025: 14; 2026: 8.
36 papers.
Sort:
By area:
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Controlcited by 46
- 2023Stochastic Optimal Control Matchingcited by 39
- 2024Doob's Lagrangian: A Sample-Efficient Variational Approach to Transition Path Samplingcited by 25
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 22
- 2025Action-Minimization Meets Generative Modeling: Efficient Transition Path Sampling with the Onsager-Machlup Functionalcited by 18
- 2025Temperature-Annealed Boltzmann Generatorscited by 14
- 2025Value Gradient Guidance for Flow Matching Alignmentcited by 14
- 2024A Taxonomy of Loss Functions for Stochastic Optimal Controlcited by 11
- 2024Transition Path Sampling with Improved Off-Policy Training of Diffusion Path Samplerscited by 10
- 2023Transition Path Sampling with Boltzmann Generator-based MCMC Movescited by 8
24 more
- 2025The Spacetime of Diffusion Models: An Information Geometry Perspectivecited by 6
- 2025Control Consistency Losses for Diffusion Bridgescited by 6
- 2025Reinforced sequential Monte Carlo for amortised samplingcited by 5
- 2024Generalized Flow Matching for Transition Dynamics Modelingcited by 4
- 2024React-OT: Optimal Transport for Generating Transition State in Chemical Reactionscited by 3
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputationcited by 3
- 2025Simulation-Free Differential Dynamics through Neural Conservation Lawscited by 2
- 2024Estimating Reaction Barriers with Deep Reinforcement Learningcited by 1
- 2024An Efficient On-Policy Deep Learning Framework for Stochastic Optimal Controlcited by 1
- 2025Entangled Schr"odinger Bridge Matchingcited by 1
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flowcited by 1
- 2026A unified perspective on fine-tuning and sampling with diffusion and flow modelscited by 1
- 2026Analytic Bridge Diffusions for Controlled Path Generationcited by 1
- 2026A Stabilized Path-Space Approach to Diffusion-Based Posterior Samplingcited by 1
- 2024Control, Transport and Sampling: Towards Better Loss Designcited by 0
- 2025Single-Step Consistent Diffusion Samplerscited by 0
- 2025Follow the MEP: Scalable Neural Representations for Minimum-Energy Path Discovery in Molecular Systemscited by 0
- 2025Machine-Learned Sampling of Conditioned Path Measurescited by 0
- 2025Error Propagation in Dynamic Programming: From Stochastic Control to American Option Pricingcited by 0
- 2026A Schr"odinger Eigenfunction Method for Long-Horizon Stochastic Optimal Controlcited by 0
- 2026Path Integral Value Matching for Linear Quadratic Stochastic Optimal Controlcited by 0
- 2026Stochastic Control Policies for Robust Molecular Transition Path Samplingcited by 0
- 2026Machine learning kinetics from molecular dynamics datacited by 0
- 2026Neural Transport Nested Samplingcited by 0
- 2026Neural Transport Nested Sampling24 Sep 2026
- 2026Machine learning kinetics from molecular dynamics data15 Sep 2026
- 2026Stochastic Control Policies for Robust Molecular Transition Path Sampling13 Aug 2026
- 2026Path Integral Value Matching for Linear Quadratic Stochastic Optimal Control11 Aug 2026
- 2026A Stabilized Path-Space Approach to Diffusion-Based Posterior Sampling10 Jun 2026
- 2026Analytic Bridge Diffusions for Controlled Path Generation3 May 2026
- 2026A unified perspective on fine-tuning and sampling with diffusion and flow models30 Apr 2026
- 2026A Schr"odinger Eigenfunction Method for Long-Horizon Stochastic Optimal Control24 Mar 2026
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flow4 Dec 2025
- 2025Value Gradient Guidance for Flow Matching Alignment4 Dec 2025
- 2025Control Consistency Losses for Diffusion Bridges4 Dec 2025
- 2025Entangled Schr"odinger Bridge Matching10 Nov 2025
24 more
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
- 2025Error Propagation in Dynamic Programming: From Stochastic Control to American Option Pricing24 Sep 2025
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
- 2025Simulation-Free Differential Dynamics through Neural Conservation Laws23 Jun 2025
- 2025Machine-Learned Sampling of Conditioned Path Measures2 Jun 2025
- 2025The Spacetime of Diffusion Models: An Information Geometry Perspective23 May 2025
- 2025Action-Minimization Meets Generative Modeling: Efficient Transition Path Sampling with the Onsager-Machlup Functional25 Apr 2025
- 2025Follow the MEP: Scalable Neural Representations for Minimum-Energy Path Discovery in Molecular Systems23 Apr 2025
- 2025Single-Step Consistent Diffusion Samplers11 Feb 2025
- 2025Temperature-Annealed Boltzmann Generators31 Jan 2025
- 2024Generalized Flow Matching for Transition Dynamics Modeling19 Oct 2024
- 2024Doob's Lagrangian: A Sample-Efficient Variational Approach to Transition Path Sampling10 Oct 2024
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputation10 Oct 2024
- 2024An Efficient On-Policy Deep Learning Framework for Stochastic Optimal Control7 Oct 2024
- 2024A Taxonomy of Loss Functions for Stochastic Optimal Control1 Oct 2024
- 2024Estimating Reaction Barriers with Deep Reinforcement Learning17 Jul 2024
- 2024Transition Path Sampling with Improved Off-Policy Training of Diffusion Path Samplers30 May 2024
- 2024RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control27 May 2024
- 2024Control, Transport and Sampling: Towards Better Loss Design22 May 2024
- 2024React-OT: Optimal Transport for Generating Transition State in Chemical Reactions20 Apr 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2023Transition Path Sampling with Boltzmann Generator-based MCMC Moves8 Dec 2023
- 2023Stochastic Optimal Control Matching4 Dec 2023
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimization4 Oct 2023
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Controlcited by 46
- 2025Action-Minimization Meets Generative Modeling: Efficient Transition Path Sampling with the Onsager-Machlup Functionalcited by 18
- 2025Value Gradient Guidance for Flow Matching Alignmentcited by 14
- 2025The Spacetime of Diffusion Models: An Information Geometry Perspectivecited by 6
- 2024Generalized Flow Matching for Transition Dynamics Modelingcited by 4
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flowcited by 1
- 2026A Stabilized Path-Space Approach to Diffusion-Based Posterior Samplingcited by 1
- 2025Single-Step Consistent Diffusion Samplerscited by 0
- 2026A Stabilized Path-Space Approach to Diffusion-Based Posterior Sampling10 Jun 2026
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flow4 Dec 2025
- 2025Value Gradient Guidance for Flow Matching Alignment4 Dec 2025
- 2025The Spacetime of Diffusion Models: An Information Geometry Perspective23 May 2025
- 2025Action-Minimization Meets Generative Modeling: Efficient Transition Path Sampling with the Onsager-Machlup Functional25 Apr 2025
- 2025Single-Step Consistent Diffusion Samplers11 Feb 2025
- 2024Generalized Flow Matching for Transition Dynamics Modeling19 Oct 2024
- 2024RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control27 May 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2023Stochastic Optimal Control Matchingcited by 39
- 2024Doob's Lagrangian: A Sample-Efficient Variational Approach to Transition Path Samplingcited by 25
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 22
- 2025Temperature-Annealed Boltzmann Generatorscited by 14
- 2024A Taxonomy of Loss Functions for Stochastic Optimal Controlcited by 11
- 2024Transition Path Sampling with Improved Off-Policy Training of Diffusion Path Samplerscited by 10
- 2023Transition Path Sampling with Boltzmann Generator-based MCMC Movescited by 8
- 2025Control Consistency Losses for Diffusion Bridgescited by 6
- 2025Reinforced sequential Monte Carlo for amortised samplingcited by 5
- 2024React-OT: Optimal Transport for Generating Transition State in Chemical Reactionscited by 3
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputationcited by 3
12 most cited of 27 All 27 (JSON)
- 2026Neural Transport Nested Sampling24 Sep 2026
- 2026Machine learning kinetics from molecular dynamics data15 Sep 2026
- 2026Stochastic Control Policies for Robust Molecular Transition Path Sampling13 Aug 2026
- 2026Path Integral Value Matching for Linear Quadratic Stochastic Optimal Control11 Aug 2026
- 2026Analytic Bridge Diffusions for Controlled Path Generation3 May 2026
- 2026A unified perspective on fine-tuning and sampling with diffusion and flow models30 Apr 2026
- 2026A Schr"odinger Eigenfunction Method for Long-Horizon Stochastic Optimal Control24 Mar 2026
- 2025Control Consistency Losses for Diffusion Bridges4 Dec 2025
- 2025Entangled Schr"odinger Bridge Matching10 Nov 2025
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
- 2025Error Propagation in Dynamic Programming: From Stochastic Control to American Option Pricing24 Sep 2025
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
12 newest of 27 All 27 (JSON)