arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.00223physics.comp-ph
收藏

通过机器学习将从头算精确度的分子动力学极限推到1亿个原子

Pushing the limit of molecular dynamics with ab initio accuracy to 100 million atoms with machine learning

Weile Jia, Han Wang, Mohan Chen, Denghui Lu, Lin Lin, Roberto Car, Weinan E, Linfeng Zhang

摘要:

35年来,{\it ab initio}分子动力学(AIMD)一直是从第一原理建模复杂原子现象的首选方法。然而,大多数有源植入式医疗器械的应用受到计算成本的限制,最多有数千个原子的系统。我们报告说,基于机器学习的模拟协议(深势分子动力学),同时保留{\it从头算}的准确性,可以模拟超过1纳秒长的轨迹超过1亿个原子每天,使用高度优化的代码(GPU DeePMD套件)的首脑会议超级计算机。我们的代码可以有效地扩展到整个首脑会议的超级计算机,获得$91$ PFLOPS在双精度($45.5\%$的峰值)和{$162$/$275$ PFLOPS在混合单/半精度}。这项工作的巨大成就是,它打开了大门,模拟前所未有的规模和时间尺度与{\it从头}精度。它也为下一代超级计算机更好地集成机器学习和物理建模提出了新的挑战。

英文摘要:

For 35 years, {\it ab initio} molecular dynamics (AIMD) has been the method of choice for modeling complex atomistic phenomena from first principles. However, most AIMD applications are limited by computational cost to systems with thousands of atoms at most. We report that a machine learning-based simulation protocol (Deep Potential Molecular Dynamics), while retaining {\it ab initio} accuracy, can simulate more than 1 nanosecond-long trajectory of over 100 million atoms per day, using a highly optimized code (GPU DeePMD-kit) on the Summit supercomputer. Our code can efficiently scale up to the entire Summit supercomputer, attaining $91$ PFLOPS in double precision ($45.5\%$ of the peak) and {$162$/$275$ PFLOPS in mixed-single/half precision}. The great accomplishment of this work is that it opens the door to simulating unprecedented size and time scales with {\it ab initio} accuracy. It also poses new challenges to the next-generation supercomputer for a better integration of machine learning and physical modeling.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