Fast modeling of turbulent transport in fusion plasmas using neural networks

TL;DR

This paper introduces QLKNN, a neural network surrogate model that predicts turbulent transport in tokamak plasmas with high speed and accuracy, enabling real-time applications in fusion plasma modeling.

Contribution

The paper presents a novel neural network model trained on extensive flux data, integrated into existing tokamak simulation frameworks for rapid and accurate transport predictions.

Findings

QLKNN predicts fluxes 3-5 orders faster than traditional models.

Profiles predicted by QLKNN closely match those from QuaLiKiz with 1%-15% discrepancy.

Dynamic plasma behavior is accurately captured with 4%-10% difference.

Abstract

We present an ultrafast neural network (NN) model, QLKNN, which predicts core tokamak transport heat and particle fluxes. QLKNN is a surrogate model based on a database of 300 million flux calculations of the quasilinear gyrokinetic transport model QuaLiKiz. The database covers a wide range of realistic tokamak core parameters. Physical features such as the existence of a critical gradient for the onset of turbulent transport were integrated into the neural network training methodology. We have coupled QLKNN to the tokamak modelling framework JINTRAC and rapid control-oriented tokamak transport solver RAPTOR. The coupled frameworks are demonstrated and validated through application to three JET shots covering a representative spread of H-mode operating space, predicting turbulent transport of energy and particles in the plasma core. JINTRAC-QLKNN and RAPTOR-QLKNN are able to accurately reproduce JINTRAC-QuaLiKiz T i,e and n e profiles, but 3 to 5 orders of magnitude faster. Simulations which take hours are reduced down to only a few tens of seconds. The discrepancy in the final source-driven predicted profiles between QLKNN and QuaLiKiz is on the order 1%-15%. Also the dynamic behaviour was well captured by QLKNN, with differences of only 4%-10% compared to JINTRAC-QuaLiKiz observed at mid-radius, for a study of density buildup following the L-H transition. Deployment of neural network surrogate models in multi-physics integrated tokamak modelling is a promising route towards enabling accurate and fast tokamak scenario optimization, Uncertainty Quantification, and control applications.