A Neural-Network-Based Mapping and Optimization Framework for High-Precision Coarse-Grained Simulation

Abstract: The accuracy and efficiency of a coarse-grained (CG) force field are pivotal for high-precision molecular simulations of large systems with complex molecules. We present an automated mapping and optimization framework for molecular simulation (AMOFMS), which is designed to streamline and improve the force field optimization process. It features a neural-network-based mapping function, DSGPM-TP (Deep Supervised Graph Partitioning Model with Type Prediction). This model can accurately and efficiently convert atomistic structures to CG mappings, reducing the need for manual intervention. By integrating bottom-up and top-down methodologies, AMOFMS allows users to freely combine these approaches or use them independently as optimization targets. Moreover, users can select and combine different optimizers to meet their specific mission. With its parallel optimizer, AMOFMS significantly accelerates the optimization process, reducing the time required to achieve optimal results. Successful applications of AMOFMS include parameter optimizations for systems such as POPC and PEO, demonstrating its robustness and effectiveness. Overall, AMOFMS provides a general and flexible framework for the automated development of high-precision CG force fields.