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.

arXiv comment27 pages, 10 figures

References in corpus (22)

• Neural Message Passing for Quantum Chemistry
• SchNet - a deep learning architecture for molecules and materials
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Machine Learning Force Fields
• Assessing the Performance of Recent Density Functionals for Bulk Solids
• By-passing the Kohn-Sham equations with machine learning
• Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
• Directional Message Passing for Molecular Graphs
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• A Transferable Machine-Learning Model of the Electron Density
• Equivariant message passing for the prediction of tensorial properties and molecular spectra
• High-throughput computational screening for solid-state Li-ion conductors
• Cormorant: Covariant Molecular Neural Networks
• 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
• Predicting charge density distribution of materials using a local-environment-based graph convolutional network
• Deep Neural Network Computes Electron Densities and Energies of a Large Set of Organic Molecules Faster than Density Functional Theory (DFT)
• The fundamentals of quantum machine learning
• DeepDFT: Neural Message Passing Network for Accurate Charge Density Prediction
• Efficient prediction of 3D electron densities using machine learning
• 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