Transition1x: a dataset for building generalizable reactive machine learning potentials

Mathias Schreiner*, Arghya Bhowmik, Tejs Vegge, Jonas Busk, Ole Winther

*Corresponding author for this work

Research output: Contribution to journalJournal articleResearchpeer-review

29 Citations (Scopus)
18 Downloads (Pure)

Abstract

Machine Learning (ML) models have, in contrast to their usefulness in molecular dynamics studies, had limited success as surrogate potentials for reaction barrier search. This is primarily because available datasets for training ML models on small molecular systems almost exclusively contain configurations at or near equilibrium. In this work, we present the dataset Transition1x containing 9.6 million Density Functional Theory (DFT) calculations of forces and energies of molecular configurations on and around reaction pathways at the ωB97x/6–31 G(d) level of theory. The data was generated by running Nudged Elastic Band (NEB) with DFT on 10k organic reactions of various types while saving intermediate calculations. We train equivariant graph message-passing neural network models on Transition1x and cross-validate on the popular ANI1x and QM9 datasets. We show that ML models cannot learn features in transition state regions solely by training on hitherto popular benchmark datasets. Transition1x is a new challenging benchmark that will provide an important step towards developing next-generation ML force fields that also work far away from equilibrium configurations and reactive systems.
Original languageEnglish
Article number779
JournalScientific Data
Volume9
Issue number1
Number of pages9
ISSN2052-4463
DOIs
Publication statusPublished - 2022

Bibliographical note

Publisher Copyright:
© 2022, The Author(s).

Cite this