Graph Neural Network Based Coarse-Grained Mapping Prediction
This work addresses the need for automated mapping selection in molecular dynamics, a domain-specific problem, but it is incremental as it builds on existing supervised learning and graph segmentation techniques.
The authors tackled the problem of selecting coarse-grained mapping operators for molecular dynamics simulation by developing an automated graph neural network approach, DSGPM, which outperforms state-of-the-art methods in graph segmentation and produces good simulation models.
The selection of coarse-grained (CG) mapping operators is a critical step for CG molecular dynamics (MD) simulation. It is still an open question about what is optimal for this choice and there is a need for theory. The current state-of-the art method is mapping operators manually selected by experts. In this work, we demonstrate an automated approach by viewing this problem as supervised learning where we seek to reproduce the mapping operators produced by experts. We present a graph neural network based CG mapping predictor called DEEP SUPERVISED GRAPH PARTITIONING MODEL(DSGPM) that treats mapping operators as a graph segmentation problem. DSGPM is trained on a novel dataset, Human-annotated Mappings (HAM), consisting of 1,206 molecules with expert annotated mapping operators. HAM can be used to facilitate further research in this area. Our model uses a novel metric learning objective to produce high-quality atomic features that are used in spectral clustering. The results show that the DSGPM outperforms state-of-the-art methods in the field of graph segmentation. Finally, we find that predicted CG mapping operators indeed result in good CG MD models when used in simulation.