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

Abstract

Electron density $\rho (\vec{r})$ is the fundamental variable in the calculation of ground state energy with density functional theory (DFT). Beyond total energy, features and changes in $\rho (\vec{r})$ distributions are often used to capture critical physicochemical phenomena in functional materials. We present a machine learning framework for the prediction of $\rho (\vec{r})$. The model is based on equivariant graph neural networks and the electron density is predicted at special query point vertices that are part of the message passing graph, but only receive messages. The model is tested across multiple data sets of molecules (QM9), liquid ethylene carbonate electrolyte (EC) and LixNiyMnzCo(1-y-z)O2 lithium ion battery cathodes (NMC). For QM9 molecules, the accuracy of the proposed model exceeds typical variability in $\rho (\vec{r})$ obtained from DFT done with different exchange-correlation functionals. The accuracy on all three datasets is beyond state of the art and the computation time is orders of magnitude faster than DFT.

27 pages, 10 figures

References in corpus (9)

• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Machine Learning Force Fields
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Equivariant message passing for the prediction of tensorial properties and molecular spectra
• Informing Geometric Deep Learning with Electronic Interactions to Accelerate Quantum Chemistry
• Quantum deep field: data-driven wave function, electron density generation, and atomization energy prediction and extrapolation with machine learning
• DeepDFT: Neural Message Passing Network for Accurate Charge Density Prediction
• Efficient determination of the Hamiltonian and electronic properties using graph neural network with complete local coordinates
• Learning electron densities in the condensed phase

Cited by in corpus (8)

• Predicting the Charge Density Response in Metal Electrodes
• Reinforcement learning for traversing chemical structure space: Optimizing transition states and minimum energy paths of molecules
• Convolutional network learning of self-consistent electron density via grid-projected atomic fingerprints
• Topological descriptors for the electron density of inorganic solids
• Learning the Electrostatic Response of the Electron Density through a Symmetry-Adapted Vector Field Model
• A parameter-free statistical model for two-dimensional carbon nanostructures
• FIRE-GNN: Force-informed, Relaxed Equivariance Graph Neural Network for Rapid and Accurate Prediction of Surface Properties
• Neural Polarization: Toward Electron Density for Molecules by Extending Equivariant Networks