气象2026，Vol.52Issue(6)：713-725,13.DOI:10.7519/j.issn.1000-0526.2025.121901
基于双分支复合小波注意力网络的雷达定量降水估计
Radar Quantitative Precipitation Estimation Based on a Dual-Branch Composite Wavelet Attention Network
摘要
Abstract
Current deep learning-based QPE methods using radar reflectivity factors mostly adopt global mapping strategies,which to some extent limits the models' ability to analyze local precipitation features.To this end,this study proposes a Dual-Branch Composite Wavelet Attention UNet(DCWA-UNet)model.The model is mainly improved from the following two aspects.Firstly,a hybrid architecture consisting of a dual-branch encoder(main branch+simplified convolutional downsampling branch)and a feature aggre-gation subnetwork is designed,which enables the end-to-end mapping from radar volume scan data to station precipitation intensity,thereby constructing a station-centered sample system.Secondly,a com-posite wavelet attention module(CWAM)is introduced to enhance the model's representation capability for radar echoes through multi-scale feature decomposition and dynamic weight allocation.Meanwhile,a weighted mean squared error loss function is adopted to emphasize the gradient contribution of moderate-to-high precipitation.Using ground-based precipitation observation and radar observation data in the Si-chuan Basin during the summers of 2019-2021,a dataset containing 4020 samples is constructed for model training and testing,and comparative experiments are conducted with deep learning models such as Sim-VP.The results show that DCWA-UNet achieves obvious comprehensive performance advantages under different precipitation intensities,with particularly significant improvements in critical success index(CSI)and mean absolute error(MAE)within the precipitation intensity below 30 mm·h-1.For precipitation in-tensities of[5,10)mm·h-1,the CSI of DCWA-UNet is significantly higher than that of SimVP and oth-er comparative models,and the MAE is reduced by 4.9％compared to SimVP;for precipitation intensities of[10,30]mm·h-1,the CSI is improved by 4.0％and the MAE is reduced by 4.0％compared to Sim-VP.Moreover,the false alarm rate is the lowest among all comparative models.关键词
雷达定量降水估计/双分支编码器/复合小波注意力模块/加权均方误差/U-NetKey words
radar-based quantitative precipitation estimation/dual-branch encoder/composite wavelet attention module/weighted mean squared error/U-Net分类
天文与地球科学引用本文复制引用
马星洪,钟琦,杨昊,陈敏,周航..基于双分支复合小波注意力网络的雷达定量降水估计[J].气象,2026,52(6):713-725,13.基金项目
国家自然科学基金项目(42030611)、国家重点研发计划(2023YFC3007502)、四川省科技成果转移转化示范项目(2024ZHCG0026)和中国气象局气象干部培训学院科研项目-培育项目(2025CMATCPY06)共同资助 （42030611）
