Search papers, labs, and topics across Lattice.
This paper introduces an AI surrogate framework for real-time plasma equilibrium prediction in tokamaks, benchmarking five neural architectures (MLP, CNN, FNO, Transformer, and KAN) against a large numerical Grad-Shafranov (GS) database. The results show that while the Transformer excels in IID accuracy, the CNN architecture strikes the best balance between accuracy, robustness, and inference speed, achieving a latency of 0.7 ms. Additionally, the study highlights the trade-off between model capacity and out-of-distribution (OOD) generalization, providing valuable insights for selecting models for fusion applications.
CNNs outperform larger models in real-time plasma equilibrium prediction, achieving a remarkable balance of speed and accuracy essential for tokamak control.
Fast and reliable plasma equilibrium prediction is essential for real-time tokamak operation and control, but conventional Grad-Shafranov (GS) solvers are often too costly for real-time deployment. We develop an AI surrogate framework and benchmark five architectures (MLP, CNN, FNO, Transformer, and KAN) on a numerical GS database with 100,000 IID and 10,000 OOD samples. Under a unified protocol, we evaluate accuracy, inference efficiency, model scaling, and robustness. We also establish device-level validation on the EXL-50U tokamak by linking numerical GS solutions, surrogate predictions, and the standard Shape Editor reference to assess simulation-to-device consistency. The surrogates achieve errors of $10^{-3}$-$10^{-2}$ relative to GS solutions, while the GS-to-device discrepancy remains at $10^{-3}$. Transformer gives the best IID accuracy, whereas CNN offers the best balance of accuracy, robustness, and speed, reaching 0.7 ms TensorRT latency. On unseen plasma geometries and parameter regimes, CNN and FNO show the strongest extrapolation stability, with 4%-5% relative $L_2$ error, while models with weaker inductive biases degrade more substantially. Scaling data and model capacity improves interpolation but not necessarily extrapolation, revealing a trade-off between capacity and OOD generalization. Overall, this work provides a systematic, device-consistent benchmark for AI-based GS prediction and practical guidance for selecting reliable surrogates for real-time plasma control and fusion applications.
