Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions
arXiv:2110.05064
Abstract
Solving the Schrödinger equation is key to many quantum mechanical properties. However, an analytical solution is only tractable for single-electron systems. Recently, neural networks succeeded at modeling wave functions of many-electron systems. Together with the variational Monte-Carlo (VMC) framework, this led to solutions on par with the best known classical methods. Still, these neural methods require tremendous amounts of computational resources as one has to train a separate model for each molecular geometry. In this work, we combine a Graph Neural Network (GNN) with a neural wave function to simultaneously solve the Schrödinger equation for multiple geometries via VMC. This enables us to model continuous subsets of the potential energy surface with a single training pass. Compared to existing state-of-the-art networks, our Potential Energy Surface Network PESNet speeds up training for multiple geometries by up to 40 times while matching or surpassing their accuracy. This may open the path to accurate and orders of magnitude cheaper quantum mechanical calculations.
arXiv commentPublished as a conference paper at ICLR 2022
References in corpus (18)
- On the difficulty of training Recurrent Neural Networks
- On representing chemical environments
- SchNet - a deep learning architecture for molecules and materials
- On Calibration of Modern Neural Networks
- Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
- Optimizing Neural Networks with Kronecker-factored Approximate Curvature
- FCHL revisited: faster and more accurate quantum machine learning
- Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation
- On the Generalization of Equivariance and Convolution in Neural Networks to the Action of Compact Groups
- Fermionic neural-network states for ab-initio electronic structure
- Solving Many-Electron Schrödinger Equation Using Deep Neural Networks
- Optimizing large parameter sets in variational quantum Monte Carlo
- Quantum deep field: data-driven wave function, electron density generation, and atomization energy prediction and extrapolation with machine learning
- On Representing (Anti)Symmetric Functions
- Making Graph Neural Networks Worth It for Low-Data Molecular Machine Learning
- Scalable variational Monte Carlo with graph neural ansatz
- Bayesian Graph Neural Networks for Molecular Property Prediction
- Simulations of state-of-the-art fermionic neural network wave functions with diffusion Monte Carlo