Molecular dynamics simulations of anisotropic particles accelerated by neural-net predicted interactions

Rigid bodies, made of smaller composite beads, are commonly used to simulate anisotropic particles with molecular dynamics or Monte Carlo methods. To accurately represent the particle shape and to obtain smooth and realistic effective pair interactions between two rigid bodies, each body may need to contain hundreds of spherical beads. Given an interacting pair of particles, traditional molecular dynamics methods calculate all the inter-body distances between the beads of the rigid bodies within a certain distance. For a system containing many anisotropic particles, these distance calculations are computationally costly and limit the attainable system size and simulation time. However, the effective interaction between two rigid particles should only depend on the distance between their center of masses and their relative orientation. Therefore, a function capable of directly mapping the center of mass distance and orientation to the interaction energy between the two rigid bodies would completely bypass inter-bead distance calculations. It is challenging to derive such a general function analytically for almost any non-spherical rigid body. In this study, we have trained neural nets, powerful tools to fit nonlinear functions to complex datasets, to achieve this task. The pair configuration (center of mass distance and relative orientation) is taken as an input, and the energy, forces, and torques between two rigid particles are predicted directly. We show that molecular dynamics simulations of cubes and cylinders performed with forces and torques obtained from the gradients of the energy neural-nets quantitatively match traditional simulations that use composite rigid bodies. Both structural quantities and dynamic measures are in agreement, while achieving up to 23 times speedup over traditional molecular dynamics, depending on hardware and system size. The method presented here can, in principle, be applied to any irregular concave or convex shape with any pair interaction, provided that sufficient training data can be obtained.