arXiv:2506.14850physics.chem-phcs.LG2025-06被引 4

预训练能提升力场模型稳定性,比单纯追求低误差更重要。

Beyond Force Metrics: Pre-Training MLFFs for Stable MD Simulations

  • 用GemNet-T模型,先在OC20上预训练再微调MD17数据。
  • 预训练模型使模拟时间延长三倍,但两模型力误差均低于5 meV/A。
  • 适合关注分子动力学长期稳定性的研究者使用。

机器学习力场(MLFF)有望加速从头算分子动力学(MD)模拟,其中精确的力预测至关重要但计算成本高。本文采用图神经网络模型GemNet-T作为MLFF,比较两种训练策略:(1) 直接在MD17数据集(1万样本)上训练;(2) 先在大规模OC20数据集上预训练,再在MD17上微调。尽管两种方法的力均方误差(MAE)均低于5 meV/A每原子,但仅靠低误差无法保证模拟稳定。值得注意的是,预训练的GemNet-T模型使轨迹持续时间显著延长,可达未预训练模型的三倍。通过分析力场的局部特性发现,预训练产生更结构化的隐层表示、对局部几何变化响应更平滑、相邻构型间力差更一致,这些共同提升了模拟的稳定性和可靠性。结果表明,大而多样的数据集预训练有助于捕捉复杂分子相互作用,且力的MAE并非衡量模拟稳定性的充分指标。

原文摘要 · Abstract (English)

Machine-learning force fields (MLFFs) have emerged as a promising solution for speeding up ab initio molecular dynamics (MD) simulations, where accurate force predictions are critical but often computationally expensive. In this work, we employ GemNet-T, a graph neural network model, as an MLFF and investigate two training strategies: (1) direct training on MD17 (10K samples) without pre-training, and (2) pre-training on the large-scale OC20 dataset followed by fine-tuning on MD17 (10K). While both approaches achieve low force mean absolute errors (MAEs), reaching 5 meV/A per atom, we find that lower force errors do not necessarily guarantee stable MD simulations. Notably, the pre-trained GemNet-T model yields significantly improved simulation stability, sustaining trajectories up to three times longer than the model trained from scratch. By analyzing local properties of the learned force fields, we find that pre-training produces more structured latent representations, smoother force responses to local geometric changes, and more consistent force differences between nearby configurations, all of which contribute to more stable and reliable MD simulations. These findings underscore the value of pre-training on large, diverse datasets to capture complex molecular interactions and highlight that force MAE alone is not always a sufficient metric of MD simulation stability.

力场模型分子动力学预训练

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