Ensuring thermodynamic consistency with invertible coarse-graining
arXiv:2210.07882 · doi:10.1063/5.0141888
Abstract
Coarse-grained models are a core computational tool in theoretical chemistry and biophysics. A judicious choice of a coarse-grained model can yield physical insight by isolating the essential degrees of freedom that dictate the thermodynamic properties of a complex, condensed-phase system. The reduced complexity of the model typically leads to lower computational costs and more efficient sampling compared to atomistic models. Designing ``good'' coarse-grained models is an art. Generally, the mapping from fine-grained configurations to coarse-grained configurations itself is not optimized in any way; instead, the energy function associated with the mapped configurations is. In this work, we explore the consequences of optimizing the coarse-grained representation alongside its potential energy function. We use a graph machine learning framework to embed atomic configurations into a low dimensional space to produce efficient representations of the original molecular system. Because the representation we obtain is no longer directly interpretable as a real space representation of the atomic coordinates, we also introduce an inversion process and an associated thermodynamic consistency relation that allows us to rigorously sample fine-grained configurations conditioned on the coarse-grained sampling. We show that this technique is robust, recovering the first two moments of the distribution of several observables in proteins such as chignolin and alanine dipeptide.
References in corpus (19)
- Automatic chemical design using a data-driven continuous representation of molecules
- SchNet - a deep learning architecture for molecules and materials
- Identification of slow molecular order parameters for Markov model construction
- VAMPnets: Deep learning of molecular kinetics
- Normalizing Flows for Probabilistic Modeling and Inference
- Breaking the Curse of Dimensionality with Convex Neural Networks
- Flow-based generative models for Markov chain Monte Carlo in lattice field theory
- Coarse Graining Molecular Dynamics with Graph Neural Networks
- Coarse-Graining Auto-Encoders for Molecular Dynamics
- E(n) Equivariant Graph Neural Networks
- Flow-matching -- efficient coarse-graining of molecular dynamics without forces
- Time step rescaling recovers continuous-time dynamical properties for discrete-time Langevin integration of nonequilibrium systems
- Trainability and Accuracy of Neural Networks: An Interacting Particle System Approach
- From data to noise to data: mixing physics across temperatures with generative artificial intelligence
- Adversarial-Residual-Coarse-Graining: Applying machine learning theory to systematic molecular coarse-graining
- Encoding and Selecting Coarse-Grain Mapping Operators with Hierarchical Graphs
- Predictive Collective Variable Discovery with Deep Bayesian Models
- Generative Coarse-Graining of Molecular Conformations
- Efficient Bayesian Sampling Using Normalizing Flows to Assist Markov Chain Monte Carlo Methods