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).

ANI-1: an extensible neural network potential with DFT accuracy at force field computational cost

ANI-1: A data set of 20M off-equilibrium DFT calculations for organic molecules

A deep learning interatomic potential developed for atomistic simulation of carbon materials

QDπ: A Quantum Deep Potential Interaction Model for Drug Discovery

Transferable Machine Learning Interatomic Potential for Carbon Hydrogen Systems

ANI/EFP: Modeling Long-Range Interactions in ANI Neural Network with Effective Fragment Potentials

Reweighting from Molecular Mechanics Force Fields to the ANI-2x Neural Network Potential

Incorporating Neural Networks into the AMOEBA Polarizable Force Field

Ab initio Accuracy Neural Network Potential for Drug-like Molecules

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

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

Transfer learning for chemically accurate interatomic neural network potentials

Comparing ANI-2x, ANI-1ccx neural networks, force field, and DFT methods for predicting conformational potential energy of organic molecules

Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation

AMARO: All Heavy-Atom Transferable Neural Network Potentials of Protein Thermodynamics

Deep Potential Molecular Dynamics: A Scalable Model with the Accuracy of Quantum Mechanics

LAMMPS Implementation of Rapid Artificial Neural Network Derived Interatomic Potentials

Training Transferable Interatomic Neural Network Potentials for Reactive Chemistry: Improved Chemical Space Sampling

Scalable Training of Neural Network Potentials for Complex Interfaces Through Data Augmentation