Small-supercell and Small-dataset Training Strategy of Machine Learning Interatomic Potentials for Point Defects

Abstract

Machine learning interatomic potentials (MLIPs) can treat large-scale material systems with near first-principles accuracy and have been widely used to accelerate point-defect simulations. However, the training of MLIPs usually relies on large amounts of DFT data. This issue is particularly pronounced for charged defects, for which DFT calculations of large supercells are required to avoid long-range Coulomb interactions and finite-size effects, making the construction of datasets computationally expensive. In this work, we propose an efficient MLIP training scheme for neutral and lowly charged point defects based on small supercells (less than 100 atoms) and limited number of DFT calculations. The scheme requires only four DFT structural relaxations to construct the training dataset and the trained MLIPs dedicated for the defect can predict the defect formation energies in larger supercells (over 200 atoms) with small errors (mostly smaller than 0.3 eV). Using defects in GaN, SiO$_2$, and $\mathrm{Cu}_2\mathrm{ZnSnS}_4$ as representative examples, we evaluate the extrapolation capability of this scheme for predicting defect total energies and structural relaxations across different supercell sizes. The results show that an MLIP trained only on single small defect supercell produces severe errors when treating large supercells. Incorporating defect data from multiple supercell sizes improves the predictive performance of the MLIP, while adding pristine defect-free bulk supercells further enhances the accuracy. These results provide practical guidance for training MLIP models of defect systems with low computational costs.