构建超百万原子构型数据集,助力机器学习势函数精准训练
LeMat-Traj: A Scalable and Unified Dataset of Materials Trajectories for Atomistic Modeling
- 整合三大材料数据库,统一格式与元数据标准
- 覆盖120万+原子构型,含高力/高能结构,提升模型泛化能力
- 配套开源工具库,支持数据持续扩展与复现
机器学习势函数(MLIP)的开发受限于量子力学轨迹数据集的分散性与格式不一。这些数据生成成本高,且因格式、元数据和可访问性差异难以整合。为此,我们提出LeMat-Traj,一个包含超过120万原子构型的标准化数据集,源自Materials Project、Alexandria和OQMD等大规模资源。该数据集统一了不同密度泛函理论(DFT)泛函(PBE、PBESol、SCAN、r2SCAN)的结果表示,并筛选高质量构型。它涵盖弛豫低能态与高能高力结构,补充分子动力学与主动学习数据集。通过在高力预训练模型上微调LeMat-Traj,松弛任务中的力预测误差显著降低。我们还发布了LeMaterial-Fetcher——一个模块化开源库,支持社区轻松集成新数据源,推动大规模材料数据集持续演进。LeMat-Traj与LeMaterial-Fetcher已在Hugging Face与GitHub公开。
原文摘要 · Abstract (English)
The development of accurate machine learning interatomic potentials (MLIPs) is limited by the fragmented availability and inconsistent formatting of quantum mechanical trajectory datasets derived from Density Functional Theory (DFT). These datasets are expensive to generate yet difficult to combine due to variations in format, metadata, and accessibility. To address this, we introduce LeMat-Traj, a curated dataset comprising over 120 million atomic configurations aggregated from large-scale repositories, including the Materials Project, Alexandria, and OQMD. LeMat-Traj standardizes data representation, harmonizes results and filters for high-quality configurations across widely used DFT functionals (PBE, PBESol, SCAN, r2SCAN). It significantly lowers the barrier for training transferrable and accurate MLIPs. LeMat-Traj spans both relaxed low-energy states and high-energy, high-force structures, complementing molecular dynamics and active learning datasets. By fine-tuning models pre-trained on high-force data with LeMat-Traj, we achieve a significant reduction in force prediction errors on relaxation tasks. We also present LeMaterial-Fetcher, a modular and extensible open-source library developed for this work, designed to provide a reproducible framework for the community to easily incorporate new data sources and ensure the continued evolution of large-scale materials datasets. LeMat-Traj and LeMaterial-Fetcher are publicly available at https://huggingface.co/datasets/LeMaterial/LeMat-Traj and https://github.com/LeMaterial/lematerial-fetcher.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。