A Transformer-Based Framework for Geomagnetic Activity Prediction
Document Type
Conference Proceeding
Publication Date
1-1-2022
Abstract
Geomagnetic activities have a crucial impact on Earth, which can affect spacecraft and electrical power grids. Geospace scientists use a geomagnetic index, called the Kp index, to describe the overall level of geomagnetic activity. This index is an important indicator of disturbances in the Earth’s magnetic field and is used by the U.S. Space Weather Prediction Center as an alert and warning service for users who may be affected by the disturbances. Early and accurate prediction of the Kp index is essential for preparedness and disaster risk management. In this paper, we present a novel deep learning method, named KpNet, to perform short-term, 1–9 hour ahead, forecasting of the Kp index based on the solar wind parameters taken from the NASA Space Science Data Coordinated Archive. KpNet combines transformer encoder blocks with Bayesian inference, which is capable of quantifying both aleatoric uncertainty (data uncertainty) and epistemic uncertainty (model uncertainty) when making Kp predictions. Experimental results show that KpNet outperforms closely related machine learning methods in terms of the root mean square error and R-squared score. Furthermore, KpNet can provide both data and model uncertainty quantification results, which the existing methods cannot offer. To our knowledge, this is the first time that Bayesian transformers have been used for Kp prediction.
Identifier
85140441664 (Scopus)
ISBN
[9783031165634]
Publication Title
Lecture Notes in Computer Science Including Subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics
External Full Text Location
https://doi.org/10.1007/978-3-031-16564-1_31
e-ISSN
16113349
ISSN
03029743
First Page
325
Last Page
335
Volume
13515 LNAI
Grant
AGS–1927578
Fund Ref
National Science Foundation
Recommended Citation
Abduallah, Yasser; Wang, Jason T.L.; Xu, Chunhui; and Wang, Haimin, "A Transformer-Based Framework for Geomagnetic Activity Prediction" (2022). Faculty Publications. 3226.
https://digitalcommons.njit.edu/fac_pubs/3226