Disruption prediction and model analysis using LightGBM on J-TEXT and HL-2A
Yi Zhong, Weijie Zheng, Zhongyong Chen, Fan Xia, Liming Yu, Qiqi Wu, Xinpei Ai, Chengshuo Shen, Zhiyou Yang, W. Yan, Yonghua Ding, Yingqin Liang, Zhipeng Chen, Ruihai Tong, W Bai, Jiarong Fang, Fang Li, J-TEXT team
Abstract
Abstract Using machine learning (ML) techniques to develop disruption predictors is an effective way to avoid or mitigate the disruption in a large-scale tokamak. The recent ML-based disruption predictors have made great progress regarding accuracy, but most of them have not achieved acceptable cross-machine performance. Before we develop a cross-machine predictor, it is very important to investigate the method of developing a cross-tokamak ML-based disruption prediction model. To ascertain the elements which impact the model’s performance and achieve a deep understanding of the predictor, multiple models are trained using data from two different tokamaks, J-TEXT and HL-2A, based on an implementation of the gradient-boosted decision trees algorithm called LightGBM, which can provide detailed information about the model and input features. The predictor models are not only built and tested for performance, but also analyzed from a feature importance perspective as well as for model performance variation. The relative feature importance ranking of two tokamaks is caused by differences in disruption types between different tokamaks. The result of two models with seven inputs showed that common diagnostics is very important in building a cross-machine predictor. This provided a strategy for selecting diagnostics and shots data for developing cross-machine predictors.