High-Throughput Aqueous Electrolyte Structure Prediction Using IonSolvR and Equivariant Graph Neural Network Potentials.

Abstract

Neural network potentials have recently emerged as an efficient and accurate tool for accelerating ab initio molecular dynamics (AIMD) in order to simulate complex condensed phases such as electrolyte solutions. Their principal limitation, however, is their requirement for sufficiently large and accurate training sets, which are often composed of Kohn-Sham density functional theory (DFT) calculations. Here we examine the feasibility of using existing density functional tight-binding (DFTB) molecular dynamics trajectory data available in the IonSolvR database in order to accelerate the training of E(3)-equivariant graph neural network potentials. We show that the solvation structure of Na+ and Cl- in aqueous NaCl solutions can be accurately reproduced with remarkably small amounts of data (i.e., 100 MD frames). We further show that these predictions can be systematically improved further via an embarrassingly parallel resampling approach.