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

作者簡介:

通訊作者:

中圖分類號:

基金項目:

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


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

Fund Project:

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

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

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

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

復(fù)制
分享
文章指標
  • 點擊次數(shù):
  • 下載次數(shù):
  • HTML閱讀次數(shù):
  • 引用次數(shù):
歷史
  • 收稿日期:2024-11-07
  • 定稿日期:2025-01-24
  • 錄用日期:
  • 在線發(fā)布日期: 2025-08-27
  • 出版日期:
您是第位訪問者
技術(shù)支持:北京勤云科技發(fā)展有限公司
双峰县| 石嘴山市| 北流市| 胶州市| 房产| 饶河县| 铁岭县| 肃宁县| 泉州市| 阿克| 自治县| 惠来县| 黄梅县| 安达市| 克拉玛依市| 乐安县| 洞头县| 松潘县| 鄂伦春自治旗| 黑山县| 六盘水市| 利津县| 九龙县| 新营市| 富阳市| 米脂县| 台北市| 杭锦后旗| 长乐市| 襄城县| 理塘县| 封开县| 吴川市| 普安县| 万盛区| 济阳县| 孝义市| 高雄市| 安仁县| 高青县| 阜平县|