Coarse-Graining Auto-Encoders for Molecular Dynamics

Abstract

Molecular dynamics simulations provide theoretical insight into the microscopic behavior of materials in condensed phase and, as a predictive tool, enable computational design of new compounds. However, because of the large temporal and spatial scales involved in thermodynamic and kinetic phenomena in materials, atomistic simulations are often computationally unfeasible. Coarse-graining methods allow simulating larger systems, by reducing the dimensionality of the simulation, and propagating longer timesteps, by averaging out fast motions. Coarse-graining involves two coupled learning problems; defining the mapping from an all-atom to a reduced representation, and the parametrization of a Hamiltonian over coarse-grained coordinates. Multiple statistical mechanics approaches have addressed the latter, but the former is generally a hand-tuned process based on chemical intuition. Here we present Autograin, an optimization framework based on auto-encoders to learn both tasks simultaneously. Autograin is trained to learn the optimal mapping between all-atom and reduced representation, using the reconstruction loss to facilitate the learning of coarse-grained variables. In addition, a force-matching method is applied to variationally determine the coarse-grained potential energy function. This procedure is tested on a number of model systems including single-molecule and bulk-phase periodic simulations.

arXiv comment8 pages, 6 figures

References in corpus (9)

• Automatic chemical design using a data-driven continuous representation of molecules
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• VAMPnets: Deep learning of molecular kinetics
• Time-lagged autoencoders: Deep learning of slow collective variables for molecular kinetics
• DeePCG: constructing coarse-grained models via deep neural networks
• Resolving Dynamic Properties of Polymers through Coarse-Grained Computational Studies
• Hydrodynamics of Diffusion in Lipid Membrane Simulations
• Encoding and Selecting Coarse-Grain Mapping Operators with Hierarchical Graphs
• Predictive Coarse-Graining

Cited by in corpus (24)

• Machine learning for molecular simulation
• Molecular Contrastive Learning of Representations via Graph Neural Networks
• Autonomous discovery in the chemical sciences part I: Progress
• TorchMD: A deep learning framework for molecular simulations
• Coarse Graining Molecular Dynamics with Graph Neural Networks
• 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
• Ensemble Learning of Coarse-Grained Molecular Dynamics Force Fields with a Kernel Approach
• PANNA: Properties from Artificial Neural Network Architectures
• Graph Neural Network Based Coarse-Grained Mapping Prediction
• Ensuring thermodynamic consistency with invertible coarse-graining
• On the design space between molecular mechanics and machine learning force fields
• Temporally coherent backmapping of molecular trajectories from coarse-grained to atomistic resolution
• A mapping space Odyssey: characterising the statistical and metric properties of reduced representations of macromolecules
• chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• A Unified Theory of Free Energy Functionals and Applications to Diffusion
• Compressing and forecasting atomic material simulations with descriptors
• Constraint Decoupled Latent Diffusion for Protein Backmapping
• Energy-Based Coarse-Graining in Molecular Dynamics: A Flow-Based Framework without Data
• Snap, Crackle, and Pop: This is why the potential of mean force clashes with the fluctuation dissipation relation
• Martini Mapper: An Automated Fragment-Based Framework for Developing Coarse-Grained Models within the Martini 3 Framework
• Adversarial reverse mapping of condensed-phase molecular structures: Chemical transferability