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

Download:

Current browse context:

physics.comp-ph

Change to browse by:

References & Citations

Bookmark

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

Physics > Computational Physics

Title: Transfer learning for chemically accurate interatomic neural network potentials

Abstract: Developing machine learning-based interatomic potentials from ab-initio electronic structure methods remains a challenging task for computational chemistry and materials science. This work studies the capability of transfer learning, in particular discriminative fine-tuning, for efficiently generating chemically accurate interatomic neural network potentials on organic molecules from the MD17 and ANI data sets. We show that pre-training the network parameters on data obtained from density functional calculations considerably improves the sample efficiency of models trained on more accurate ab-initio data. Additionally, we show that fine-tuning with energy labels alone can suffice to obtain accurate atomic forces and run large-scale atomistic simulations, provided a well-designed fine-tuning data set. We also investigate possible limitations of transfer learning, especially regarding the design and size of the pre-training and fine-tuning data sets. Finally, we provide GM-NN potentials pre-trained and fine-tuned on the ANI-1x and ANI-1ccx data sets, which can easily be fine-tuned on and applied to organic molecules.
Subjects: Computational Physics (physics.comp-ph); Machine Learning (stat.ML)
DOI: 10.1039/D2CP05793J
Cite as: arXiv:2212.03916 [physics.comp-ph]
  (or arXiv:2212.03916v2 [physics.comp-ph] for this version)

Submission history

From: Viktor Zaverkin [view email]
[v1] Wed, 7 Dec 2022 19:21:01 GMT (2120kb,D)
[v2] Sat, 28 Jan 2023 11:26:23 GMT (3226kb,D)

Link back to: arXiv, form interface, contact.