Abstract:Having access to accurate electron densities in chemical systems, especially for dynamical systems involving chemical reactions, ion transport, and other charge transfer processes, is crucial for numerous applications in materials chemistry. Traditional methods for computationally predicting electron density data for such systems include quantum mechanical (QM) techniques, such as density functional theory. However, poor scaling of these QM methods restricts their use to relatively small system sizes and short dynamic time scales. To overcome this limitation, we have developed a deep neural network machine learning formalism, which we call deep charge density prediction (DeepCDP), for predicting charge densities by only using atomic positions for molecules and condensed phase (periodic) systems. Our method uses the weighted smooth overlap of atomic positions to fingerprint environments on a grid-point basis and map it to electron density data generated from QM simulations. We trained models for bulk systems of copper, LiF, and silicon; for a molecular system, water; and for two-dimensional charged and uncharged systems, hydroxyl-functionalized graphane, with and without an added proton. We showed that DeepCDP achieves prediction R2 values greater than 0.99 and mean squared error values on the order of 10−5e2 Å−6 for most systems. DeepCDP scales linearly with system size, is highly parallelizable, and is capable of accurately predicting the excess charge in protonated hydroxyl-functionalized graphane. We demonstrate how DeepCDP can be used to accurately track the location of charges (protons) by computing electron densities at a few selected grid points in the materials, thus significantly reducing the computational cost. We also show that our models can be transferable, allowing prediction of electron densities for systems on which it has not been trained but that contain a subset of atomic species on which it has been trained. Our approach can be used to develop models that span different chemical systems and train them for the study of large-scale charge transport and chemical reactions.

Gaussian Plane-Wave Neural Operator for Electron Density Estimation

Training Machine-Learned Density Functionals on Band Gaps

Highly Accurate Real-space Electron Densities with Neural Networks

Gaussian and Plane-Wave Mixed Density Fitting for Periodic Systems.

From plane waves to local Gaussians for the simulation of correlated periodic systems

Learning Equivariant Non-Local Electron Density Functionals

Equivariant graph neural networks for fast electron density estimation of molecules, liquids, and solids

Efficient Sampling for Machine Learning Electron Density and Its Response in Real Space

Time-Dependent Density Functional Theory Excited-State Forces with the Gaussian and Augmented Plane Wave Method

Generalizing deep learning electronic structure calculation to the plane-wave basis

Learning electron densities in the condensed phase

Towards Gaussian Process for operator learning: an uncertainty aware resolution independent operator learning algorithm for computational mechanics

Incorporating Electronic Information into Machine Learning Potential Energy Surfaces via Approaching the Ground-State Electronic Energy as a Function of Atom-Based Electronic Populations

Dynamic Gaussian Graph Operator: Learning parametric partial differential equations in arbitrary discrete mechanics problems

Interpretable delta-learning of GW quasiparticle energies from GGA-DFT

All electron GW with linearized augmented plane waves for metals and semiconductors

Efficient Implementation of Density Functional Theory Based Embedding for Molecular and Periodic Systems Using Gaussian Basis Functions

Machine Learning Electron Density Prediction Using Weighted Smooth Overlap of Atomic Positions

High-Dimensional Operator Learning for Molecular Density Functional Theory

Neural network Gaussian processes as efficient models of potential energy surfaces for polyatomic molecules

Coarse-Grained Density Functional Theory Predictions via Deep Kernel Learning