There is a newer version of the record available.

Published August 3, 2020 | Version v1
Dataset Open

Fast Bayesian force fields from active learning and mapped Gaussian processes: application to structural phase transition of stanene

  • 1. John A. Paulson School of Engineering and Applied Sciences, Harvard University, Cambridge, MA 02138, USA
  • 2. Robert Bosch LLC, Research and Technology Center, Cambridge, Massachusetts 02142, USA

* Contact person

Description

Gaussian process (GP) regression is one promising technique of constructing machine learning force fields with built-in uncertainty quantification, which can be used to monitor the quality of model predictions. A current limitation of existing GP force fields is that the prediction cost grows linearly with the size of the training data set, making accurate GP predictions slow. In this work, we exploit the special structure of the kernel function to construct a mapping of the trained Gaussian process model, including both forces and their uncertainty predictions, onto spline functions of low-dimensional structural features. This method is incorporated in the Bayesian active learning workflow for training of Bayesian force fields. To demonstrate the capabilities of this method, we construct a force field for stanene and perform large scale dynamics simulation of its structural evolution. We provide a fully open-source implementation of our method, as well as the training and testing examples with the stanene dataset.

Files

File preview

files_description.md

All files

Files (92.2 MiB)

Name Size
md5:b978e36acfef7db47a058d30da5b0091
597 Bytes Preview Download
md5:a595dc4a44ed4f193fb5c633c2ac5109
88.3 MiB Preview Download
md5:a147d2f8a8d8f2479110735a4159f61d
374.8 KiB Preview Download
md5:62cbb48b16b2703d0eb31dfc7cc4341a
94.1 KiB Preview Download
md5:980dac59df745cebcd8acb74f3616a27
3.4 MiB Preview Download
md5:341a49a9f0fae27ff5e71615cdbd5584
733 Bytes Preview Download

References

Preprint
Y. Xie, J. Vandermause, L. Sun, A. Cepellotti, B. Kozinsky, in preparation.

Journal reference (Paper and code on which this method is based)
J. Vandermause, S.B. Torrisi, S. Batzner, Y. Xie, L. Sun, A.M. Kolpak, B. Kozinsky, npj Comput Mater 6, 20 (2020)., doi: 10.1038/s41524-020-0283-z

Software (Code in which the method is implemented)
J. Vandermause, S.B. Torrisi, S. Batzner, Y. Xie, L. Sun, A.M. Kolpak, B. Kozinsky, npj Comput Mater 6, 20 (2020).