Solar Wind Speed Prediction via Graph Attention Network

Abstract The solar wind is a plasma flow formed by the expansion of the high‐temperature corona and propagates in the interplanetary space with speeds between 200 km/s and 900 km/s. Accurate solar wind speed prediction and longer lead time will help mitigate the impact of solar storms on aerospace e...

Full description

Saved in:
Bibliographic Details
Main Authors: Yanru Sun, Zongxia Xie, Haocheng Wang, Xin Huang, Qinghua Hu
Format: Article
Language:English
Published: Wiley 2022-07-01
Series:Space Weather
Subjects:
Online Access:https://doi.org/10.1029/2022SW003128
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841536459050319872
author Yanru Sun
Zongxia Xie
Haocheng Wang
Xin Huang
Qinghua Hu
author_facet Yanru Sun
Zongxia Xie
Haocheng Wang
Xin Huang
Qinghua Hu
author_sort Yanru Sun
collection DOAJ
description Abstract The solar wind is a plasma flow formed by the expansion of the high‐temperature corona and propagates in the interplanetary space with speeds between 200 km/s and 900 km/s. Accurate solar wind speed prediction and longer lead time will help mitigate the impact of solar storms on aerospace equipment and the Earth's magnetic field. Recently, most approaches do not explicitly capture the relationships between different solar wind features, and the prediction accuracy of 96‐hr is still not good enough. This paper elaborately designs an end‐to‐end model: Graph‐Temporal‐AR model (GTA) for solar wind speed prediction. First, our framework considers each feature as the node to construct the graph structure and adopts the graph attention module to learn the complex dependencies among features. Second, our approach employs the dilated causal convolution to extend the receptive field and prolong the prediction time. Furthermore, we leverage the autoregressive model to solve the scale insensitive problem of the neural network, making our model more robust. Specifically, we combine the OMNI data measured at Lagrangian Point 1 (L1) with the extreme ultraviolet images observed by the Solar Dynamics Observatory satellite to predict the solar wind speed at L1. Compared with the baseline models, GTA obtains significant performance improvements. Through visualization, we find GTA excavates the relationships between multiply variables without domain prior knowledge, which may help us find other unknown associations in heliophysics data sets. The data and code are available from https://github.com/syrGitHub/GTA.
format Article
id doaj-art-0e46ff5b786d4f6c99a40b49d8b8af3e
institution Kabale University
issn 1542-7390
language English
publishDate 2022-07-01
publisher Wiley
record_format Article
series Space Weather
spelling doaj-art-0e46ff5b786d4f6c99a40b49d8b8af3e2025-01-14T16:26:59ZengWileySpace Weather1542-73902022-07-01207n/an/a10.1029/2022SW003128Solar Wind Speed Prediction via Graph Attention NetworkYanru Sun0Zongxia Xie1Haocheng Wang2Xin Huang3Qinghua Hu4College of Intelligence and Computing Tianjin University Tianjin ChinaCollege of Intelligence and Computing Tianjin University Tianjin ChinaCollege of Intelligence and Computing Tianjin University Tianjin ChinaNational Astronomical Observatories Chinese Academy of Sciences Beijing ChinaCollege of Intelligence and Computing Tianjin University Tianjin ChinaAbstract The solar wind is a plasma flow formed by the expansion of the high‐temperature corona and propagates in the interplanetary space with speeds between 200 km/s and 900 km/s. Accurate solar wind speed prediction and longer lead time will help mitigate the impact of solar storms on aerospace equipment and the Earth's magnetic field. Recently, most approaches do not explicitly capture the relationships between different solar wind features, and the prediction accuracy of 96‐hr is still not good enough. This paper elaborately designs an end‐to‐end model: Graph‐Temporal‐AR model (GTA) for solar wind speed prediction. First, our framework considers each feature as the node to construct the graph structure and adopts the graph attention module to learn the complex dependencies among features. Second, our approach employs the dilated causal convolution to extend the receptive field and prolong the prediction time. Furthermore, we leverage the autoregressive model to solve the scale insensitive problem of the neural network, making our model more robust. Specifically, we combine the OMNI data measured at Lagrangian Point 1 (L1) with the extreme ultraviolet images observed by the Solar Dynamics Observatory satellite to predict the solar wind speed at L1. Compared with the baseline models, GTA obtains significant performance improvements. Through visualization, we find GTA excavates the relationships between multiply variables without domain prior knowledge, which may help us find other unknown associations in heliophysics data sets. The data and code are available from https://github.com/syrGitHub/GTA.https://doi.org/10.1029/2022SW003128solar wind speed predictiondeep learninggraph attention network
spellingShingle Yanru Sun
Zongxia Xie
Haocheng Wang
Xin Huang
Qinghua Hu
Solar Wind Speed Prediction via Graph Attention Network
Space Weather
solar wind speed prediction
deep learning
graph attention network
title Solar Wind Speed Prediction via Graph Attention Network
title_full Solar Wind Speed Prediction via Graph Attention Network
title_fullStr Solar Wind Speed Prediction via Graph Attention Network
title_full_unstemmed Solar Wind Speed Prediction via Graph Attention Network
title_short Solar Wind Speed Prediction via Graph Attention Network
title_sort solar wind speed prediction via graph attention network
topic solar wind speed prediction
deep learning
graph attention network
url https://doi.org/10.1029/2022SW003128
work_keys_str_mv AT yanrusun solarwindspeedpredictionviagraphattentionnetwork
AT zongxiaxie solarwindspeedpredictionviagraphattentionnetwork
AT haochengwang solarwindspeedpredictionviagraphattentionnetwork
AT xinhuang solarwindspeedpredictionviagraphattentionnetwork
AT qinghuahu solarwindspeedpredictionviagraphattentionnetwork