基于稀疏注意力的雷達回波外推方法及應用
作者:
作者單位:

作者簡介:

通訊作者:

中圖分類號:

基金項目:

四川省科技計劃項目(2023JDZH0034)、四川省科技成果轉移轉化示范項目(2024ZHGG0026)資助


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

Fund Project:

  • 摘要
  • |
  • 圖/表
  • |
  • 訪問統計
  • |
  • 參考文獻
  • |
  • 相似文獻
  • |
  • 引證文獻
  • |
  • 資源附件
  • |
  • 文章評論
    摘要:

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

    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.

    參考文獻
    相似文獻
    引證文獻
引用本文

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

復制
分享
文章指標
  • 點擊次數:
  • 下載次數:
  • HTML閱讀次數:
  • 引用次數:
歷史
  • 收稿日期:2024-11-07
  • 定稿日期:2025-01-24
  • 錄用日期:
  • 在線發布日期: 2025-08-27
  • 出版日期:
您是第位訪問者
技術支持:北京勤云科技發展有限公司
午夜欧美大片免费观看,欧美激情综合五月色丁香,亚洲日本在线视频观看,午夜精品福利在线
>