Abstract:The simulation of large-scale systems with complex electron interactions remains one of the greatest challenges for the atomistic modeling of materials. Although classical force fields often fail to describe the coupling between electronic states and ionic rearrangements, the more accurate \textit{ab-initio} molecular dynamics suffers from computational complexity that prevents long-time and large-scale simulations, which are essential to study many technologically relevant phenomena, such as reactions, ion migrations, phase transformations, and degradation. In this work, we present the Crystal Hamiltonian Graph neural Network (CHGNet) as a novel machine-learning interatomic potential (MLIP), using a graph-neural-network-based force field to model a universal potential energy surface. CHGNet is pretrained on the energies, forces, stresses, and magnetic moments from the Materials Project Trajectory Dataset, which consists of over 10 years of density functional theory static and relaxation trajectories of $\sim 1.5$ million inorganic structures. The explicit inclusion of magnetic moments enables CHGNet to learn and accurately represent the orbital occupancy of electrons, enhancing its capability to describe both atomic and electronic degrees of freedom. We demonstrate several applications of CHGNet in solid-state materials, including charge-informed molecular dynamics in Li$_x$MnO$_2$, the finite temperature phase diagram for Li$_x$FePO$_4$ and Li diffusion in garnet conductors. We critically analyze the significance of including charge information for capturing appropriate chemistry, and we provide new insights into ionic systems with additional electronic degrees of freedom that can not be observed by previous MLIPs.

PiNN: equivariant neural network suite for modelling electrochemical systems

PiNN: A Python Library for Building Atomic Neural Networks of Molecules and Materials

PiNNwall: Heterogeneous Electrode Models from Integrating Machine Learning and Atomistic Simulation

SpaiNN: Equivariant Message Passing for Excited-State Nonadiabatic Molecular Dynamics

PiNNAcLe: Adaptive Learn-On-The-Fly Algorithm for Machine-Learning Potential

Neural Network Potential with Multi-Resolution Approach Enables Accurate Prediction of Reaction Free Energies in Solution

An Investigation of Physics Informed Neural Networks to solve the Poisson-Boltzmann Equation in Molecular Electrostatics

AIMNet2: A Neural Network Potential to Meet your Neutral, Charged, Organic, and Elemental-Organic Needs

CHGNet as a pretrained universal neural network potential for charge-informed atomistic modelling

CHGNet: Pretrained universal neural network potential for charge-informed atomistic modeling

Ab initio Accuracy Neural Network Potential for Drug-like Molecules

Simulating Diffusion Properties of Solid‐State Electrolytes via a Neural Network Potential: Performance and Training Scheme

A physics-informed neural network approach to parameter estimation of lithium-ion battery electrochemical model

A metalearning approach for Physics-Informed Neural Networks (PINNs): Application to parameterized PDEs

Including physics-informed atomization constraints in neural networks for reactive chemistry

Physics-Informed Neural Networks (PINNs) for Parameterized PDEs: A Metalearning Approach

The Potential of Neural Network Potentials

AENET-LAMMPS and AENET-TINKER: Interfaces for Accurate and Efficient Molecular Dynamics Simulations with Machine Learning Potentials

Kernel charge equilibration: efficient and accurate prediction of molecular dipole moments with a machine-learning enhanced electron density model

Incorporating Neural Networks into the AMOEBA Polarizable Force Field