Density limit disruption prediction using a long short-term memory network on EAST

Disruption prediction using a long short-term memory (LSTM) algorithm has been developed on EAST, due to its inherent advantages in time series data processing. In the present work, LSTM is used as the model and the AUC (area under receiver operation characteristic curve) is used as the evaluation index. When the model is trained on data from the plasma current flattop phase and tested on data from the same period multiple times, the highest AUC is 0.8646 and the training time is about 6900 s per epoch. For comparison, the last 1000 ms of the flattop phases are intercepted as short time sequences. When the model is trained on data from short time sequences and tested on data from the same period, the highest AUC is increased to 0.9379 and the training time is restricted to 36 s per epoch. When the best model trained on the short time sequences is applied to the flattop phase for testing, the AUC is up to 0.9189. The experiment results show that it is possible for LSTM to train the model on data from short time sequences and migrate the model to the entire flattop phase, with a shorter training time and higher AUC value.