Abstract:Density functional theory (DFT) is one of the main methods in Quantum Chemistry that offers an attractive trade off between the cost and accuracy of quantum chemical computations. The electron density plays a key role in DFT. In this work, we explore whether machine learning - more specifically, deep neural networks (DNNs) - can be trained to predict electron densities faster than DFT. First, we choose a practically efficient combination of a DFT functional and a basis set (PBE0/pcS-3) and use it to generate a database of DFT solutions for more than 133,000 organic molecules from a previously published database QM9. Next, we train a DNN to predict electron densities and energies of such molecules. The only input to the DNN is an approximate electron density computed with a cheap quantum chemical method in a small basis set (HF/cc-VDZ). We demonstrate that the DNN successfully learns differences in the electron densities arising both from electron correlation and small basis set artifacts in the HF computations. All qualitative features in density differences, including local minima on lone pairs, local maxima on nuclei, toroidal shapes around C-H and C-C bonds, complex shapes around aromatic and cyclopropane rings and CN group, etc. are captured by the DNN. Accuracy of energy predictions by the DNN is ~ 1 kcal/mol, on par with other models reported in the literature, while those models do not predict the electron density. Computations with the DNN, including HF computations, take much less time that DFT computations (by a factor of ~20-30 for most QM9 molecules in the current version, and it is clear how it could be further improved).

Toward Orbital-Free Density Functional Theory with Small Data Sets and Deep Learning

Deepks: A Comprehensive Data-Driven Approach Toward Chemically Accurate Density Functional Theory

KineticNet: Deep learning a transferable kinetic energy functional for orbital-free density functional theory

Accelerating finite-temperature Kohn-Sham density functional theory with deep neural networks

Deep learning density functional theory Hamiltonian in real space

Machine Learning Approaches toward Orbital-free Density Functional Theory: Simultaneous Training on the Kinetic Energy Density Functional and Its Functional Derivative

D4FT: A Deep Learning Approach to Kohn-Sham Density Functional Theory

Deep learning and density-functional theory

Overcoming the Barrier of Orbital-Free Density Functional Theory for Molecular Systems Using Deep Learning

Deep-learning density functional theory Hamiltonian for efficient ab initio electronic-structure calculation

Highly Accurate Real-space Electron Densities with Neural Networks

By-passing the Kohn-Sham equations with machine learning

Neural-network Density Functional Theory Based on Variational Energy Minimization

Deep Neural Network Computes Electron Densities and Energies of a Large Set of Organic Molecules Faster than Density Functional Theory (DFT)

Neural network distillation of orbital dependent density functional theory

Deep Charge: A Deep Learning Model of Electron Density from One-Shot Density Functional Theory Calculation

Variational principle to regularize machine-learned density functionals: the non-interacting kinetic-energy functional

Density functionals from deep learning

Pushing the frontiers of density functionals by solving the fractional electron problem