ART-SM: Boosting Fragment-based Backmapping by Machine Learning

In sequential multiscale molecular dynamics simulations, which advantageously combine the increased sampling and dynamics at coarse-grained resolution with the higher accuracy of atomistic simulations, the resolution is altered over time. While coarse-graining is straightforward once the mapping between atomistic and coarse-grained resolution is defined, reintroducing the atomistic details is still a non-trivial process called backmapping. Here, we present ART-SM, a fragment-based backmapping framework that learns from atomistic simulation data to seamlessly switch from coarse-grained to atomistic resolution. ART-SM requires minimal user input and goes beyond state-of-the-art fragment-based approaches by selecting from multiple conformations per fragment via machine learning to simultaneously reflect the coarse-grained structure and the Boltzmann distribution. Additionally, we introduce a novel refinement step to connect individual fragments by optimizing specific bonds, angles, and dihedral angles in the backmapping process. We demonstrate that our algorithm accurately restores the atomistic bond length, angle, and dihedral angle distributions for various small and linear molecules from Martini coarse-grained beads and that the resulting high-resolution structures are representative of the input coarse-grained conformations. Moreover, the reconstruction of the TIP3P water model is fast and robust, and we demonstrate that ART-SM can be applied to larger linear molecules as well. To illustrate the efficiency of the local and autoregressive approach of ART-SM, we simulated a large realistic system containing the surfactants TAPB and SDS in solution using the Martini3 force field. The self-assembled micelles of various shapes were backmapped with ART-SM after training on only short atomistic simulations of a single water-solvated SDS or TAPB molecule. Together, these results indicate the potential for the method to be extended to more complex molecules such as lipids, proteins, macromolecules, and materials in the future.