dpdata: A Scalable Python Toolkit for Atomistic Machine Learning Data Sets
Seamless management of atomistic data sets is a critical prerequisite for the successful development and deployment of machine learning potentials (MLPs). Here, we present dpdata, an open-source Python library designed to streamline every aspect of MLP data handling. Built upon a flexible, plugin-based architecture, dpdata supports reading, writing, and converting between a broad range of file formats─from popular quantum-chemistry packages and molecular-dynamics engines to specialized MLP frameworks. Users may define custom data types, formats, drivers, and minimizers, enabling effortless extension to emerging software. Key utilities include automated train-test splitting, coordinate perturbation for active learning, outlier-energy removal, Δ-learning data set generation, error-metric computation, and unit conversion. Through efficient NumPy-backed storage and system-level operations, dpdata achieves significant memory saving and inference speedups over configuration-by-configuration tools such as ASE. We also highlight practical impact, with dpdata used across published studies, for format conversion, data storage, coordinate perturbation, and utilization in other projects for data processing.
Authors
- Cao Thang Nguyen
- Ping Tuo (ORCID: https://orcid.org/0000-0002-6477-5900)
- Rundong Liao
- Anyang Peng (ORCID: https://orcid.org/0000-0002-0630-2187)
- Logan Ward (ORCID: https://orcid.org/0000-0002-1323-5939)
- Chenxing Luo (ORCID: https://orcid.org/0000-0003-4116-6851)
- Wanrun Jiang (ORCID: https://orcid.org/0000-0001-7833-9862)
- Pan Xiang (ORCID: https://orcid.org/0000-0003-3880-6221)
- Yixiao Chen (ORCID: https://orcid.org/0000-0001-8201-5887)
- Liping Liu
- Fengbo Yuan (ORCID: https://orcid.org/0009-0002-3949-8785)
- Jinzhe Zeng (ORCID: https://orcid.org/0000-0002-1515-8172)
- Jiaxin Zhu (ORCID: https://orcid.org/0000-0002-3471-4728)
- Chengqian Zhang (ORCID: https://orcid.org/0000-0001-7829-0587)
- R. Zhou (ORCID: https://orcid.org/0000-0002-5789-4629)
- Renxi Liu (ORCID: https://orcid.org/0000-0002-5288-4149)
- Marián Rynik
- Junjie Wang (ORCID: https://orcid.org/0000-0003-1411-3324)
- Xuyuan Zhou
- Denan Li
- Yong‐Bin Zhuang (ORCID: https://orcid.org/0000-0001-5182-8084)
- Kaiwei Wan (ORCID: https://orcid.org/0000-0001-8518-6550)
- Bowen Li (ORCID: https://orcid.org/0009-0004-8729-9224)
- Wenshuo Liang (ORCID: https://orcid.org/0000-0003-0646-8803)
- Haidi Wang (ORCID: https://orcid.org/0000-0003-4768-2136)
- Jiahao Fan
- Haoxi Li (ORCID: https://orcid.org/0009-0004-8369-1042)
- Yifan Li (ORCID: https://orcid.org/0000-0003-3217-0592)
- Handong Wang (ORCID: https://orcid.org/0000-0001-5989-9338)
- Duo Zhang (ORCID: https://orcid.org/0000-0001-9591-2659)
- Yuzhi Zhang (ORCID: https://orcid.org/0000-0002-5841-1107)
- Linfeng Zhang (ORCID: https://orcid.org/0000-0002-8470-5846)
- Weihong Xu
- Tao Chen (ORCID: https://orcid.org/0000-0002-7990-8788)
- Jiameng Huang (ORCID: https://orcid.org/0009-0002-9701-8145)
- Xulong Peng (ORCID: https://orcid.org/0000-0003-3232-3798)
- Zezhong Zhang (ORCID: https://orcid.org/0000-0002-2062-8327)
- Jinchao Zhang (ORCID: https://orcid.org/0000-0003-4611-9675)
- Yingze Wang (ORCID: https://orcid.org/0000-0002-1706-3791)
Institutions
- University of North Carolina at Chapel Hill (US)
- University of Science and Technology of China (CN)
- Institute of Applied Physics (UA)
- Argonne National Laboratory (US)
- Microsoft (United States) (US)
- Hefei University of Technology (CN)
- City University of Hong Kong (HK)
- Iowa State University (US)
- Xiamen University (CN)
- Chinese Academy of Sciences (CN)
- Princeton University (US)
- Peking University (CN)
- Baker Hughes (United States) (US)
- Art Institute of Portland (US)
- Suzhou University of Science and Technology (CN)
- Westlake University (CN)
- University of Alabama at Birmingham (US)
- Laboratory of Molecular Genetics (PL)
- Microsoft (Finland) (FI)
- IRD Fuel Cells (Denmark) (DK)
- Beijing National Laboratory for Molecular Sciences (CN)
- FZU ‒ Institute of Physics of the Academy of Sciences of the Czech Republic (CZ)
- National Test Pilot School (US)
- Center for Interdisciplinary Studies (IN)
- Computational Physics (United States) (US)
- Collaborative Innovation Center of Advanced Microstructures (CN)
- SP Technology (South Korea) (KR)
- Institute of Applied Physics and Computational Mathematics (CN)
- Institute of Semiconductors (CN)
- Collaborative Innovation Center of Chemistry for Energy Materials (CN)
- Institute of Physics (CN)
- Ningbo Institute of Industrial Technology (CN)
- Centre for Innovation Studies (CA)
- Preferred Networks (Japan) (JP)
- Ulsan National Institute of Science and Technology (KR)
- East China Normal University (CN)
- Comenius University Bratislava (SK)
- Columbia University (US)
- Beihang University (CN)
- Nanjing University (CN)
- University of California, Berkeley (US)
Publication Details
- Journal
- Journal of Chemical Information and Modeling
- Published
- 2025-10-20
- DOI
- https://doi.org/10.1021/acs.jcim.5c01767
- Citations
- 1
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.52