WSSM: Geographic-enhanced hierarchical state-space model for global station weather forecast

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yang, Songru, Liu, Zili, Shi, Zhenwei, Zou, Zhengxia
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913658555072512
author Yang, Songru
Liu, Zili
Shi, Zhenwei
Zou, Zhengxia
author_facet Yang, Songru
Liu, Zili
Shi, Zhenwei
Zou, Zhengxia
contents Global Station Weather Forecasting (GSWF), a prominent meteorological research area, is pivotal in providing timely localized weather predictions. Despite the progress existing models have made in the overall accuracy of the GSWF, executing high-precision extreme event prediction still presents a substantial challenge. The recent emergence of state-space models, with their ability to efficiently capture continuous-time dynamics and latent states, offer potential solutions. However, early investigations indicated that Mamba underperforms in the context of GSWF, suggesting further adaptation and optimization. To tackle this problem, in this paper, we introduce Weather State-space Model (WSSM), a novel Mamba-based approach tailored for GSWF. Geographical knowledge is integrated in addition to the widely-used positional encoding to represent the absolute special-temporal position. The multi-scale time-frequency features are synthesized from coarse to fine to model the seasonal to extreme weather dynamic. Our method effectively improves the overall prediction accuracy and addresses the challenge of forecasting extreme weather events. The state-of-the-art results obtained on the Weather-5K subset underscore the efficacy of the WSSM
format Preprint
id arxiv_https___arxiv_org_abs_2501_11238
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle WSSM: Geographic-enhanced hierarchical state-space model for global station weather forecast
Yang, Songru
Liu, Zili
Shi, Zhenwei
Zou, Zhengxia
Machine Learning
Artificial Intelligence
Atmospheric and Oceanic Physics
Global Station Weather Forecasting (GSWF), a prominent meteorological research area, is pivotal in providing timely localized weather predictions. Despite the progress existing models have made in the overall accuracy of the GSWF, executing high-precision extreme event prediction still presents a substantial challenge. The recent emergence of state-space models, with their ability to efficiently capture continuous-time dynamics and latent states, offer potential solutions. However, early investigations indicated that Mamba underperforms in the context of GSWF, suggesting further adaptation and optimization. To tackle this problem, in this paper, we introduce Weather State-space Model (WSSM), a novel Mamba-based approach tailored for GSWF. Geographical knowledge is integrated in addition to the widely-used positional encoding to represent the absolute special-temporal position. The multi-scale time-frequency features are synthesized from coarse to fine to model the seasonal to extreme weather dynamic. Our method effectively improves the overall prediction accuracy and addresses the challenge of forecasting extreme weather events. The state-of-the-art results obtained on the Weather-5K subset underscore the efficacy of the WSSM
title WSSM: Geographic-enhanced hierarchical state-space model for global station weather forecast
topic Machine Learning
Artificial Intelligence
Atmospheric and Oceanic Physics
url https://arxiv.org/abs/2501.11238