Fusing Global Diagnostic Signals and Spectral Data for Electron and Ion Temperature Profile Prediction on EAST via Attention-based Neural Networks

Abstract

Ion temperature Ti and electron temperature Te are fundamental kinetic parameters for evaluating tokamak plasma confinement, turbulent transport and magnetohydrodynamic stability. Reliable prediction of their spatiotemporal profiles is essential for long-pulse steady-state operation on the Experimental Advanced Superconducting Tokamak (EAST). However, existing EAST diagnostic systems have obvious measurement limitations, including narrow applicable conditions of charge exchange recombination spectroscopy and severe signal degradation of edge temperature diagnostics during long discharges. Classical physics-based transport codes suffer from high computational cost and error accumulation, while existing data-driven models only separately predict Ti or Te relying on spectral data, lacking joint Ti/Te modeling and failing under missing spectral measurements. In this work, we propose a multi-modal (i.e., global macroscopic diagnostic signals and spectral data) fusion attention neural network for simultaneous reconstruction of Ti/Te profiles on EAST. It leverages an adaptive attention module to weight global diagnostic features and suppress noise interference, employs convolutional layers to extract deep spectral representations, and utilizes multi-head self-attention to facilitate cross-modal feature interaction. Two separate prediction heads are constructed to output full radial distributions of Ti and Te. This work builds the first EAST-oriented joint Ti/Te prediction model compatible with operating scenarios without spectral diagnostics. Hierarchical attention mechanisms excavate nonlinear multi-modal correlations to realize high-performance dual-temperature profile prediction. The source code of this paper is available on https://github.com/Event-AHU/OpenFusion/tree/main/TiTe_Prediction