Gaussian process surrogate models for the properties of micro-tearing modes in spherical tokamaks
Authors:
William Hornsby,
Ander Gray,
James Buchanan,
Daniel Kenndy,
Bhavin Patel,
Francis Casson,
Colin Roach,
Mikkel Lykkegaard,
Huy Nguyen,
Nikolaos Papadimas,
Ben Fourcin,
Jordan Hart
Abstract:
Spherical tokamaks (STs) have many desirable features that make them a suitable choice for fusion power plants. To understand their confinement properties, accurate calculation of turbulent micro-instabilities is necessary for tokamak design. Presented is a novel surrogate model for Micro-tearing modes (MTMs), the micro-instability thought to be dominant in high beta STs. Direct numerical calculat…
▽ More
Spherical tokamaks (STs) have many desirable features that make them a suitable choice for fusion power plants. To understand their confinement properties, accurate calculation of turbulent micro-instabilities is necessary for tokamak design. Presented is a novel surrogate model for Micro-tearing modes (MTMs), the micro-instability thought to be dominant in high beta STs. Direct numerical calculation of micro-instabilities is computationally expensive and is a significant bottleneck in integrated plasma modelling. The considerable number of geometric and thermodynamic parameters, the interactions that influence these coefficients and the resolutions needed to accurately resolve these modes, makes direct numerical simulation for parameter space exploration computationally extremely challenging. However, this and the dearth of accurate reduced physics models for MTMs makes it suitable for surrogate modelling using Gaussian Process Regression, a modern machine learning technique. This paper outlines the further development of a data-driven reduced-order model across a spherical tokamak reactor-relevant parameter space utilising Gaussian Process Regression (GPR) and classification; techniques from machine learning. To build the original simple GP model these two components were used in an active learning loop to maximise the efficiency of data acquisition thus minimising computational cost. The `simple' GP was seen to show a plateau of fidelity with more data and to be under-confident, particular in areas of parameter space close to marginal stability. It is postulated that the presence of multiple sub-types of MTM could be the root cause, with the underlying function being less smooth than expected. An expansion of the model using clustering algorithms to find optimal sub models using a mixture of experts approach is shown to greatly improve the variances in the outputs of the GP model.
△ Less
Submitted 14 March, 2024;
originally announced March 2024.
Gaussian Process Regression models for the properties of micro-tearing modes in spherical tokamak
Authors:
William Hornsby,
Ander Gray,
James Buchanan,
Bhavin Patel,
Daniel Kennedy,
Francis Casson,
Colin Roach,
Mikkel Lykkegaard,
Huy Nguyen,
Nikolaos Papadimas,
Ben Fourcin,
Jordan Hart
Abstract:
Spherical tokamaks (STs) have many desirable features that make them an attractive choice for a future fusion power plant. Power plant viability is intrinsically related to plasma heat and particle confinement and this is often determined by the level of micro-instability driven turbulence. Accurate calculation of the properties of turbulent micro-instabilities is therefore critical for tokamak de…
▽ More
Spherical tokamaks (STs) have many desirable features that make them an attractive choice for a future fusion power plant. Power plant viability is intrinsically related to plasma heat and particle confinement and this is often determined by the level of micro-instability driven turbulence. Accurate calculation of the properties of turbulent micro-instabilities is therefore critical for tokamak design, however, the evaluation of these properties is computationally expensive. The considerable number of geometric and thermodynamic parameters and the high resolutions required to accurately resolve these instabilities makes repeated use of direct numerical simulations in integrated modelling workflows extremely computationally challenging and creates the need for fast, accurate, reduced-order models.
This paper outlines the development of a data-driven reduced-order model, often termed a {\it surrogate model} for the properties of micro-tearing modes (MTMs) across a spherical tokamak reactor-relevant parameter space utilising Gaussian Process Regression (GPR) and classification; techniques from machine learning. These two components are used in an active learning loop to maximise the efficiency of data acquisition thus minimising computational cost. The high-fidelity gyrokinetic code GS2 is used to calculate the linear properties of the MTMs: the mode growth rate, frequency and normalised electron heat flux; core components of a quasi-linear transport model. Five-fold cross-validation and direct validation on unseen data is used to ascertain the performance of the resulting surrogate models.
△ Less
Submitted 14 November, 2023; v1 submitted 18 September, 2023;
originally announced September 2023.