Flow-matching -- efficient coarse-graining of molecular dynamics without forces

Abstract

Coarse-grained (CG) molecular simulations have become a standard tool to study molecular processes on time- and length-scales inaccessible to all-atom simulations. Parameterizing CG force fields to match all-atom simulations has mainly relied on force-matching or relative entropy minimization, which require many samples from costly simulations with all-atom or CG resolutions, respectively. Here we present flow-matching, a new training method for CG force fields that combines the advantages of both methods by leveraging normalizing flows, a generative deep learning method. Flow-matching first trains a normalizing flow to represent the CG probability density, which is equivalent to minimizing the relative entropy without requiring iterative CG simulations. Subsequently, the flow generates samples and forces according to the learned distribution in order to train the desired CG free energy model via force matching. Even without requiring forces from the all-atom simulations, flow-matching outperforms classical force-matching by an order of magnitude in terms of data efficiency, and produces CG models that can capture the folding and unfolding transitions of small proteins.

References in corpus (3)

• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Machine Learning Implicit Solvation for Molecular Dynamics
• Ensemble Learning of Coarse-Grained Molecular Dynamics Force Fields with a Kernel Approach

Cited by in corpus (9)

• Ensuring thermodynamic consistency with invertible coarse-graining
• chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics
• Spiers Memorial Lecture: How to do impactful research in artificial intelligence for chemistry and materials science
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• Scaling Graph Neural Networks to Large Proteins
• Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Data
• Internal-Coordinate Density Modelling of Protein Structure: Covariance Matters
• Knowledge Distillation of a Protein Language Model Yields a Foundational Implicit Solvent Model
• Knowledge Distillation of Noisy Force Labels for Improved Coarse-Grained Force Fields