EuroPED-NN: uncertainty aware surrogate model

A. Panera Alvarez*, A. Ho, Aaro Jarvinen, S. Saarelma, S. Wiesen, JET Contributors, ASDEX Upgrade Team

*Corresponding author for this work

Research output: Contribution to journalArticleScientificpeer-review

1 Citation (Scopus)

Abstract

This work successfully generates an uncertainty-aware surrogate model of the EuroPED plasma pedestal model using the Bayesian neural network with noise contrastive prior (BNN-NCP) technique. This model is trained using data from the JET-ILW pedestal database and subsequent model evaluations, conforming to EuroPED-NN. The BNN-NCP technique has been proven to be a suitable method for generating uncertainty-aware surrogate models. It matches the output results of a regular neural network while providing confidence estimates for predictions as uncertainties. Additionally, it highlights out-of-distribution regions using surrogate model uncertainties. This provides critical insights into model robustness and reliability. EuroPED-NN has been physically validated, first, analyzing electron density n e ( ψ pol = 0.94 ) with respect to increasing plasma current, I p , and second, validating the Δ − β p , ped relation associated with the EuroPED model. This affirms the robustness of the underlying physics learned by the surrogate model. On top of that, the method was used to develop a EuroPED-like model fed with experimental data, i.e. an uncertainty aware experimental model, which is functional in JET database. Both models have been also tested in ∼50 AUG shots.

Original languageEnglish
Article number095012
JournalPlasma Physics and Controlled Fusion
Volume66
Issue number9
DOIs
Publication statusPublished - Sept 2024
MoE publication typeA1 Journal article-refereed

Keywords

  • artificial intelligence
  • Bayesian neural network
  • machine learning
  • nuclear fusion
  • pedestal
  • uncertainty

Fingerprint

Dive into the research topics of 'EuroPED-NN: uncertainty aware surrogate model'. Together they form a unique fingerprint.

Cite this