基于稀疏注意力的雷達(dá)回波外推方法及應(yīng)用
作者:
作者單位:

作者簡介:

通訊作者:

中圖分類號:

基金項(xiàng)目:

四川省科技計(jì)劃項(xiàng)目(2023JDZH0034)、四川省科技成果轉(zhuǎn)移轉(zhuǎn)化示范項(xiàng)目(2024ZHGG0026)資助


Research on Radar Echo Extrapolation Method Based on Sparse Attention
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 圖/表
  • |
  • 訪問統(tǒng)計(jì)
  • |
  • 參考文獻(xiàn)
  • |
  • 相似文獻(xiàn)
  • |
  • 引證文獻(xiàn)
  • |
  • 資源附件
  • |
  • 文章評論
    摘要:

    基于深度學(xué)習(xí)的雷達(dá)回波外推方法已被廣泛應(yīng)用于具有挑戰(zhàn)性的短臨降水預(yù)報任務(wù)中。然而,現(xiàn)有方法在預(yù)測精度上存在不夠高的問題,而且在處理高分辨率和長時間序列數(shù)據(jù)時,訓(xùn)練速度較慢。為此,本文提出了一種基于稀疏融合注意力的深度學(xué)習(xí)模型——PFA-TransUNet(Probsparse Fusion Attention TransUNet)。該模型是一種編碼器-解碼器架構(gòu),通過在編碼器路徑中引入多層Transformer,然后將傳統(tǒng)多頭自注意力分解為時空維度計(jì)算,從而充分融合時空信息。同時,引入稀疏注意力方法,降低了自注意力計(jì)算復(fù)雜度,縮短了訓(xùn)練時間。在河北省雷達(dá)數(shù)據(jù)集上的實(shí)驗(yàn)結(jié)果表明,與其他先進(jìn)的經(jīng)典模型相比,PFA-TransUNet在外推精度,均方根誤差,結(jié)構(gòu)相似性,雷達(dá)回波為20、30、40 dBz時的臨界成功指數(shù),訓(xùn)練速度等多個評價指標(biāo)上均有所提升,展現(xiàn)出了優(yōu)異的整體性能。

    Abstract:

    The deep learning-based radar echo extrapolation method is widely applied to the challenging task of short-term precipitation forecasting. However, existing methods still face issues with prediction accuracy, and when dealing with high-resolution and long-time sequence data, the training speed tends to be slow. To address these problems, this paper proposes a deep learning model based on sparse fusion attention - PFA-TransUNet (ProbSparse Fusion Attention TransUNet). This model is an encoder-decoder architecture, where a multi-layer Transformer is introduced in the encoder path. It then decomposes the traditional multi-head self-attention mechanism into computations in the spatiotemporal dimensions, allowing for the full integration of spatiotemporal information. In addition, the sparse attention method is incorporated to reduce the computational complexity of self-attention, significantly shortening the training time. Experimental results on the Hebei Province radar dataset show that compared to other advanced classical models, PFA-TransUNet outperforms them in various evaluation metrics such as extrapolation accuracy, Mean Squared Error (MSE), Structural Similarity Index (SSIM), Critical Success Index (CSI) at 20, 30, and 40 dBz, and training speed. The model demonstrates exceptional overall performance. In recent years, radar echo extrapolation becomes an increasingly important approach in precipitation forecasting, especially for nowcasting (short-term forecasting) tasks, where the ability to predict precipitation with high accuracy and efficiency is critical. However, due to the complex spatiotemporal nature of radar echoes, previous methods struggle to efficiently capture both spatial and temporal dependencies, which leads to suboptimal forecasting results. Furthermore, the computational cost associated with high-resolution and long-time series data further hampers the efficiency of current deep learning models. PFA-TransUNet addresses these limitations by incorporating a sparse attention mechanism, which helps reduce the computational load without sacrificing model performance. Traditional self-attention mechanisms in Transformer models can be computationally expensive due to the quadratic complexity of attention calculations, especially when applied to large datasets. By leveraging sparse attention, the proposed model focuses on the most relevant parts of the input data, thus improving computational efficiency and speeding up training. Another key feature of PFA-TransUNet is its ability to effectively model spatiotemporal dependencies. By decomposing the multi-head self-attention into spatiotemporal dimensions, the model captures the intricate relationships between space and time, leading to more accurate extrapolations of radar echoes. This is crucial in precipitation forecasting, as both spatial distribution and temporal evolution play a significant role in the prediction accuracy. The experimental results from the Hebei radar dataset indicate that PFA-TransUNet achieves superior performance compared to traditional models. The model shows a substantial improvement in forecast accuracy, with lower MSE values and higher SSIM scores, indicating better preservation of the structure of radar echoes. Furthermore, the model excels in terms of CSI at different dBz thresholds, demonstrating its robustness in detecting precipitation events under various conditions. Most importantly, the model’s training speed is significantly improved due to the sparse attention mechanism, making it suitable for real-time forecasting applications. In conclusion, PFA-TransUNet presents a promising solution for radar echo extrapolation tasks, especially in the context of short-term precipitation forecasting. Its combination of sparse fusion attention and spatiotemporal modelling makes it a powerful tool for improving the accuracy and efficiency of radar-based forecasting systems.

    參考文獻(xiàn)
    相似文獻(xiàn)
    引證文獻(xiàn)
引用本文

王杰,陳靜,楊昊,陳敏,周航,王勇.基于稀疏注意力的雷達(dá)回波外推方法及應(yīng)用[J].氣象科技,2025,53(4):468~478

復(fù)制
分享
文章指標(biāo)
  • 點(diǎn)擊次數(shù):
  • 下載次數(shù):
  • HTML閱讀次數(shù):
  • 引用次數(shù):
歷史
  • 收稿日期:2024-11-07
  • 定稿日期:2025-01-24
  • 錄用日期:
  • 在線發(fā)布日期: 2025-08-27
  • 出版日期:
您是第位訪問者
技術(shù)支持:北京勤云科技發(fā)展有限公司
延边| 观塘区| 盐城市| 福贡县| 北宁市| 海安县| 南投市| 登封市| 固原市| 恩施市| 广灵县| 巴彦淖尔市| 班戈县| 柳江县| 阿瓦提县| 宝丰县| 三明市| 怀宁县| 涿鹿县| 石林| 普定县| 宁阳县| 本溪市| 阜宁县| 定边县| 喜德县| 湛江市| 余干县| 永新县| 江山市| 扎赉特旗| 康保县| 高陵县| 南丹县| 垦利县| 定结县| 图们市| 沽源县| 南召县| 泊头市| 祁东县|