A Swin Transformer for Multiparameter Solar Wind Forecasting from MHD Simulations
- 1Nanchang University
- 2National Satellite Meteorological Center (National Center for Space Weather)
- 3University of Electronic Science and Technology of China
Abstract
Accurate forecasting of solar wind parameters is crucial for space weather forecasting and the protection of infrastructure. Existing physics-based magnetohydrodynamic (MHD) models are computationally expensive and often rely on assumptions. Most deep learning methods focus only on single-parameter predictions at spatial locations, making it difficult to characterize the multiparameter coupled dynamics that govern space weather effects. The model proposed in this study is based on the Transformer and adopts a Swin Transformer encoder–decoder architecture. It is trained on MHD simulation data and can predict eight solar wind parameters (number density, three velocity components, temperature, and three magnetic field components) on a two-dimensional polar coordinate grid of the solar equatorial plane. In order to handle the nonuniform spatial distribution of the grid in the polar coordinate system, this study introduces a position-dependent relative bias to encode the variation of physical spacing with heliocentric distance. The model was trained on MHD simulation data covering 631 Carrington rotations. In single-step (24 hr) predictions on the test set, the weighted correlation coefficients (wCC) exceed 0.98 for all parameters. In the autoregressive rolling prediction mode, the model maintains high accuracy for approximately 72 hr. When compared with OMNI observations covering the 2009–2023 testing period, the correlation coefficient for near-Earth radial velocity is 0.71. These results demonstrate that the Transformer-based architecture can effectively learn the multiparameter coupled dynamics of the solar wind from MHD simulations, serving as a computationally efficient alternative model and possessing the potential for operational evaluation.
Related
- Short-term solar flare forecasting with a hybrid ConvNeXtV2 and transformer–temporal convolutional network
- Global solar compressible convection simulations on non-Cartesian mappings
- Bifrost spectral inversions. Fast non-LTE solar chromospheric diagnostics from 3D simulations
- Influence of local and global solar magnetic fields on medium-term forecasting of the number and parameters of coronal mass ejections
- PROSWIN: Probabilistic Solar Wind Speed Forecasting Using Deep Distributional Regression From Solar Images
BibTeX
@article{deng2026swin,
title = {{A Swin Transformer for Multiparameter Solar Wind Forecasting from MHD Simulations}},
author = {Yiwan Deng and Yike Qiu and Zhou Chen and Kening Zhang and Tingfeng Xie and Jing-Song Wang},
journal = {The Astrophysical Journal},
year = {2026},
month = nov,
doi = {10.3847/1538-4357/aeab9e},
url = {https://iopscience.iop.org/article/10.3847/1538-4357/aeab9e},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - A Swin Transformer for Multiparameter Solar Wind Forecasting from MHD Simulations
AU - Yiwan Deng
AU - Yike Qiu
AU - Zhou Chen
AU - Kening Zhang
AU - Tingfeng Xie
AU - Jing-Song Wang
PY - 2026
DA - 2026/11/03
JO - The Astrophysical Journal
DO - 10.3847/1538-4357/aeab9e
UR - https://iopscience.iop.org/article/10.3847/1538-4357/aeab9e
AB - Accurate forecasting of solar wind parameters is crucial for space weather forecasting and the protection of infrastructure. Existing physics-based magnetohydrodynamic (MHD) models are computationally expensive and often rely on assumptions. Most deep learning methods focus only on single-parameter predictions at spatial locations, making it difficult to characterize the multiparameter coupled dynamics that govern space weather effects. The model proposed in this study is based on the Transformer and adopts a Swin Transformer encoder–decoder architecture. It is trained on MHD simulation data and can predict eight solar wind parameters (number density, three velocity components, temperature, and three magnetic field components) on a two-dimensional polar coordinate grid of the solar equatorial plane. In order to handle the nonuniform spatial distribution of the grid in the polar coordinate system, this study introduces a position-dependent relative bias to encode the variation of physical spacing with heliocentric distance. The model was trained on MHD simulation data covering 631 Carrington rotations. In single-step (24 hr) predictions on the test set, the weighted correlation coefficients (wCC) exceed 0.98 for all parameters. In the autoregressive rolling prediction mode, the model maintains high accuracy for approximately 72 hr. When compared with OMNI observations covering the 2009–2023 testing period, the correlation coefficient for near-Earth radial velocity is 0.71. These results demonstrate that the Transformer-based architecture can effectively learn the multiparameter coupled dynamics of the solar wind from MHD simulations, serving as a computationally efficient alternative model and possessing the potential for operational evaluation.
SN - 1538-4357
SN - 0004-637X
ER -