Machine Learning Difference Charge Density

Xiwen Li, LiangLiang Hong, Yingwei Chen, Hongjun Xiang

Published: 2025/10/1

Abstract

In density functional theory (DFT), the ground state charge density is the fundamental variable which determines all other ground state properties. Many machine learning charge density models are developed by prior efforts, which have been proven useful to accelerate DFT calculations. Yet they all use the total charge density (TCD) as the training target. In this work, we advocate predicting difference charge density (DCD) instead. We term this simple technique by $\Delta$-SAED, which leverages the prior physical information of superposition of atomic electron densities (SAED). The robustness of $\Delta$-SAED is demonstrated through evaluations over diverse benchmark datasets, showing an extra accuracy gain for more than 90% structures in the test sets. Using a Si allotropy dataset, $\Delta$-SAED is demonstrated to advance model's transferability to chemical accuracy for non-self-consistent calculations. By incorporating physical priors to compensate for the limited expressive power of machine learning models, $\Delta$-SAED offers a cost-free yet robust approach to improving charge density prediction and enhancing non-self-consistent performance.