Charge-Optimized Electrostatic Interaction Atom-Centered Neural Network Algorithm

Machine-learning algorithms have been proposed to capture electrostatic interactions by using effective partial charges. These algorithms often rely on a pretrained model for partial charge prediction using density functional theory-calculated partial charges as references, which introduces complexity to the force field model. The accuracy of the trained model also depends on the reliability of charge partition methods, which can be dependent on the specific system and methodology employed. In this study, we propose an atom-centered neural network (ANN) algorithm that eliminates the need for reference charges. Our algorithm requires only a single NN model for each element to obtain both atomic energy and charges. These atomic charges are then employed to compute electrostatic energies using the Ewald summation algorithm. Subsequently, the force field model is trained on total energy and forces, with the inclusion of electrostatic energy. To evaluate the performance of our algorithm, we conducted tests on three benchmark systems, including a Ge slab with an O adatom system, a TiO₂ crystalline system, and a Pd-O nanoparticle system. Our results demonstrate reasonably accurate predictions of partial charges and electrostatic interactions. This algorithm provides a self-consistent charge prediction strategy and possibilities for robust and reliable modeling of electrostatic interactions in machine-learning potentials.