Ensemble Learning of Coarse-Grained Molecular Dynamics Force Fields with a Kernel Approach

Abstract

Gradient-domain machine learning (GDML) is an accurate and efficient approach to learn a molecular potential and associated force field based on the kernel ridge regression algorithm. Here, we demonstrate its application to learn an effective coarse-grained (CG) model from all-atom simulation data in a sample efficient manner. The coarse-grained force field is learned by following the thermodynamic consistency principle, here by minimizing the error between the predicted coarse-grained force and the all-atom mean force in the coarse-grained coordinates. Solving this problem by GDML directly is impossible because coarse-graining requires averaging over many training data points, resulting in impractical memory requirements for storing the kernel matrices. In this work, we propose a data-efficient and memory-saving alternative. Using ensemble learning and stratified sampling, we propose a 2-layer training scheme that enables GDML to learn an effective coarse-grained model. We illustrate our method on a simple biomolecular system, alanine dipeptide, by reconstructing the free energy landscape of a coarse-grained variant of this molecule. Our novel GDML training scheme yields a smaller free energy error than neural networks when the training set is small, and a comparably high accuracy when the training set is sufficiently large.

14 pages, 6 figures

References in corpus (25)

• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Popular Ensemble Methods: An Empirical Study
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• Deep Potential Molecular Dynamics: a scalable model with the accuracy of quantum mechanics
• SchNet - a deep learning architecture for molecules and materials
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Identification of slow molecular order parameters for Markov model construction
• Machine learning for molecular simulation
• Less is more: sampling chemical space with active learning
• Machine Learning Unifies the Modelling of Materials and Molecules
• Towards Exact Molecular Dynamics Simulations with Machine-Learned Force Fields
• VAMPnets: Deep learning of molecular kinetics
• Time-lagged autoencoders: Deep learning of slow collective variables for molecular kinetics
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• sGDML: Constructing Accurate and Data Efficient Molecular Force Fields Using Machine Learning
• DeePCG: constructing coarse-grained models via deep neural networks
• Variational Encoding of Complex Dynamics
• Coarse-Graining Auto-Encoders for Molecular Dynamics
• Non-covalent interactions across organic and biological subsets of chemical space: Physics-based potentials parametrized from machine learning
• Comparison of permutationally invariant polynomials, neural networks, and Gaussian approximation potentials in representing water interactions through many-body expansions
• End-to-end Symmetry Preserving Inter-atomic Potential Energy Model for Finite and Extended Systems
• Coarse-graining Molecular Systems by Spectral Matching

Cited by in corpus (16)

• Machine Learning Force Fields
• Combining Machine Learning and Computational Chemistry for Predictive Insights Into Chemical Systems
• TorchMD: A deep learning framework for molecular simulations
• Perspective on integrating machine learning into computational chemistry and materials science
• Coarse Graining Molecular Dynamics with Graph Neural Networks
• Machine Learning Coarse-Grained Potentials of Protein Thermodynamics
• Machine Learning Implicit Solvation for Molecular Dynamics
• Flow-matching -- efficient coarse-graining of molecular dynamics without forces
• Temperature-transferable coarse-graining of ionic liquids with dual graph convolutional neural networks
• Machine Learning of Free Energies in Chemical Compound Space Using Ensemble Representations: Reaching Experimental Uncertainty for Solvation
• Molecular Force Fields with Gradient-Domain Machine Learning (GDML): Comparison and Synergies with Classical Force Fields
• Dynamical Strengthening of Covalent and Non-Covalent Molecular Interactions by Nuclear Quantum Effects at Finite Temperature
• Relevance of Rotationally Equivariant Convolutions for Predicting Molecular Properties
• Quantum dynamics using path integral coarse-graining
• Efficient interpolation of molecular properties across chemical compound space with low-dimensional descriptors
• Knowledge Distillation of Noisy Force Labels for Improved Coarse-Grained Force Fields