Reducing training data needs with minimal multilevel machine learning (M3L)

Stefan Heinen, Danish Khan, Guido Falk von Rudorff, Konstantin Karandashev, Daniel Jose Arismendi Arrieta, Alastair J A Price, Surajit Nandi, Arghya Bhowmik, Kersti Hermansson, O Anatole von Lilienfeld*

*Corresponding author for this work

Research output: Contribution to journalJournal articleResearchpeer-review

9 Downloads (Pure)

Abstract

For many machine learning applications in science, data acquisition, not training, is the bottleneck even when avoiding experiments and relying on computation and simulation. Correspondingly, and in order to reduce cost and carbon footprint, training data efficiency is key. We introduce minimal multilevel machine learning (M3L) which optimizes training data set sizes using a loss function at multiple levels of reference data in order to minimize a combination of prediction error with overall training data acquisition costs (as measured by computational wall-times). Numerical evidence has been obtained for calculated atomization energies and electron affinities of thousands of organic molecules at various levels of theory including HF, MP2, DLPNO-CCSD(T), DFHFCABS, PNOMP2F12, and PNOCCSD(T)F12, and treating them with basis sets TZ, cc-pVTZ, and AVTZ-F12. Our M3L benchmarks for reaching chemical accuracy in distinct chemical compound sub-spaces indicate substantial computational cost reductions by factors of ∼1.01, 1.1, 3.8, 13.8, and 25.8 when compared to heuristic sub-optimal multilevel machine learning (M2L) for the data sets QM7b, QM9 LCCSD (T) , Electrolyte Genome Project, QM9AECCSD(T) , and QM9EACCSD(T) , respectively. Furthermore, we use M2L to investigate the performance for 76 density functionals when used within multilevel learning and building on the following levels drawn from the hierarchy of Jacobs Ladder: LDA, GGA, mGGA, and hybrid functionals. Within M2L and the molecules considered, mGGAs do not provide any noticeable advantage over GGAs. Among the functionals considered and in combination with LDA, the three on average top performing GGA and Hybrid levels for atomization energies on QM9 using M3L correspond respectively to PW91, KT2, B97D, and τ-HCTH, B3LYP∗(VWN5), and TPSSH.
Original languageEnglish
Article number025058
JournalMachine Learning: Science and Technology
Volume5
Issue number2
Number of pages14
ISSN2632-2153
DOIs
Publication statusPublished - 2024

Keywords

  • Quantum machine learning
  • Computational cost
  • Multilevel machine learning
  • Delta learning
  • Quantum chemistry
  • Kernel ridge regression

Fingerprint

Dive into the research topics of 'Reducing training data needs with minimal multilevel machine learning (M3L)'. Together they form a unique fingerprint.

Cite this