Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions
(Submitted on 11 Oct 2021 (v1), revised 26 Nov 2021 (this version, v2), latest version 29 Mar 2022 (v3))
Abstract: Solving the Schr\"odinger 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\"odinger 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.
Submission history
From: Nicholas Gao [view email][v1] Mon, 11 Oct 2021 07:58:31 GMT (2222kb,D)
[v2] Fri, 26 Nov 2021 08:28:58 GMT (2238kb,D)
[v3] Tue, 29 Mar 2022 07:21:24 GMT (2208kb,D)
Link back to: arXiv, form interface, contact.