DiffGLE: Differentiable Coarse-Grained Dynamics using Generalized Langevin Equation

Jinu Jeong,Ishan Nadkarni,Narayana. R. Aluru
2024-10-11
Abstract:Capturing the correct dynamics at the Coarse-Grained (CG) scale remains a central challenge in the advancement of systematic CG models for soft matter simulations. The Generalized Langevin Equation (GLE), rooted in the Mori-Zwanzig formalism, provides a robust framework for incorporating friction and stochastic forces into CG models, that are lost due to the reduction in degrees of freedom. Leveraging recent advancements in Automatic Differentiation (AD) and reformulating the non-Markovian GLE using a colored noise ansatz, we present a top-down approach for accurately parameterizing the non-Markovian GLE for different coarse-grained fluids that accurately reproduces the velocity-autocorrelation function of the original All-Atom (AA) model. We demonstrate our approach on two different fluids namely, SPC/E water and carbon dioxide which have distinct structure and dynamical characteristics. Importantly, by being end-to-end differentiable, this approach offers a simplified and efficient solution to achieving accurate CG dynamics, effectively bypassing the complexities inherent in most bottom-up methods.
Soft Condensed Matter
What problem does this paper attempt to address?
The paper attempts to address the issue of accurately capturing system dynamics in Coarse-Grained (CG) models. Specifically, the authors focus on how to construct dynamically accurate CG models through the Generalized Langevin Equation (GLE) to overcome the problem of inaccurate dynamics predictions in existing CG models when simulating soft matter. This inaccuracy is mainly reflected in the fact that, compared to All-Atom (AA) models, the dynamics processes in CG models are too fast, making it impossible to accurately study the dynamic effects and transport properties of the system on relevant time scales. To achieve this goal, the authors propose an end-to-end differentiable method based on Automatic Differentiation (AD) technology for accurately parameterizing non-Markovian GLE, thereby accurately reproducing the Velocity-Autocorrelation Function (VACF) of the original AA model. This method not only simplifies the parameterization process but also improves computational efficiency, avoiding the complexity of traditional bottom-up approaches. In the paper, the authors demonstrate the effectiveness of this method in two different fluids—SPC/E water and carbon dioxide—proving its effectiveness in maintaining dynamic accuracy and structural properties.