Physics-Informed Weakly Supervised Learning for Interatomic Potentials

Published: 17 Jun 2024, Last Modified: 17 Jul 2024ICML2024-AI4Science PosterEveryoneRevisionsBibTeXCC BY 4.0
Keywords: machine learning interatomic potential, physics informed method, weakly supervised method, molecule
TL;DR: Physics-Informed Weakly Supervised Learning
Abstract: Machine learning plays an increasingly important role in computational chemistry and materials science, complementing computationally intensive ab initio and first-principles methods. Despite their utility, machine-learning models often lack generalization capability and robustness during atomistic simulations, yielding unphysical energy and force predictions that hinder their real-world applications. We address this challenge by introducing a physics-informed, weakly supervised approach for training machine-learned interatomic potentials (MLIPs). We introduce two novel loss functions, using the concept of conservative forces and extrapolating the total energy via a Taylor expansion. Our approach enhances the accuracy of MLIPs applied to learning tasks with sparse training data set sizes and reduces the need for pre-training computationally demanding models. Particularly, we perform extensive experiments demonstrating reduced energy and force errors---often lower by a factor of two---for various baseline models and benchmark data sets. We also enhance the accuracy of energy and force predictions compared to previous methods that employ data augmentation via a Taylor expansion. Finally, we show that our approach facilitates MLIPs' training in a setting where the computation of forces is infeasible at the reference level, such as those employing complete-basis-set extrapolation.
Submission Number: 76
Loading