On the equivalence of molecular graph convolution and molecular wave function with poor basis set

Abstract

In this study, we demonstrate that the linear combination of atomic orbitals (LCAO), an approximation of quantum physics introduced by Pauling and Lennard-Jones in the 1920s, corresponds to graph convolutional networks (GCNs) for molecules. However, GCNs involve unnecessary nonlinearity and deep architecture. We also verify that molecular GCNs are based on a poor basis function set compared with the standard one used in theoretical calculations or quantum chemical simulations. From these observations, we describe the quantum deep field (QDF), a machine learning (ML) model based on an underlying quantum physics, in particular the density functional theory (DFT). We believe that the QDF model can be easily understood because it can be regarded as a single linear layer GCN. Moreover, it uses two vanilla feedforward neural networks to learn an energy functional and a Hohenberg--Kohn map that have nonlinearities inherent in quantum physics and the DFT. For molecular energy prediction tasks, we demonstrated the viability of an ``extrapolation,'' in which we trained a QDF model with small molecules, tested it with large molecules, and achieved high extrapolation performance. This will lead to reliable and practical applications for discovering effective materials. The implementation is available at https://github.com/masashitsubaki/QuantumDeepField_molecule.

References in corpus (17)

• Adam: A Method for Stochastic Optimization
• Neural Message Passing for Quantum Chemistry
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• SchNet - a deep learning architecture for molecules and materials
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Molecular Graph Convolutions: Moving Beyond Fingerprints
• Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Hidden Physics Models: Machine Learning of Nonlinear Partial Differential Equations
• Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation
• A Transferable Machine-Learning Model of the Electron Density
• How Neural Networks Extrapolate: From Feedforward to Graph Neural Networks
• Solid Harmonic Wavelet Scattering for Predictions of Molecule Properties
• Predicting charge density distribution of materials using a local-environment-based graph convolutional network
• Alchemy: A Quantum Chemistry Dataset for Benchmarking AI Models
• Deep Neural Network Computes Electron Densities and Energies of a Large Set of Organic Molecules Faster than Density Functional Theory (DFT)