cs.LGFeb 18, 2026

Prediction of Major Solar Flares Using Interpretable Class-dependent Reward Framework with Active Region Magnetograms and Domain Knowledge

Authors: Zixian Wu, Xuebao Li, Yanfang Zheng, Rui Wang, Shunhuang Zhang, Jinfang Wei, Yongshang Lv, Liang Dong, 

Organizations: School of Computer Science, Jiangsu University of Science and Technology, Zhenjiang 212100, People′s Republic of China · State Key Laboratory of Space Weather, Chinese Academy of Sciences, Beijing 100190, People′s Republic of China · School of software, Southeast University, Nanjing, People′s Republic of China · Yunnan Astronomical Observatory, Chinese Academy of Sciences, Kunming 650216, People′s Republic of China · 

Abstract

In this work, we develop, for the first time, a supervised classification framework with class-dependent rewards (CDR) to predict ≥\geqMM flares within 24 hr. We construct multiple datasets, covering knowledge-informed features and line-of sight (LOS) magnetograms. We also apply three deep learning models (CNN, CNN-BiLSTM, and Transformer) and three CDR counterparts (CDR-CNN, CDR-CNN-BiLSTM, and CDR-Transformer). First, we analyze the importance of LOS magnetic field parameters with the Transformer, then compare its performance using LOS-only, vector-only, and combined magnetic field parameters. Second, we compare flare prediction performance based on CDR models versus deep learning counterparts. Third, we perform sensitivity analysis on reward engineering for CDR models. Fourth, we use the SHAP method for model interpretability. Finally, we conduct performance comparison between our models and NASA/CCMC. The main findings are: (1)Among LOS feature combinations, R_VALUE and AREA_ACR consistently yield the best results. (2)Transformer achieves better performance with combined LOS and vector magnetic field data than with either alone. (3)Models using knowledge-informed features outperform those using magnetograms. (4)While CNN and CNN-BiLSTM outperform their CDR counterparts on magnetograms, CDR-Transformer is slightly superior to its deep learning counterpart when using knowledge-informed features. Among all models, CDR-Transformer achieves the best performance. (5)The predictive performance of the CDR models is not overly sensitive to the reward choices.(6)Through SHAP analysis, the CDR model tends to regard TOTUSJH as more important, while the Transformer tends to prioritize R_VALUE more.(7)Under identical prediction time and active region (AR) number, the CDR-Transformer shows superior predictive capabilities compared to NASA/CCMC.

Figures & tables

Explore similar work

CardsList
  1. Predicting Associations between Solar Flares and Coronal Mass Ejections Using SDO/HMI Magnetograms and a Hybrid Neural Network

    Apr 11, 2026Jialiang Li, Vasyl Yurchyshyn, Jason T. L. Wang +6Solar Physics

  2. Prediction of Solar Flares Using Photospheric Magnetic Field Parameters with Deep Learning

    Jun 20, 2026Yash Chaudhary, Jason T. L. Wang, Chunhui Xu +2Solar Physics

  3. A Deep Learning Framework for Predicting Solar EUV Irradiance During Significant Flares

    Jul 21, 2026Sathvik Soman, Jason T. L. Wang, Haimin Wang +1Solar Irradiance ForecastingTime Series Forecasting