Predicting Magnetic Structures from Atoms Alone: A Neural Network Trained on Real Experiments

Paper: 2605.16230 Authors: Abhijatmedhi Chotrattanapituk, Ryotaro Okabe, Eunbi Rha, Mariya Al-Hinai, Eugene Jiang, Daniel Pajerowski, Yongqiang Cheng, Joshua J. Turner, Mingda Li Categories: cond-mat.mtrl-sci, cs.LG

The Gap

Determining magnetic structure is expensive and slow. Neutron scattering experiments require specialized facilities and weeks of beamtime. First-principles methods like density functional theory can predict simple ferromagnets, but struggle with the non-collinear and incommensurate orders found in real materials—spins that spiral, cant at angles, or have periodicities incommensurate with the crystal lattice. Existing ML approaches either assume collinear order or require symmetry constraints that break down for complex structures.

The field needed a method that could predict arbitrary magnetic orders—collinear, non-collinear, commensurate, incommensurate—directly from atomic positions, trained on what actually exists in nature rather than what theory can easily compute.

Problem: Magnetic structure unknown
   |
   v
Experiments: Slow, expensive, specialized facilities
DFT: Fails on non-collinear/incommensurate orders
Prior ML: Assumes collinearity or symmetry
   |
   v
Gap: No universal predictor for real magnetic complexity
   |
   v
Method: E(3)-equivariant GNN + modulated structure encoding
   |
   v
Evidence: Trained on MAGNDATA (experimental structures)
          Predicts all order types with high fidelity
   |
   v
Conclusion: Scalable prediction of arbitrary magnetic orders

The Increment

One sentence: Before this paper, predicting complex magnetic structures required either expensive experiments or symmetry assumptions that excluded most real materials; after, a single neural network predicts arbitrary magnetic orders from atomic coordinates with near-experimental accuracy.

Core Mechanism

The Magnetic Structure Network (MSN) is an E(3)-equivariant graph neural network that treats atoms as nodes and their interactions as edges. The key innovation is the Primitive Modulated Structure Representation (PMSR), which encodes magnetic moments as Fourier series in the primitive cell. For each magnetic atom, instead of predicting a single spin vector, the model predicts amplitudes and phases for multiple wave vectors, allowing it to represent both simple ferromagnets (one wave vector at zero) and complex spirals (multiple incommensurate wave vectors).

The network processes atomic positions through message-passing layers that respect rotational and translational symmetry. Each layer updates node features by aggregating information from neighbors, weighted by learned edge features that depend on interatomic distances and angles. After several layers, the model outputs Fourier coefficients for each magnetic site. These coefficients are then transformed back into real-space magnetic moments at each atomic position.

Training uses experimental structures from MAGNDATA, a curated database of magnetically ordered materials determined by neutron diffraction. The loss function compares predicted and experimental magnetic moments in real space, averaged over the unit cell. This direct supervision on experimental data means the model learns the actual distribution of magnetic orders found in nature, not just what theory predicts should exist.

Input: Atomic positions \{r_i\}
   |
   v
[Graph Construction]
   Nodes: atoms
   Edges: interatomic vectors
   |
   v
[E(3)-Equivariant Message Passing]
   h_i^(l+1) = Update(h_i^(l), \{h_j^(l), e_ij\})
   (respects rotations/translations)
   |
   v
[Fourier Coefficient Prediction]
   For each atom i, predict:
   A_k, phi_k for wave vectors k
   |
   v
[Real-Space Reconstruction]
   m_i(r) = sum_k A_k * exp(i*k*r + phi_k)
   |
   v
Output: Magnetic moment at each site

Think of PMSR like describing ocean waves. A simple wave has one frequency and amplitude—that’s a ferromagnet. But real ocean surfaces have multiple overlapping waves with different wavelengths and phases, some matching the underlying seafloor grid (commensurate) and some not (incommensurate). Instead of trying to draw the entire choppy surface, you record the amplitude and phase of each component wave. Given those Fourier components, you can reconstruct the full surface at any point. MSN does this for magnetic moments: it learns to predict the “wave components” of magnetic order, then reconstructs the full 3D spin texture. The E(3) equivariance is like ensuring your wave description doesn’t change if you rotate your coordinate system—the physics stays the same regardless of how you orient your axes.

Key Concepts

• E(3) Equivariance: A function is E(3)-equivariant if rotating or translating the input produces the same rotation or translation in the output. For magnetic structures, this means if you rotate the crystal, the predicted magnetic moments rotate by the same amount. This is crucial because physical laws don’t depend on your choice of coordinate system. Standard neural networks aren’t equivariant—they’d have to relearn the same physics for every possible orientation. E(3)-equivariant architectures bake this symmetry into the network structure, making learning more efficient and predictions more reliable. Imagine teaching someone to recognize a chair: you could show them millions of photos of chairs at every possible angle, or you could teach them what makes a chair a chair (seat, back, legs) and let them apply that understanding to any orientation. Equivariance is the second approach.

