Federated Transfer Learning for Soalr Flare Forecasting

Junfeng Fu,Jie Wan,Zhongrui Zhao,Koeun Han,E Peng
DOI: https://doi.org/10.21203/rs.3.rs-3134736/v1
2023-01-01
Abstract:Abstract The critical role of solar flare forecasting in shielding space-based systems and essential infrastructures from potential hazards is universally acknowledged. Most research in solar flare forecasting predominantly utilizes machine learning techniques, primarily focusing on the development and optimization of a single, comprehensive model. This strategy, however, tends to overlook the intrinsic variability amongst individual sunspots located on the solar photosphere. Such sunspot variations, which are manifested in distinct flare progression patterns and other unique physical parameters, play a pivotal role in the forecasting process. Ignoring these nuances while training machine learning models can unintentionally compromise the precision of solar flare predictions. To mitigate this issue, we propose a methodology rooted in federated transfer learning. This approach commences with the segregation of the acquired dataset using NOAA numbers, thereby enabling the tailored training of models based on these subdivisions. Subsequently, these individually trained models are integrated into a unified ”global model”. The parameters of this global model are then redistributed to each sub-model, facilitating additional training in an iterative cycle. This iterative process culminates when the output deviations between the global and sub-models satisfy a predetermined boundary. Our proposed methodology has yielded a commendable Area Under the Curve (AUC) score of 93.6%, underscoring its superior predictive performance relative to traditional techniques. These results underscore the importance and accuracy of factoring in the regional differences of sunspots in solar flare forecasting, thereby validating the effectiveness of our federated transfer learning approach.
What problem does this paper attempt to address?