EuroPED-NN: Uncertainty aware surrogate model

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Alvarez, A. Panera, Ho, A., Jarvinen, A., Saarelma, S., Wiesen, S., Contributors, JET, team, the AUG
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929480145043456
author Alvarez, A. Panera
Ho, A.
Jarvinen, A.
Saarelma, S.
Wiesen, S.
Contributors, JET
team, the AUG
author_facet Alvarez, A. Panera
Ho, A.
Jarvinen, A.
Saarelma, S.
Wiesen, S.
Contributors, JET
team, the AUG
contents 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 (OOD) 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\!\left(ψ_{\text{pol}}=0.94\right)$ 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 $\sim 50$ AUG shots.
format Preprint
id arxiv_https___arxiv_org_abs_2402_00760
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle EuroPED-NN: Uncertainty aware surrogate model
Alvarez, A. Panera
Ho, A.
Jarvinen, A.
Saarelma, S.
Wiesen, S.
Contributors, JET
team, the AUG
Plasma Physics
Machine Learning
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 (OOD) 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\!\left(ψ_{\text{pol}}=0.94\right)$ 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 $\sim 50$ AUG shots.
title EuroPED-NN: Uncertainty aware surrogate model
topic Plasma Physics
Machine Learning
url https://arxiv.org/abs/2402.00760