TY - JOUR
T1 - EuroPED-NN
T2 - uncertainty aware surrogate model
AU - Panera Alvarez, A.
AU - Ho, A.
AU - Jarvinen, A.
AU - Saarelma, S.
AU - Wiesen, S.
AU - JET Contributors
AU - ASDEX Upgrade Team
N1 - Publisher Copyright:
© 2024 The Author(s). Published by IOP Publishing Ltd.
PY - 2024/9
Y1 - 2024/9
N2 - 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.
AB - 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.
KW - artificial intelligence
KW - Bayesian neural network
KW - machine learning
KW - nuclear fusion
KW - pedestal
KW - uncertainty
UR - http://www.scopus.com/inward/record.url?scp=85201240134&partnerID=8YFLogxK
U2 - 10.1088/1361-6587/ad6707
DO - 10.1088/1361-6587/ad6707
M3 - Article
AN - SCOPUS:85201240134
SN - 0741-3335
VL - 66
JO - Plasma Physics and Controlled Fusion
JF - Plasma Physics and Controlled Fusion
IS - 9
M1 - 095012
ER -