P-DRUM: Post-hoc Descriptor-based Residual Uncertainty Modeling for Machine Learning Potentials
This addresses computational efficiency for researchers using machine learning potentials in materials science and chemistry, though it appears incremental as it builds on existing descriptor-based approaches.
The paper tackles the high computational cost of ensemble methods for uncertainty quantification in machine learning interatomic potentials by proposing P-DRUM, a post-hoc framework that uses descriptors from trained graph neural networks to estimate residual errors as uncertainty proxies, achieving competitive performance with established methods.
Ensemble method is considered the gold standard for uncertainty quantification (UQ) in machine learning interatomic potentials (MLIPs). However, their high computational cost can limit its practicality. Alternative techniques, such as Monte Carlo dropout and deep kernel learning, have been proposed to improve computational efficiency; however, some of these methods cannot be applied to already trained models and may affect the prediction accuracy. In this paper, we propose a simple and efficient post-hoc framework for UQ that leverages the descriptor of a trained graph neural network potential to estimate residual errors. We refer to this method as post-hoc descriptor-based residual uncertainty modeling (P-DRUM). P-DRUM models the discrepancy between MLIP predictions and ground truth values, allowing these residuals to act as proxies for prediction uncertainty. We explore multiple variants of P-DRUM and benchmark them against established UQ methods, evaluating both their effectiveness and limitations.