STAA: Spatio-Temporal Alignment Attention for Short-Term Precipitation Forecasting

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Chen, Min, Yang, Hao, Li, Shaohan, Qin, Xiaolin
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914947099787264
author Chen, Min
Yang, Hao
Li, Shaohan
Qin, Xiaolin
author_facet Chen, Min
Yang, Hao
Li, Shaohan
Qin, Xiaolin
contents There is a great need to accurately predict short-term precipitation, which has socioeconomic effects such as agriculture and disaster prevention. Recently, the forecasting models have employed multi-source data as the multi-modality input, thus improving the prediction accuracy. However, the prevailing methods usually suffer from the desynchronization of multi-source variables, the insufficient capability of capturing spatio-temporal dependency, and unsatisfactory performance in predicting extreme precipitation events. To fix these problems, we propose a short-term precipitation forecasting model based on spatio-temporal alignment attention, with SATA as the temporal alignment module and STAU as the spatio-temporal feature extractor to filter high-pass features from precipitation signals and capture multi-term temporal dependencies. Based on satellite and ERA5 data from the southwestern region of China, our model achieves improvements of 12.61\% in terms of RMSE, in comparison with the state-of-the-art methods.
format Preprint
id arxiv_https___arxiv_org_abs_2409_06732
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle STAA: Spatio-Temporal Alignment Attention for Short-Term Precipitation Forecasting
Chen, Min
Yang, Hao
Li, Shaohan
Qin, Xiaolin
Atmospheric and Oceanic Physics
Machine Learning
There is a great need to accurately predict short-term precipitation, which has socioeconomic effects such as agriculture and disaster prevention. Recently, the forecasting models have employed multi-source data as the multi-modality input, thus improving the prediction accuracy. However, the prevailing methods usually suffer from the desynchronization of multi-source variables, the insufficient capability of capturing spatio-temporal dependency, and unsatisfactory performance in predicting extreme precipitation events. To fix these problems, we propose a short-term precipitation forecasting model based on spatio-temporal alignment attention, with SATA as the temporal alignment module and STAU as the spatio-temporal feature extractor to filter high-pass features from precipitation signals and capture multi-term temporal dependencies. Based on satellite and ERA5 data from the southwestern region of China, our model achieves improvements of 12.61\% in terms of RMSE, in comparison with the state-of-the-art methods.
title STAA: Spatio-Temporal Alignment Attention for Short-Term Precipitation Forecasting
topic Atmospheric and Oceanic Physics
Machine Learning
url https://arxiv.org/abs/2409.06732