Universal Magnetic Structure Prediction from Atomic Coordinates with Near-Experimental Accuracy
Organizations: Quantum Measurement Group, MIT, Cambridge, MA 02139, USA · Department of Electrical Engineering and Computer Science, MIT, Cambridge, MA 02139, USA · Department of Chemistry, MIT, Cambridge, MA 02139, USA · Department of Nuclear Science and Engineering, MIT, Cambridge, MA 02139, USA · Department of Physics, MIT, Cambridge, MA 02139, USA · Neutron Scattering Division, Oak Ridge National Laboratory, Oak Ridge, TN 37831, USA · SLAC National Accelerator Laboratory, Menlo Park, CA 94025, USA
Abstract
Magnetic order is a fundamental property of materials, governing collective behavior and enabling a broad range of functionalities. Yet magnetic structure remains difficult to determine: experiments are costly and specialized, while first-principles methods often struggle with the noncollinear and incommensurate orders found in real materials. Here we introduce magnetic structure network (MSN), an E(3) equivariant graph neural network that predicts both collinear and non-collinear magnetic structures directly from atomic crystal structures, trained directly on experimentally determined structures from MAGNDATA. By proposing the primitive modulated structure representation (PMSR), we are able to encode commensurate and incommensurate structures in a unified way without symmetry assumptions. The model achieves strong performance across all modulation components and reconstructs experimental magnetic structures with high fidelity. Our approach provides a scalable framework for rapid magnetic structure prediction and opens a route to data-driven discovery of magnetic materials.