用于从头算分子动力学模拟的模型知识迁移
Model Knowledge Transfer for Ab Initio Molecular Dynamics Simulations
-
摘要:研究背景 随着机器学习驱动的潜在能量面(PES)模型的快速发展,基于从头计算的分子动力学(AIMD)模拟得到了极大的促进,提供了比传统电子结构方法更快速的替代方案。然而,尽管这些模型在计算速度上具有优势,当直接用于预测分子属性时,预训练模型仍然会产生较大的计算开销和资源低效性,从而降低了其总体效率。因此,如何提高模型效率同时保留准确性,仍然是当前研究中的一大挑战。目的 本研究的主要目的是提出一种面向于从头算分子动力学模拟的模型知识迁移策略(KTAIMD),在不牺牲模型的准确性的情况下提高基于从头计算的分子动力学模拟(AIMD)的速率。我们旨在通过这一策略,解决现有机器学习驱动的分子模拟中仍存在计算开销和效率低下的问题,从而使得大规模模拟变得更加高效,并能够在更大范围内推广应用。方法 为了解决上述问题,本研究一种提出了面向从头算分子动力学模拟的模型知识迁移策略(KTAIMD)。具体地,我们基于可支持多种加速器和并行模式的PWDFT 计算软件构建了数据集,并采用了Data Parallelism数据并行方式实现多卡GPU加速训练预训练模型。然后,我们还基于多项式结构形式构建了轻量级神经网络,并采用模型迁移策略训练对该模型进行训练。结果 实验结果表明,KTAIMD策略的应用显著降低了内存使用量,减少了约一个数量级的内存消耗。此外,该方法成功地在保持原始模型准确性的同时,显著提高了计算效率。关键结果如下:内存使用量减少:通过KTAIMD策略,内存消耗降低了一个数量级。计算效率提高:通过知识迁移,计算效率得到了显著提升。模型精度保持:与传统方法相比,迁移后的模型准确性没有明显下降。结论 本研究提出的KTAIMD策略为分子动力学模拟提供了一种新的高效方法,显著降低了内存使用和计算资源消耗,且保持了原模型较高的模型精度。这一策略为大规模分子模拟提供了实践价值,并为进一步提高模拟效率和可扩展性设立了新的基准。未来的研究可以探索该方法在不同类型的分子系统中的应用,进一步拓展其应用范围,提升在高效计算中的潜力。Abstract: Machine learning based potential energy surface (PES) models have revolutionized ab initio molecular dynamics (AIMD) simulations, providing a much faster alternative to traditional electronic structure methods. Nevertheless, despite their speed advantages, these models incur significant computational overhead and resource inefficiencies when directly used for predicting molecular properties with pre-trained models, which ultimately reduces their overall efficiency. To address this, we propose a novel model knowledge transfer strategy for enhancing ab initio molecular dynamics (KTAIMD) simulations without compromising accuracy. This strategy, applied as a post-training refinement step, significantly reduces computational overhead while retaining the precision of the original model. Extensive experiments conducted on the plane-wave density functional theory (PWDFT) platform validate the KTAIMD strategy, demonstrating a notable reduction in memory usage by an order of magnitude. Consequently, this approach offers substantial practical value for large-scale simulations and establishes a new benchmark for efficiency and scalability in molecular dynamics.
下载: