Deep learning approaches to recover the plasma current density profile from the safety factor based on Grad–Shafranov solutions across multiple tokamaks

Many magnetohydrodynamic stability analyses require generation of a set of equilibria with a fixed safety factor q-profile while varying other plasma parameters.A neural network(NN)-based approach is investigated that facilitates such a process.Both multilayer perceptron(MLP)-based NN and convolutional neural network(CNN)models are trained to map the q-profile to the plasma current density J-profile,and vice versa,while satisfying the Grad–Shafranov radial force balance constraint.When the initial target models are trained,using a database of semianalytically constructed numerical equilibria,an initial CNN with one convolutional layer is found to perform better than an initial MLP model.In particular,a trained initial CNN model can also predict the q-or J-profile for experimental tokamak equilibria.The performance of both initial target models is further improved by fine-tuning the training database,i.e.by adding realistic experimental equilibria with Gaussian noise.The fine-tuned target models,referred to as fine-tuned MLP and fine-tuned CNN,well reproduce the target q-or J-profile across multiple tokamak devices.As an important application,these NN-based equilibrium profile convertors can be utilized to provide a good initial guess for iterative equilibrium solvers,where the desired input quantity is the safety factor instead of the plasma current density.