Data-driven approach to model the influence of magnetic geometry in the confinement of fusion devices

Abstract Designing a fusion energy device to produce as much energy as possible requires balancing numerous physics constraints and metrics. In stellarators, the geometry is very flexible and involves a large number of free parameters. These can be optimized to achieve good performance. One of the main optimization targets is omnigenity, that is, the confinement of alpha particles stemming from the fusion reactions. In this work, two classes of omnigenous stellarators are studied, namely quasisymmetric and quasi-isodynamic stellarators. The goal is to determine the influence of the geometry on omnigenity, which can lead to greater insight into the design space of stellarators. For this purpose, a database of stellarator configurations with two field periods and a single dominant mode is created and analyzed for correlations, pair-wise distributions, and dimensionality reduction using a supervised autoencoder framework. Then, a classification model is trained on this database to predict the convergence of numerical solvers. Finally, two regression models, LightGBM and its probabilistic version, LightGBM LSS, as well as a feed-forward neural network, are trained to predict quasisymmetry and quasi-isodynamiticity and find the design parameters that most influence omnigenity.