High-speed and low-power molecular dynamics processing unit (MDPU) with ab initio accuracy
Abstract Molecular dynamics (MD) is an indispensable atomistic-scale computational tool widely-used in various disciplines. In the past decades, nearly all ab initio MD and machine-learning MD have been based on the general-purpose central/graphics processing units (CPU/GPU), which are well-known to suffer from their intrinsic “memory wall” and “power wall” bottlenecks. Consequently, nowadays MD calculations with ab initio accuracy are extremely time-consuming and power-consuming, imposing serious restrictions on the MD simulation size and duration. To solve this problem, here we propose a special-purpose MD processing unit (MDPU), which could reduce MD time and power consumption by about 10 3 times (10 9 times) compared to state-of-the-art machine-learning MD (ab initio MD) based on CPU/GPU, while keeping ab initio accuracy. With significantly-enhanced performance, the proposed MDPU may pave a way for the accurate atomistic-scale analysis of large-size and/or long-duration problems which were impossible/impractical to compute before.
Authors
- Denghui Lu (ORCID: https://orcid.org/0000-0003-0977-3635)
- Jun Cheng (ORCID: https://orcid.org/0000-0001-6971-0797)
- Hanhan Sun (ORCID: https://orcid.org/0000-0002-0653-1385)
- Yujia Zhang (ORCID: https://orcid.org/0009-0006-0647-7899)
- Bin Liang (ORCID: https://orcid.org/0000-0003-2655-9024)
- Xi Ding
- Pinghui Mo (ORCID: https://orcid.org/0009-0009-4591-8644)
- Xin Zhang (ORCID: https://orcid.org/0000-0002-6858-7835)
- Zhuoying Zhao (ORCID: https://orcid.org/0000-0001-9561-1362)
- E Weinan
- Mohan Chen (ORCID: https://orcid.org/0000-0002-8071-5633)
- Jianxing Huang (ORCID: https://orcid.org/0000-0001-6363-5562)
- Jiayu Dai (ORCID: https://orcid.org/0000-0001-7423-7500)
- Bo Chen (ORCID: https://orcid.org/0000-0002-0018-6891)
- Junhua Li (ORCID: https://orcid.org/0000-0002-0944-7891)
- Fei Liu (ORCID: https://orcid.org/0000-0003-3212-7594)
- Xinyu Li (ORCID: https://orcid.org/0000-0002-3730-0360)
- Yinan Wang (ORCID: https://orcid.org/0000-0002-9292-0781)
- Handong Wang (ORCID: https://orcid.org/0000-0001-5989-9338)
- Jie Liu (ORCID: https://orcid.org/0000-0001-7350-9850)
- Duo Zhang (ORCID: https://orcid.org/0000-0001-9591-2659)
- Linfeng Zhang (ORCID: https://orcid.org/0000-0002-8470-5846)
- Dawei Guan (ORCID: https://orcid.org/0000-0002-9439-6557)
- Mengchao Shi
Institutions
- Hunan University (CN)
- National University of Defense Technology (CN)
- Xiamen University (CN)
- University of Washington (US)
- Peking University (CN)
- Bay Institute (US)
- Institute of Applied Physics and Computational Mathematics (CN)
- Collaborative Innovation Center of Chemistry for Energy Materials (CN)
Publication Details
- Journal
- npj Computational Materials
- Published
- 2024-11-07
- DOI
- https://doi.org/10.1038/s41524-024-01422-3
- Citations
- 8
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.94
Funders
- National Natural Science Foundation of China