Plasma Surrogate Modelling using Fourier Neural Operators

Abstract

Predicting plasma evolution within a Tokamak reactor is crucial to realizing the goal of sustainable fusion. Capabilities in forecasting the spatio-temporal evolution of plasma rapidly and accurately allow us to quickly iterate over design and control strategies on current Tokamak devices and future reactors. Modelling plasma evolution using numerical solvers is often expensive, consuming many hours on supercomputers, and hence, we need alternative inexpensive surrogate models. We demonstrate accurate predictions of plasma evolution both in simulation and experimental domains using deep learning-based surrogate modelling tools, viz., Fourier Neural Operators (FNO). We show that FNO has a speedup of six orders of magnitude over traditional solvers in predicting the plasma dynamics simulated from magnetohydrodynamic models, while maintaining a high accuracy (MSE in the normalised domain $\approx$ $10^{-5}$). Our modified version of the FNO is capable of solving multi-variable Partial Differential Equations (PDE), and can capture the dependence among the different variables in a single model. FNOs can also predict plasma evolution on real-world experimental data observed by the cameras positioned within the MAST Tokamak, i.e., cameras looking across the central solenoid and the divertor in the Tokamak. We show that FNOs are able to accurately forecast the evolution of plasma and have the potential to be deployed for real-time monitoring. We also illustrate their capability in forecasting the plasma shape, the locations of interactions of the plasma with the central solenoid and the divertor for the full (available) duration of the plasma shot within MAST. The FNO offers a viable alternative for surrogate modelling as it is quick to train and infer, and requires fewer data points, while being able to do zero-shot super-resolution and getting high-fidelity solutions.

References in corpus (16)

• Association of genomic subtypes of lower-grade gliomas with shape features automatically extracted by a deep learning algorithm
• The JOREK non-linear extended MHD code and applications to large-scale instabilities and their control in magnetically confined fusion plasmas
• Simulation Intelligence: Towards a New Generation of Scientific Methods
• Message Passing Neural PDE Solvers
• Applications of physics informed neural operators
• Neural network surrogate of QuaLiKiz using JET experimental data to populate training space
• Fast Dynamic 1D Simulation of Divertor Plasmas with Neural PDE Surrogates
• Nonlocality and Nonlinearity Implies Universality in Operator Learning
• Representation Equivalent Neural Operators: a Framework for Alias-free Operator Learning
• Fourier Neural Operator for Plasma Modelling
• Autoregressive Models: What Are They Good For?
• Multi-Grid Tensorized Fourier Neural Operator for High-Resolution PDEs
• Fast Regression of the Tritium Breeding Ratio in Fusion Reactors
• Incremental Spatial and Spectral Learning of Neural Operators for Solving Large-Scale PDEs
• Fourier-RNNs for Modelling Noisy Physics Data
• Autoregressive Renaissance in Neural PDE Solvers

Cited by in corpus (7)

• Accelerating the convergence of Newton's method for nonlinear elliptic PDEs using Fourier neural operators
• Uncertainty Quantification of Surrogate Models using Conformal Prediction
• Neural-Parareal: Dynamically Training Neural Operators as Coarse Solvers for Time-Parallelisation of Fusion MHD Simulations
• Fast prediction of plasma instabilities with sparse-grid-accelerated optimized dynamic mode decomposition
• Hybrid Fourier Neural Operator-Plasma Fluid Model for Fast and Accurate Multiscale Simulations of High Power Microwave Breakdown
• Modeling turbulent and self-gravitating fluids with Fourier neural operators
• Challenges and opportunities for AI to help deliver fusion energy