• Incommensurate Magnetic Order: In a crystal, atoms sit on a periodic lattice with some repeat distance. Commensurate magnetic order means the magnetic pattern repeats with the same period—if the crystal repeats every 5 atoms, so does the spin pattern. Incommensurate order means the magnetic period doesn’t match: maybe spins rotate by 137 degrees between neighbors, so the pattern never exactly repeats within the crystal lattice. This happens when competing magnetic interactions favor a wavelength that doesn’t fit neatly into the atomic grid. It’s like trying to tile a floor with tiles that are 1.618 times the width of the floorboards underneath—the pattern never lines up. Incommensurate orders are common in frustrated magnets and multiferroics, but they’re hard to represent in standard computational frameworks that assume periodicity.

• Primitive Modulated Structure Representation (PMSR): Traditional approaches represent magnetic structures in a supercell large enough to contain one full period of the magnetic order. For incommensurate structures, this supercell can be enormous or impossible to define. PMSR instead works in the primitive cell (the smallest repeating unit of the crystal) and represents magnetic moments as Fourier series: m(r) = sum over k of A_k ** exp(ikr + phi_k). Each wave vector k, amplitude A_k, and phase phi_k is a Fourier component. This representation naturally handles both commensurate orders (k vectors are reciprocal lattice vectors) and incommensurate orders (k vectors are arbitrary). It’s like the difference between storing a song as a massive waveform file versus storing it as sheet music with notes and frequencies—the second representation is compact and works for any melody, even ones with irrational frequency ratios.

Framework Shift

Before (DFT + symmetry):              After (MSN + PMSR):

[Crystal Structure]                   [Crystal Structure]
        |                                     |
        v                                     v
  Assume symmetry                       E(3)-equivariant GNN
  (collinear, commensurate)                   |
        |                                     v
        v                               Predict Fourier
  DFT energy minimization               coefficients (A_k, phi_k)
  (fails on complex orders)                   |
        |                                     v
        v                               Reconstruct any order
  [Limited order types]                 (collinear, non-collinear,
                                        commensurate, incommensurate)
                                              |
                                              v
                                        [Universal prediction]

From symmetry-constrained energy minimization to data-driven Fourier decomposition, the core shift is learning the distribution of real magnetic orders rather than computing idealized ground states.

Expert Assessment

Problem choice: This is a real gap. Magnetic structure determination is a genuine bottleneck in materials discovery, and the inability of DFT to handle non-collinear/incommensurate orders isn’t a minor limitation—it excludes a large fraction of interesting magnetic materials. The choice to train on experimental data (MAGNDATA) rather than DFT-computed structures is smart; it sidesteps the “garbage in, garbage out” problem that plagues many materials ML models.

Method maturity: The PMSR representation is genuinely clever—it’s not obvious that Fourier decomposition in the primitive cell would work for incommensurate structures, but it does. The E(3) equivariance is table stakes for this problem, not a novelty. The architecture itself (message-passing GNN) is standard; the innovation is in the representation and training data. One concern: the model predicts Fourier coefficients independently for each atom, which might miss global constraints (like the total magnetization or symmetry-enforced relationships between sites). A more sophisticated approach might predict coefficients jointly with explicit constraint satisfaction.

Experimental integrity: The baselines are reasonable but not exhaustive. Comparing to DFT is fair, but the paper doesn’t compare to other ML approaches for magnetic structure prediction (though admittedly, few exist that handle non-collinear orders). The train/test split is by material, not by structure type, which is good. The error metrics (MAE on magnetic moments) are appropriate. One red flag: the paper doesn’t discuss failure modes in detail. What kinds of structures does the model get wrong? Are there systematic biases? The high-level accuracy numbers look good, but without failure analysis, it’s hard to know where the method’s boundaries are.

Writing quality: The introduction is clear and motivates the problem well. The methods section is dense—PMSR is explained mathematically but could use more intuition-building before diving into equations. The results section focuses on aggregate metrics; more case studies of specific materials (especially failures) would strengthen the paper. The discussion is brief and doesn’t engage deeply with limitations or future directions. Rewriting the methods section to lead with intuition (the ocean waves analogy) before formalism would make the paper more accessible.

Verdict: weak accept — Solid contribution with a clever representation and real-world training data, but lacks depth in failure analysis and comparison to alternative ML approaches.

Takeaways

Fourier representation for periodic-but-not-quite data: PMSR’s trick of using Fourier series in the primitive cell to handle incommensurate patterns transfers to any domain where you have approximate periodicity with mismatched length scales. Think protein folding (secondary structure motifs that don’t align with sequence periodicity), time series with multiple incommensurate cycles (economic data with daily, weekly, and lunar cycles), or spatial patterns in ecology (animal territories that don’t tile neatly).

Train on experiments, not simulations: The decision to use MAGNDATA (experimental structures) rather than DFT-computed structures is a template for other materials ML work. Simulations are cheap and abundant, but they encode the biases of the simulation method. Experiments are sparse and noisy, but they reflect reality. If your goal is to predict real-world behavior, bite the bullet and curate experimental training data, even if it means a smaller dataset.

Equivariance as inductive bias: E(3) equivariance isn’t just a nice-to-have—it’s a way to encode domain knowledge (physical laws don’t depend on coordinate systems) that dramatically reduces the amount of training data needed. This principle applies beyond physics: if your problem has known symmetries (permutation invariance in sets, time-translation invariance in stationary processes), build them into the architecture rather than hoping the model learns them from data.