Coarse Graining Molecular Dynamics with Graph Neural Networks

Abstract

Coarse graining enables the investigation of molecular dynamics for larger systems and at longer timescales than is possible at atomic resolution. However, a coarse graining model must be formulated such that the conclusions we draw from it are consistent with the conclusions we would draw from a model at a finer level of detail. It has been proven that a force matching scheme defines a thermodynamically consistent coarse-grained model for an atomistic system in the variational limit. Wang et al. [ACS Cent. Sci. 5, 755 (2019)] demonstrated that the existence of such a variational limit enables the use of a supervised machine learning framework to generate a coarse-grained force field, which can then be used for simulation in the coarse-grained space. Their framework, however, requires the manual input of molecular features upon which to machine learn the force field. In the present contribution, we build upon the advance of Wang et al.and introduce a hybrid architecture for the machine learning of coarse-grained force fields that learns their own features via a subnetwork that leverages continuous filter convolutions on a graph neural network architecture. We demonstrate that this framework succeeds at reproducing the thermodynamics for small biomolecular systems. Since the learned molecular representations are inherently transferable, the architecture presented here sets the stage for the development of machine-learned, coarse-grained force fields that are transferable across molecular systems.

17 pages, 9 figures

References in corpus (34)

• Array Programming with NumPy
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Semi-Supervised Classification with Graph Convolutional Networks
• Neural Message Passing for Quantum Chemistry
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• 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
• Molecular Graph Convolutions: Moving Beyond Fingerprints
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Interatomic potentials from first-principles calculations: the force-matching method
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Identification of slow molecular order parameters for Markov model construction
• Machine learning for molecular simulation
• Convolutional Networks on Graphs for Learning Molecular Fingerprints
• Less is more: sampling chemical space with active learning
• Machine Learning Unifies the Modelling of Materials and Molecules
• SchNet: A continuous-filter convolutional neural network for modeling quantum interactions
• SchNetPack: A Deep Learning Toolbox For Atomistic Systems
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• DeePCG: constructing coarse-grained models via deep neural networks
• Rational Construction of Stochastic Numerical Methods for Molecular Sampling
• 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
• A First Principle Approach to Rescale the Dynamics of Simulated Coarse-Grained Macromolecular Liquids
• Variational Selection of Features for Molecular Kinetics
• Temperature-transferable coarse-graining of ionic liquids with dual graph convolutional neural networks
• Ensemble Learning of Coarse-Grained Molecular Dynamics Force Fields with a Kernel Approach
• Coarse-graining Molecular Systems by Spectral Matching

Cited by in corpus (25)

• TorchMD: A deep learning framework for molecular simulations
• Machine Learning Coarse-Grained Potentials of Protein Thermodynamics
• Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting
• Machine Learning Implicit Solvation for Molecular Dynamics
• Flow-matching -- efficient coarse-graining of molecular dynamics without forces
• Graph Neural Networks Accelerated Molecular Dynamics
• GraphVAMPNet, using graph neural networks and variational approach to markov processes for dynamical modeling of biomolecules
• Deep Coarse-grained Potentials via Relative Entropy Minimization
• Quantum dynamics using path integral coarse-graining
• Contrastive Learning of Coarse-Grained Force Fields
• Ensuring thermodynamic consistency with invertible coarse-graining
• Temporally coherent backmapping of molecular trajectories from coarse-grained to atomistic resolution
• Recent advances in describing and driving crystal nucleation using machine learning and artificial intelligence
• Spiers Memorial Lecture: How to do impactful research in artificial intelligence for chemistry and materials science
• Automatic Multi-Objective Optimization of Coarse-Grained Lipid Force Fields Using SwarmCG
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• EPR-Net: Constructing non-equilibrium potential landscape via a variational force projection formulation
• AMARO: All Heavy-Atom Transferable Neural Network Potentials of Protein Thermodynamics
• Data-driven dynamical coarse-graining for condensed matter systems
• A neural network potential with self-trained atomic fingerprints: a test with the mW water potential
• Scaling Graph Neural Networks to Large Proteins
• Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Data
• Martini Mapper: An Automated Fragment-Based Framework for Developing Coarse-Grained Models within the Martini 3 Framework
• Knowledge Distillation of Noisy Force Labels for Improved Coarse-Grained Force Fields
• Knowledge Distillation of a Protein Language Model Yields a Foundational Implicit Solvent Model