Bayesian neural networks for predicting tokamak energy confinement time with uncertainty quantification

Accurate estimation of the tokamak energy confinement time ( τ _E ) is crucial for optimizing the operation and design of fusion devices. Traditional methods, such as the ITER scaling law, often lack predictive precision and provide limited uncertainty quantification. To address these limitations, w...

Full description

Saved in:
Bibliographic Details
Main Authors: Enliang Gao, Chenguang Wan, Youjun Hu, Minglong Wang, Jingjing Lu, Zhisong Qu, Xinghao Wen, Jia Huang, Ying Chen, Heru Guo, Zhengping Luo, Zhi Yu, Xiaojuan Liu, Qiping Yuan, Jiangang Li
Format: Article
Language:English
Published: IOP Publishing 2025-01-01
Series:Nuclear Fusion
Subjects:
Online Access:https://doi.org/10.1088/1741-4326/ade8fd
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Accurate estimation of the tokamak energy confinement time ( τ _E ) is crucial for optimizing the operation and design of fusion devices. Traditional methods, such as the ITER scaling law, often lack predictive precision and provide limited uncertainty quantification. To address these limitations, we develop two Bayesian neural network (BNN) models: VIBNN and NUTSBNN. These models leverage variational inference (VI) and the No-U-Turn Sampler (NUTS), respectively, for posterior inference. Evaluations on the multi-machine ITPA global H-mode confinement database and individual datasets from JET, DIII-D, and ASDEX-Upgrade, both models outperform conventional methods in terms of accuracy and provide reliable uncertainty estimates. NUTSBNN achieves higher accuracy with an inference time of ∼3 s, making it suitable for offline analysis, while VIBNN provides faster inference ( ${\sim}0.05$ s), which is promising for real-time applications. These results highlight the potential of BNNs for accurate and interpretable confinement time predictions in fusion research.
ISSN:0029-5515