TokaMark: A Comprehensive Benchmark for MAST Tokamak Plasma Models
Not on the map. Why?
Abstract (arXiv)
Development and operation of commercially viable fusion energy reactors such as tokamaks require accurate predictions of plasma dynamics from sparse, noisy, and incomplete sensors readings. The complexity of the underlying physics and the heterogeneity of experimental data pose formidable challenges for conventional numerical methods, and highlight the promise of modern data-native approaches. A major obstacle in realizing this potential is, however, the lack of curated, openly available datasets and standardized benchmarks. Existing fusion datasets are scarce, fragmented across institutions, facility-specific, and inconsistently annotated, which limits reproducibility and prevents a fair and scalable comparison of AI approaches. In this paper, we introduce TokaMark, a structured benchmark to evaluate AI models on real experimental data collected from the Mega Ampere Spherical Tokamak (MAST). TokaMark provides a comprehensive suite of tools designed to unify access to multi-modal fusion data and standardize evaluation protocols. The benchmark includes a curated list of 14 tasks spanning a range of physical mechanisms, exploiting a variety of diagnostics and covering multiple operational use cases. A baseline model is provided to facilitate transparent comparison and validation within a unified framework. By establishing a unified benchmark, TokaMark aims to accelerate progress in data-driven AI-based plasma modeling, contributing to the broader goal of achieving sustainable and stable fusion energy. The dataset, benchmark, documentation, and tooling are open-sourced under https://github.com/UKAEA-IBM-STFC-Fusion-FMs/tokamark_baseline.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 10 of its 51 references that are in the map. First posted on arXiv 5 Feb 2026.
Builds on
10 of 51 references.
References · 10
- 2020Hybrid deep learning architecture for general disruption prediction across tokamakscited by 17
- 2024Full Shot Predictions for the DIII-D Tokamak via Deep Recurrent Networkscited by 9
- 2019Deep Learning for Plasma Tomography and Disruption Prediction from Bolometer Datacited by 5
- 2021Scenario adaptive disruption prediction study for next generation burning-plasma tokamakscited by 5
- 2022Neural net modeling of equilibria in NSTX-Ucited by 5
- 2022A machine-learning-based tool for last closed-flux surface reconstruction on tokamakscited by 3
- 2024Active Disruption Avoidance and Trajectory Design for Tokamak Ramp-downs with Neural Differential Equations and Reinforcement Learningcited by 3
- 2025Learning Plasma Dynamics and Robust Rampdown Trajectories with Predict-First Experiments at TCVcited by 3
- 2020Experiment data-driven modeling of tokamak discharge in EASTcited by 2
- 2025FusionMAE: large-scale pretrained model to optimize and simplify diagnostic and control of fusion plasmacited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026TokaMind: A Multi-Modal Transformer Foundation Model for Tokamak Plasma Dynamicscited by 3
- 2026Benchmarking Sensor Robustness in Plasma Diagnostic Models: A Systematic Evaluation on TokaMarkcited by 0
- 2026FusionMMT: A Unified Multimodal and Multitask Learning Framework for Nuclear Fusioncited by 0
- 2026FusionMMT: A Unified Multimodal and Multitask Learning Framework for Nuclear Fusion6 Aug 2026
- 2026Benchmarking Sensor Robustness in Plasma Diagnostic Models: A Systematic Evaluation on TokaMark5 Jul 2026
- 2026TokaMind: A Multi-Modal Transformer Foundation Model for Tokamak Plasma Dynamics16 Feb 2026