提出可同时输出原子力与不确定性的等变深度学习框架,提升分子模拟可靠性。
Equivariant Evidential Deep Learning for Interatomic Potentials
- 用3×3协方差张量建模原子力不确定性,保证旋转不变性
- 在多个分子基准上实现更高精度与数据效率,优于基线和集成方法
- 单模型推理,适合主动学习与高可靠性分子动力学模拟
不确定性量化(UQ)对评估机器学习势函数(MLIP)在分子动力学(MD)模拟中的可靠性至关重要,可识别外推区域并支持不确定性感知的训练流程,如主动学习。现有MLIP的UQ方法常受限于高计算成本或性能不足。证据深度学习(EDL)提供了一种理论完备的单模型方案,可在一次前向传播中同时估计随机不确定性和认知不确定性。然而,将证据形式从标量目标推广到原子力等矢量值量时,面临重大挑战,尤其在旋转变换下保持统计自洽性。为此,我们提出“等变证据深度学习用于原子间势”(e²IP),一种无需依赖主干网络的框架,通过表示一个在旋转下等变的3×3对称正定协方差张量,联合建模原子力及其不确定性。在多种分子基准上的实验表明,e²IP在准确性、效率与可靠性之间表现更优,优于非等变证据基线与广泛使用的集成方法。其全等变架构还提升了数据效率,同时保持单模型推理速度。
原文摘要 · Abstract (English)
Uncertainty quantification (UQ) is critical for assessing the reliability of machine learning interatomic potentials (MLIPs) in molecular dynamics (MD) simulations, identifying extrapolation regimes and enabling uncertainty-aware workflows such as active learning for training dataset construction. Existing UQ approaches for MLIPs are often limited by high computational cost or suboptimal performance. Evidential deep learning (EDL) provides a theoretically grounded single-model alternative that determines both aleatoric and epistemic uncertainty in a single forward pass. However, extending evidential formulations from scalar targets to vector-valued quantities such as atomic forces introduces substantial challenges, particularly in maintaining statistical self-consistency under rotational transformations. To address this, we propose \textit{Equivariant Evidential Deep Learning for Interatomic Potentials} ($\text{e}^2$IP), a backbone-agnostic framework that models atomic forces and their uncertainty jointly by representing uncertainty as a full $3\times3$ symmetric positive definite covariance tensor that transforms equivariantly under rotations. Experiments on diverse molecular benchmarks show that $\text{e}^2$IP provides a stronger accuracy-efficiency-reliability balance than the non-equivariant evidential baseline and the widely used ensemble method. It also achieves better data efficiency through the fully equivariant architecture while retaining single-model inference efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。