Search

We propose a non-collinear spin-constrained method that generates training data for deep-learning-based magnetic model, which provides a powerful tool for studying complex magnetic phenomena at the atomic scale. First, we propose a projection method for atomic magnetic moments by applying a radial truncation to the numerical atomic orbitals. We then implement a Lagrange multiplier method that can yield the magnetic torques of atoms by constraining the magnitude and direction of atomic magnetic moments. The method is implemented in ABACUS with both plane wave basis and numerical atomic orbital basis. We benchmark the iron (Fe) systems with the new method and analyze differences from calculations with the plane wave basis and numerical atomic orbitals basis in describing magnetic energy barriers. Based on more than 30,000 first-principles data with the information of magnetic torque, we train a deep-learning-based magnetic model DeePSPIN for the Fe system. By utilizing the model in large-scale molecular dynamics simulations, we successfully predict Curie temperatures of $\alpha$-Fe close to experimental values.

ABACUS (Atomic-orbital Based Ab-initio Computation at USTC) is an open-source software for first-principles electronic structure calculations and molecular dynamics simulations. It mainly features density functional theory (DFT) and is compatible with both plane-wave basis sets and numerical atomic orbital basis sets. ABACUS serves as a platform that facilitates the integration of various electronic structure methods, such as Kohn-Sham DFT, stochastic DFT, orbital-free DFT, and real-time time-dependent DFT, etc. In addition, with the aid of high-performance computing, ABACUS is designed to perform efficiently and provide massive amounts of first-principles data for generating general-purpose machine learning potentials, such as DPA models. Furthermore, ABACUS serves as an electronic structure platform that interfaces with several AI-assisted algorithms and packages, such as DeePKS-kit, DeePMD, DP-GEN, DeepH, DeePTB, etc.