Extreme Precipitation Nowcasting using Multi-Task Latent Diffusion Models

Li Chaorong,Ling Xudong,Yang Qiang,Qin Fengqing,Huang Yuanyuan
2024-10-26
Abstract:Deep learning models have made remarkable strides in precipitation prediction, yet they continue to struggle with capturing the spatial details of the features of radar images, particularly over high precipitation intensity areas. This shortcoming is evident in the form of low forecast accuracy in the spatial positioning of radar echo images across varying precipitation intensity regions. To address this challenge, we introduce the multi-task latent diffusion model(MTLDM), a novel approach for precipitation prediction. The basic concept of the MTLDM is based on the understanding that the radar image representing precipitation is the result of multiple factors. Therefore, we adopt a divide-and-conquer approach, that is, we decompose the radar image using decomposition technology and then predict the decomposed sub-images separately. We conceptualize the precipitation image as a composition of various components corresponding to different precipitation intensities. The MTLDM decomposes the precipitation image into these distinct components and employs a dedicated task to predict each one. This method enables spatiotemporally consistent prediction of real-world precipitation areas up to 5-80 min in advance, outperforming existing state-of-the-art techniques across multiple evaluation metrics.
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?
The problem that this paper attempts to solve is that in extreme precipitation forecasting, the existing deep - learning models are deficient in capturing the spatial - detail features of radar images, especially in high - precipitation - intensity areas. This results in low accuracy in the spatial positioning of the forecasting results. Specifically, the radar echo images generated by the existing models are prone to blurring or loss of details, especially performing poorly when predicting fine structures such as small islands. In addition, for the prediction of high - intensity precipitation (≥16 mm/h), the Critical Success Index (CSI) of the existing models is usually less than 0.5, indicating that they have significant problems in predicting the detailed features of the images and cannot accurately predict precipitation conditions. As the prediction time extends, the spatial prediction accuracy drops sharply, which is one of the main problems in current deep - learning research. To address these challenges, the author introduced a Multi - Task Latent Diffusion Model (MTLDM). By decomposing radar images and predicting these decomposed sub - images separately, the spatial and temporal consistency of precipitation forecasting is improved. This method regards the precipitation image as being composed of various components corresponding to different precipitation intensities. The MTLDM decomposes the precipitation image into these different components and assigns a specific task to each component for prediction. This method enables the model to predict precipitation more accurately at different precipitation - intensity levels, thus improving the overall accuracy of the forecast.