Temperature-transferable coarse-graining of ionic liquids with dual graph convolutional neural networks
Jurğis Ruža, Wujie Wang, Daniel Schwalbe‐Koda, Simon Axelrod, William H. Harris, Rafael Gómez‐Bombarelli
Abstract
Computer simulations can provide mechanistic insight into ionic liquids (ILs) and predict the properties of experimentally unrealized ion combinations. However, ILs suffer from a particularly large disparity in the time scales of atomistic and ensemble motion. Coarse-grained models are therefore used in place of costly all-atom simulations, accessing longer time scales and larger systems. Nevertheless, constructing the many-body potential of mean force that defines the structure and dynamics of a coarse-grained system can be complicated and computationally intensive. Machine learning shows great promise for the linked challenges of dimensionality reduction and learning the potential of mean force. To improve the coarse-graining of ILs, we present a neural network model trained on all-atom classical molecular dynamics simulations. The potential of mean force is expressed as two jointly trained neural network interatomic potentials that learn the coupled short-range and many-body long range molecular interactions. These interatomic potentials treat temperature as an explicit input variable to capture its influence on the potential of mean force. The model reproduces structural quantities with high fidelity, outperforms the temperature-independent baseline at capturing dynamics, generalizes to unseen temperatures, and incurs low simulation cost.