DisruptionBench and Complimentary New Models: Two Advancements in Machine Learning Driven Disruption Prediction
Name
10894_2025_Article_495.pdf
Size
5.9 MB
Format
Adobe PDF
Checksum (MD5)
f5d347fb1471ac6e6467bef7f70f94ae
Author(s)
Spangher, Lucas
Bonotto, Matteo
Arnold, William
Chayapathy, Dhruva
Gallingani, Tommaso
Spangher, Alexander
Cannarile, Francesco
Bigoni, Daniele
de Marchi, Eliana
Rea, Cristina
Date Issued
May 24, 2025
Journal
Journal of Fusion Energy
Publisher
Springer US
Citation
Spangher, L., Bonotto, M., Arnold, W. et al. DisruptionBench and Complimentary New Models: Two Advancements in Machine Learning Driven Disruption Prediction. J Fusion Energ 44, 26 (2025).
Version
Final published version
Abstract
Plasma disruptions remain a major obstacle to sustained commercial operation of tokamak-based fusion devices. Although machine learning (ML) methods have shown promise for predicting disruptions, their performance and generalizability suffer from a lack of common benchmarks and comprehensive multi-device evaluations. To address this, we present DisruptionBench, a new benchmarking platform designed to standardize how ML-driven disruption prediction systems are trained and evaluated on multi-machine data. DisruptionBench spans three devices - Alcator C-Mod, DIII-D, and EAST - and includes tasks of varying difficulty: zero-shot, few-shot, and many-shot training regimes to assess each model’s ability to transfer learned representations to new or data-limited machines. We evaluate four state-of-the-art ML architectures. Two are re-implementations of notable prior work: a random forest (Cristina Rea in PPCF 60:084008, 2018) and the Hybrid Deep Learner (HDL) (Zhu in NC 61: 026607, 2020). We also propose two new approaches tailored for disruption prediction: a transformer-based model inspired by GPT-2, capable of learning long-range temporal dependencies through self-attention, and a Continuous Convolutional Neural Network (CCNN) that leverages continuous kernels to capture subtle variations in plasma signals. Across the nine benchmarking tasks, the CCNN demonstrates consistently strong performance and achieves the highest overall Area Under the ROC Curve (AUC) in intra-machine tests (up to 0.97 on C-Mod). Nevertheless, the GPT-2-based approach and HDL can outperform CCNN in specific transfer scenarios, particularly when the test machine is underrepresented in training data. We further analyze the significance of memory length in capturing precursor phenomena, providing evidence that longer context windows can boost predictive accuracy.
MIT Department
Massachusetts Institute of Technology. Plasma Science and Fusion Center
Terms of Use
Creative Commons Attribution
Persistent DSpace Link
DOI of Published Version
https://doi.org/10.1007/s10894-025-00495-2
