NSGT: A Plasma Position Prediction Method for the HL-2A Tokamak Device Based on Transformer

In the discharge experiments of the HL-2A Tokamak device, plasma position prediction plays a crucial role in aspects such as the closed - loop control of HL-2A and verifying the correctness of the plasma controller design. In previous studies, methods based on traditional equations and LSTM have made significant progress in plasma position prediction. However, plasma position prediction is of extremely high complexity, and these methods have difficulty in achieving further improvement on this issue. To address the problem that the performance of traditional methods is hard to enhance further, we propose a Non - Stationary - Gated Linear Unit Transformer (NSGT) model integrating multi - source data processing. In this model, we select data from historical discharge experiments as the input of the NSGT model, and use the improved Transformer structure to model the global features of the multi - source fused data. The experimental results show that the NSGT model has a significant reduction in the root - mean - square error of prediction compared with the LSTM method, effectively supporting the real - time adjustment of the closed - loop control system. This method not only improves the prediction accuracy but also provides important technical support for achieving more stable and safe operation of the Tokamak device.