LLM2D
通过降低特征相关性提高GNN力场模型的稳定性
Improving the Stability of GNN Force Field Models by Reducing Feature Correlation
作者: Yujie Zeng, Wenlong He, Ihor Vasyltsov, Jiaxin Wei, Ying Zhang, Lin Chen, Yuehua Dai
发布日期: 2/19/2025
arXiv ID: oai:arXiv.org:2502.12548v1

摘要

arXiv:2502.12548v1 宣告类型: cross 摘要: 近年来,基于图神经网络的力场模型(GNNFF)在分子动力学(MD)模拟中得到了广泛应用,而MD模拟是半导体材料研究中最经济有效的方法之一。然而,即使这些模型在训练数据集(内部数据集)上提供了高精度的能量和力的绝对误差均值(MAE),在使用于外部数据集进行长时间MD模拟时,它们通常会变得不稳定。本文提出了一种基于特征相关性的方法,以增强GNNFF模型在MD模拟中的稳定性。我们揭示了特征相关性与GNNFF模型稳定性之间的负相关关系,并设计了一种具有动态损失系数调度器的损失函数,可以减少边缘特征的相关性,并适用于一般的GNNFF训练。我们还提出了一个经验度量来评估MD模拟中的稳定性。实验表明,我们的方法可以显著提高GNNFF模型的稳定性,尤其是在外部数据上的稳定性提升,仅有不到3%的计算开销。例如,我们可以通过这项方法将Allegro模型的稳定MD模拟时间从0.03皮秒扩展到10皮秒。