用粒子动态生成高效训练数据,提升分子力场模型精度与效率
Stein Kernelized Molecular Dynamics for Active Learning of Interatomic Potentials

- 基于斯坦因变分梯度的粒子动力学,自适应采样高信息量构型
- 相同样本数下,模型精度更高,训练迭代次数更少
- 适合需要高效数据采集的分子模拟与机器学习势能研究
机器学习原子间势能(MLIPs)可实现高效精确的原子模拟,但其性能高度依赖训练数据的质量与多样性。本文提出斯坦因核化分子动力学(SKMD),一种增强采样方法,通过相互作用粒子动力学获取用于主动学习和微调的高信息量训练构型。SKMD是斯坦因变分梯度下降的随机版本,结合异步粒子更新与全局原子描述符核函数,提供对称性感知的构型相似性度量。与传统增强采样方法不同,SKMD保持玻尔兹曼分布为动力学的渐近分布,平衡了构型探索与能量景观高概率区域的吸引。我们进一步提出基于自适应停止准则的在线数据采集策略,有效筛选非冗余数据。在穆勒-布朗势能与丙氨酸二肽的MACE势能模型上验证,相比基准方法,本方法以相同样本数实现更高精度且减少训练迭代。
原文摘要 · Abstract (English)
Machine learning interatomic potentials (MLIPs) enable efficient and accurate atomistic simulations but depend critically on the quality and diversity of the training data. We introduce Stein kernelized molecular dynamics (SKMD), an enhanced sampling method that uses interacting particle dynamics to acquire informative training configurations for the active learning and fine-tuning of MLIPs. SKMD corresponds to a stochastic variant of Stein variational gradient descent that is adapted for molecular dynamics by incorporating asynchronous particle updates and a kernel of global atomic descriptors, which provides a symmetry-aware measure of configurational similarity. Unlike other enhanced samplers used in molecular dynamics, SKMD preserves the Boltzmann distribution as the asymptotic distribution of the dynamics. This property enforces a balance between the exploration of diverse configurations and attraction toward high-probability regions of the energy landscape. We further propose an approach to efficient online data acquisition using an adaptive stopping criterion that selects non-redundant training data over the course of simulation. We demonstrate SKMD for the active learning of a neural network model of the Müller-Brown potential and the fine-tuning of a MACE interatomic potential for alanine dipeptide. Compared to active learning baselines, our method achieves higher model accuracy in fewer training iterations with the same number of acquired training samples.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。