Files in this item



application/pdfWK01_5528.pdf (1MB)


application/pdf5528.pdf (18kB)


Title:Utilization Of Neural Networks For Gpu-accelerated Diffusion Monte Carlo For Vibrational Problems
Author(s):Lu, Fenris
Contributor(s):McCoy, Anne B.; DiRisio, Ryan J.
Subject(s):Theory and Computation
Abstract:In recent years, there has been increasing interest in combining machine learning with computational chemistry. Many machine learning applications in theoretical chemistry have focused on the development of potential energy surfaces (PES) based on \emph{ab initio} data points\footnote{Manzhos, S.; Carrington, T. Chem. Rev. Article ASAP.}. Another computational resource that has gained popularity in recent years is the use of Graphics Processing Units (GPUs) to accelerate calculations such as molecular dynamics simulations \footnote{Abraham, M. J.; Murtola, T.; Schulz, R.; Páll, S.; Smith, J. C.; Hess, B.; Lindahl, E. SoftwareX 2015, 1–2, 19–25}. In this work, we take advantage of the knowledge gained from both of these areas, and apply these approaches to the evaluations of the potential energy for Diffusion Monte Carlo (DMC) calculations. Diffusion Monte Carlo is a stochastic simulation approach through which an ensemble of localized functions (walkers) converges to the ground state vibrational wave function at long propagation times. This is accomplished through the ensemble randomly sampling a potential energy surface. These calculations require on the order of 10$^8$ to 10$^10$ potential energy evaluations, making the call to a potential energy surface the most expensive part of these calculations. To this end, we have developed a general technique for “learning” already existing potential energy surfaces using neural networks, which can then easily be evaluated on GPUs using TensorFlow. The technique entails running a small-scale DMC simulation to collect training data, coordinates and energies, and using this data to fit a subspace of the potential energy surface that is relevant for the DMC simulations. We find that using the neural network potential energy surface maintains the fidelity of the wave function while also enabling the use of ensemble sizes orders of magnitude larger than previously used for the same amount of wall time. We test this method on small systems such as \chem{CH5+}, \chem{H2O}, and water dimer.
Issue Date:2021-06-23
Publisher:International Symposium on Molecular Spectroscopy
Genre:Conference Paper / Presentation
Date Available in IDEALS:2021-09-24

This item appears in the following Collection(s)

Item Statistics