GPU-Accelerated Approximate Kernel Method for Quantum Machine Learning
arXiv:2206.01580 · doi:10.1063/5.0108967
Abstract
Conventional kernel-based machine learning models for ab initio potential energy surfaces, while accurate and convenient in small data regimes, suffer immense computational cost as training set sizes increase. We introduce QML-Lightning, a PyTorch package containing GPU-accelerated approximate kernel models, which reduces the training time by several orders of magnitude, yielding trained models within seconds. QML-Lightning includes a cost-efficient GPU implementation of FCHL19, which together can yield energy and force predictions with competitive accuracy on a microsecond-per-atom timescale. Using modern GPU hardware, we report learning curves of energies and forces as well as timings as numerical evidence for select legacy benchmarks from atomisitic simulation including QM9, MD-17, and 3BPA.
arXiv comment25 pages, 5 figures
References in corpus (17)
- Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
- On representing chemical environments
- SchNet - a deep learning architecture for molecules and materials
- ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
- E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
- Quantum-Chemical Insights from Deep Tensor Neural Networks
- Machine Learning of Accurate Energy-Conserving Molecular Force Fields
- Towards Exact Molecular Dynamics Simulations with Machine-Learned Force Fields
- Alchemical and structural distribution based representation for improved QML
- WACSF - Weighted Atom-Centered Symmetry Functions as Descriptors in Machine Learning Potentials
- Equivariant message passing for the prediction of tensorial properties and molecular spectra
- sGDML: Constructing Accurate and Data Efficient Molecular Force Fields Using Machine Learning
- Operators in Machine Learning: Response Properties in Chemical Space
- Efficient non-parametric fitting of potential energy surfaces for polyatomic molecules with Gaussian processes
- Permutationally invariant polynomial regression for energies and gradients, using reverse differentiation, achieves orders of magnitude speed-up with high precision compared to other machine learning methods
- Challenges for Machine Learning Force Fields in Reproducing Potential Energy Surfaces of Flexible Molecules
- Efficient implementation of atom-density representations