arXiv:2510.03046cs.LG2025-10被引 2

提出带不确定性的原子势能模型,实现高精度与可靠性兼顾的分子模拟。

Bayesian E(3)-Equivariant Interatomic Potential with Iterative Restratification of Many-body Message Passing

  • 基于能量力联合负对数似然损失,建模能量与原子力的不确定性
  • 在分布外检测和主动学习任务中优于传统方法,提升20%以上效率
  • 适合需要可靠预测的材料发现与动态模拟场景

机器学习势能(MLPs)已成为大规模原子模拟的关键工具,可在计算效率下实现从头算级精度。然而,现有MLPs在不确定性量化方面存在不足,限制了其在主动学习、校准和分布外(OOD)检测中的应用。本文提出贝叶斯E(3)等变的MLP,结合多体消息传递的迭代重分层机制。引入联合能量-力负对数似然(NLL$_\text{JEF}$)损失函数,显式建模能量与原子力的不确定性,在多个基准测试中显著优于传统损失函数。系统评估了深度集成、随机权重平均高斯、改进的变分在线牛顿法及拉普拉斯近似等多种贝叶斯方法,在不确定性预测、OOD检测、校准和主动学习任务上表现优异。结果表明,使用基于分歧的主动学习(BALD),该框架在能量和力不确定性量化基础上,性能超越随机采样与仅依赖能量不确定性的采样方式。本工作证实,贝叶斯等变神经网络可实现与先进模型相当的精度,并支持不确定性引导的主动学习、分布外检测与能量/力校准,为大规模原子模拟建立可靠的不确定性感知框架。

原文摘要 · Abstract (English)

Machine learning potentials (MLPs) have become essential for large-scale atomistic simulations, enabling ab initio-level accuracy with computational efficiency. However, current MLPs struggle with uncertainty quantification, limiting their reliability for active learning, calibration, and out-of-distribution (OOD) detection. We address these challenges by developing Bayesian E(3) equivariant MLPs with iterative restratification of many-body message passing. Our approach introduces the joint energy-force negative log-likelihood (NLL$_\text{JEF}$) loss function, which explicitly models uncertainty in both energies and interatomic forces, yielding substantially improved accuracy compared to conventional NLL losses. We systematically benchmark multiple Bayesian approaches, including deep ensembles with mean-variance estimation, stochastic weight averaging Gaussian, improved variational online Newton, and Laplace approximation by evaluating their performance on uncertainty prediction, OOD detection, calibration, and active learning tasks. We further demonstrate that NLL$_\text{JEF}$ facilitates efficient active learning by quantifying energy and force uncertainties. Using Bayesian active learning by disagreement (BALD), our framework outperforms random sampling and energy-uncertainty-based sampling. Our results demonstrate that Bayesian MLPs achieve competitive accuracy with state-of-the-art models while enabling uncertainty-guided active learning, OOD detection, and energy/forces calibration. This work establishes Bayesian equivariant neural networks as a powerful framework for developing uncertainty-aware MLPs for atomistic simulations at scale.

机器学习势不确定性量化主动学习等变网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。