ARENA: Active Region attENtion Aggregation for interpretable solar flare prediction
Organizations: MPI for Intelligent Systems, Tübingen · University of Tübingen · MPI for Gravitational Physics, Potsdam · Instituto de Astrofísica de Canarias ·
Abstract
Predicting solar flares is a challenging space-weather forecasting problem with potentially significant impacts on Earth's upper atmosphere, including disruptions to radio communications. We propose Active Region attENtion Aggregation (ARENA), a neural network architecture that predicts solar flares from a variable number of active-region cutouts while learning to identify the region most relevant to the prediction. ARENA is trained using multiple-instance learning under weak supervision, requiring only a flare/no-flare label for the observation and no labels specifying the flare origin. The learned attention mechanism therefore provides a means of localizing the active region driving the prediction without explicit origin annotations. ARENA achieves a heidke skill score of 0.45 which is comparable to other flare prediction approaches. These results demonstrate that active-region-based aggregation can provide an efficient and interpretable approach to solar flare prediction. The code is available at https://github.com/LuisBauer22/ARENA.
Figures & tables
| TSS | HSS | F1 | Precision | TPR | TNR | |
|---|---|---|---|---|---|---|
| Per AR (instance level) | 0.85 | 0.17 | 0.19 | 0.10 | 0.96 | 0.90 |
| ARENA (bag level) | 0.79 | 0.33 | 0.37 | 0.23 | 0.92 | 0.88 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | TSS | HSS | F1 | Precision | TPR | TNR |
|---|---|---|---|---|---|---|
| Validation | ||||||
| Test 2017–2024 | ||||||
| Test 2017 | ||||||
| AR subset 2017 |
| Training set | TSS | HSS | F1 | Precision | TPR | TNR | |
|---|---|---|---|---|---|---|---|
| P-CNN (ensemble) [ 10 ] | 2010–2019 | 0.58 | 0.42 | 0.53 | 0.39 | 0.82 | 0.76 |
| Full disk (ensemble) | 2010–2017 | 0.52 | 0.34 | 0.49 | 0.34 | 0.81 | 0.71 |
| Fine-tuned Surya [ 24 ] | 2010–2017 | 0.48 | 0.46 | 0.55 | 0.54 | 0.57 | 0.90 |
| ARENA (ensemble) | 2010–2017 | 0.51 | 0.30 | 0.46 | 0.31 | 0.87 | 0.64 |
Attention Mechanisms
Attention mechanisms let neural networks weight information from other tokens or regions according to relevance, enabling context-dependent processing.
See all 517 papers