Accelerating fourth-generation machine learning potentials by quasi-linear scaling particle mesh charge equilibration

Moritz Gubler,Jonas A. Finkler,Moritz R. Schäfer,Jörg Behler,Stefan Goedecker
2024-03-15
Abstract:Machine learning potentials (MLP) have revolutionized the field of atomistic simulations by describing the atomic interactions with the accuracy of electronic structure methods at a small fraction of the costs. Most current MLPs construct the energy of a system as a sum of atomic energies, which depend on information about the atomic environments provided in form of predefined or learnable feature vectors. If, in addition, non-local phenomena like long-range charge transfer are important, fourth-generation MLPs need to be used, which include a charge equilibration (Qeq) step to take the global structure of the system into account. This Qeq can significantly increase the computational cost and thus can become the computational bottleneck for large systems. In this paper we present a highly efficient formulation of Qeq that does not require the explicit computation of the Coulomb matrix elements resulting in a quasi-linearly scaling method. Moreover, our approach also allows for the efficient calculation of energy derivatives, which explicitly consider the global structure-dependence of the atomic charges as obtained from Qeq. Due to its generality, the method is not restricted to MLPs but can also be applied within a variety of other force fields.
Computational Physics
What problem does this paper attempt to address?
This paper addresses the issue of inefficient calculation of long-range charge transfer and the resulting electrostatic interaction in the fourth-generation machine learning potential (MLP). Existing methods have high computational costs due to the need to solve linear equations, which limits their application to large-scale systems. The paper proposes a new fast conjugate gradient method that achieves approximate linear complexity for charge balance and improves the calculation of energy derivatives. It is suitable for various types of systems, especially 4G-HDNNP that rely on atomic charges. This makes it possible to simulate large-scale systems containing thousands of atoms.