改进U-Net的风云四号卫星降水估计算法研究

Research on Precipitation Estimation Algorithm from Fengyun-4 Satellite Based on Improved U-Net

针对强对流天气条件下利用卫星图像进行降水量估计时精度不高、时空分辨率低的问题,提出了一种改进U-Net的降水量估计算法。将U-Net模型的编码器通过残差模块与解码器相结合,使得模型参数可以共享,避免深层网络模型梯度消失的情况。在此结构基础上引入了空间金字塔模块进行多尺度特征提取,保留更多的图像特征,加强对细小降水云团信息的特征提取能力;引入了注意力机制模块,提取重要降水特征信息。实验结果表明,该算法在命中率、虚警率、相关系数分别为0.84、0.48、0.59;均方根误差、平均绝对误差分别为1.354 mm/h、0.432 mm/h。与PERSIANN-CNN、U-Net算法相比,有效地提升了降水量估计精度。与其他降水产品对比,能更好地识别出降水区。该算法可以实现近实时的降水估计,能够有效地提升降水估计精度,对低时间分辨率的降水估计研究具有一定的价值。

Aiming at the problems of low accuracy and low spatial-temporal resolution of precipitation estimation using satellite images under the condition of severe convective weather,an improved U-Net precipitation estimation algorithm is proposed.Firstly the encoder of the U-Net model is combined with the decoder through the residual module,so that the model parameters can be shared to avoid the disappearance of deep network model gradients.Based on this structure,the spatial pyramid module is introduced for multi-scale feature extraction to retain more image features and strengthen the feature extraction ability of small precipitation cloud information.The attention mechanism module is added to extract important precipitation feature information.The experimental results show that probability of detection,false alarm ration,critical success index of the proposed algorithm are 0.84,0.48 and 0.59,respectively.The root mean square error and mean absolute error are 1.354 mm/h and 0.432 mm/h respectively.Compared with PERSIANN-CNN and U-Net,the proposed algorithm effectively improves the accuracy of precipitation estimation.Compared with other precipitation products,it also has certain advantages.Therefore,the algorithm can achieve near-real-time results and effectively improve the accuracy of precipitation estimation,which is valuable for the research of precipitation estimation with low tempotal resolution.