DeePMD-kit v2: A software package for Deep Potential models

Jinzhe Zeng,Duo Zhang,Denghui Lu,Pinghui Mo,Zeyu Li,Yixiao Chen,Marián Rynik,Li'ang Huang,Ziyao Li,Shaochen Shi,Yingze Wang,Haotian Ye,Ping Tuo,Jiabin Yang,Ye Ding,Yifan Li,Davide Tisi,Qiyu Zeng,Han Bao,Yu Xia,Jiameng Huang,Koki Muraoka,Yibo Wang,Junhan Chang,Fengbo Yuan,Sigbjørn Løland Bore,Chun Cai,Yinnian Lin,Bo Wang,Jiayan Xu,Jia-Xin Zhu,Chenxing Luo,Yuzhi Zhang,Rhys E. A. Goodall,Wenshuo Liang,Anurag Kumar Singh,Sikai Yao,Jingchao Zhang,Renata Wentzcovitch,Jiequn Han,Jie Liu,Weile Jia,Darrin M. York,Weinan E,Roberto Car,Linfeng Zhang,Han Wang

DOI: https://doi.org/10.1063/5.0155600

2023-04-19

Abstract:DeePMD-kit is a powerful open-source software package that facilitates molecular dynamics simulations using machine learning potentials (MLP) known as Deep Potential (DP) models. This package, which was released in 2017, has been widely used in the fields of physics, chemistry, biology, and material science for studying atomistic systems. The current version of DeePMD-kit offers numerous advanced features such as DeepPot-SE, attention-based and hybrid descriptors, the ability to fit tensile properties, type embedding, model deviation, Deep Potential - Range Correction (DPRc), Deep Potential Long Range (DPLR), GPU support for customized operators, model compression, non-von Neumann molecular dynamics (NVNMD), and improved usability, including documentation, compiled binary packages, graphical user interfaces (GUI), and application programming interfaces (API). This article presents an overview of the current major version of the DeePMD-kit package, highlighting its features and technical details. Additionally, the article benchmarks the accuracy and efficiency of different models and discusses ongoing developments.

Chemical Physics,Atomic and Molecular Clusters

What problem does this paper attempt to address?

The paper introduces the latest version (v2.2.1) of the DeePMD-kit software package, which is an open-source tool for molecular dynamics simulation using the Deep Potential (DP) model. The DP model is a machine learning-based potential energy model widely used in fields such as physics, chemistry, biology, and materials science for studying atomic systems. Since its release in 2017, DeePMD-kit has developed many advanced features, including DeepPot-SE, attention-based and mixed descriptors, tensor property fitting, type embedding, model bias, Deep Potential - Range Correction (DPRc), and Deep Potential Long Range (DPLR).

The paper focuses on the following:

1. Provides a detailed introduction to the current features of DeePMD-kit, including various descriptors (such as local frames, DeepPot-SE, attention-based, mixed descriptors, etc.), network structures (such as adaptive switching functions, tensor property fitting), GPU support, and model compression.
2. Conducts benchmark tests on the accuracy and efficiency of different models.
3. Discusses the continuous development and future improvements of the software.

In summary, this paper aims to address how to enhance the efficiency and accuracy of molecular dynamics simulation using deep learning techniques through the DeePMD-kit software package, particularly in dealing with various atomic systems. It also discusses how to continuously update and expand its functionality to meet different research demands.