We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Physics > Chemical Physics

Title: GPU-Accelerated Approximate Kernel Method for Quantum Machine Learning

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.
Comments: 25 pages, 5 figures
Subjects: Chemical Physics (physics.chem-ph)
DOI: 10.1063/5.0108967
Cite as: arXiv:2206.01580 [physics.chem-ph]
  (or arXiv:2206.01580v2 [physics.chem-ph] for this version)

Submission history

From: Nicholas Browning [view email]
[v1] Fri, 3 Jun 2022 14:04:44 GMT (1125kb,D)
[v2] Mon, 6 Jun 2022 13:07:10 GMT (1125kb,D)

Link back to: arXiv, form interface, contact.