TensorMD: Molecular Dynamics Simulation with Ab Initio Accuracy of 50 Billion Atoms
Molecular dynamics simulation emerges as an important area that HPC+AI helps to investigate the physical properties, with machine-learning interatomic potentials (MLIPs) being used. General-purpose machine-learning (ML) tools have been leveraged in MLIPs, but they are not perfectly matched with each other, since many optimization opportunities in MLIPs have been missed by ML tools. This inefficiency arises from the fact that HPC+AI applications work with far more computational complexity compared with pure AI scenarios. This paper has developed an MLIP, named TensorMD, independently from any ML tool. TensorMD has been evaluated on two supercomputers and scaled to 51.8 billion atoms, i.e., ~ 3× compared with state-of-the-art.
Authors
- Honghui Shang (ORCID: https://orcid.org/0000-0003-4957-4251)
- Jiahao Shan (ORCID: https://orcid.org/0000-0003-0783-2240)
- Zhenchuan Chen (ORCID: https://orcid.org/0009-0002-3756-3950)
- Haifeng Song (ORCID: https://orcid.org/0000-0001-6215-0125)
- Xingyu Gao (ORCID: https://orcid.org/0000-0002-6736-9377)
- Yucheng Ouyang (ORCID: https://orcid.org/0000-0001-8315-3667)
- Xiaobing Feng (ORCID: https://orcid.org/0000-0003-2909-7750)
- Xin Chen (ORCID: https://orcid.org/0000-0002-0562-0319)
- Ying Liu (ORCID: https://orcid.org/0000-0003-1599-9665)
- Huimin Cui (ORCID: https://orcid.org/0000-0002-2491-7679)
- Rongfen Lin (ORCID: https://orcid.org/0000-0003-0321-5405)
- Lifang Wang (ORCID: https://orcid.org/0000-0002-0408-549X)
- Xin Chen (ORCID: https://orcid.org/0000-0001-9643-0870)
- Fang Li (ORCID: https://orcid.org/0000-0002-3555-2126)
Institutions
- Chinese Academy of Sciences (CN)
- Institute of Computing Technology (CN)
- Institute of Applied Physics and Computational Mathematics (CN)
Publication Details
- Published
- 2025-02-28
- DOI
- https://doi.org/10.1145/3710848.3710882
- Citations
- 1
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.52