vix.ing · top · new · best · stats

High Accuracy Uncertainty-Aware Interatomic Force Modeling with Equivariant Bayesian Neural Networks

2023/04/05 by Tim Rensmeyer, Rensmeyer, Tim, Ben Craig +5 · 1 citation
Computer Science · Materials Science · #Chemical Physics (physics.chem-ph) #Computational Physics (physics.comp-ph) #FOS: Computer and information sciences #FOS: Physical sciences #Machine Learning (cs.LG) #Machine Learning and Algorithms #Machine Learning in Materials Science #Neural Networks and Applications

paper · pdf · doi:10.48550/arxiv.2304.03694

openalex publication_date 2023/04/05 · openalex created_date 2023/04/11 · openalex updated_date 2026/07/28

Abstract

Even though Bayesian neural networks offer a promising framework for modeling uncertainty, active learning and incorporating prior physical knowledge, few applications of them can be found in the context of interatomic force modeling. One of the main challenges in their application to learning interatomic forces is the lack of suitable Monte Carlo Markov chain sampling algorithms for the posterior density, as the commonly used algorithms do not converge in a practical amount of time for many of the state-of-the-art architectures. As a response to this challenge, we introduce a new Monte Carlo Markov chain sampling algorithm in this paper which can circumvent the problems of the existing sampling methods. In addition, we introduce a new stochastic neural network model based on the NequIP architecture and demonstrate that, when combined with our novel sampling algorithm, we obtain predictions with state-of-the-art accuracy as well as a good measure of uncertainty.

Cited by

Related