返回
RNDiff: Rainfall nowcasting with Condition Diffusion Model
DOI:10.1016/j.patcog.2024.111193.png)
摘要
En 中文
The Diffusion Models are widely used in image generation because they can generate high-quality and realistic samples. In contrast, generative adversarial networks (GANs) and variational autoencoders (VAEs) have some limitations in terms of image quality. We introduce a diffusion model to the precipitation forecasting task and propose a short-term precipitation nowcasting with condition diffusion model based on historical observational data, which is referred to as Rainfall nowcasting with Condition Diffusion Model(RNDiff). By incorporating an additional conditional decoder module in the denoising process, RNDiff achieves end-to-end conditional rainfall prediction. RNDiff is composed of two networks: a denoising network and a conditional encoder network. The conditional network is composed of multiple independent UNet networks. These networks extract conditional feature maps at different resolutions, providing accurate conditional information that guides the diffusion model for conditional generation. RNDiff surpasses GANs in terms of prediction accuracy, although it requires more computational resources. The RNDiff model exhibits higher stability and efficiency during training than GANs-based approaches, and generates high-quality precipitation distribution samples that better reflect future actual precipitation conditions. Compared to the current state-of-the-art GAN-based methods, our proposed approach achieves significant improvements on key evaluation metrics. Specifically, our method leads to improvements in the CSI, HSS, and FSS, which are increased by around 8%, 5%, and 6%, respectively. The experiment fully verified the advantages and potential of RNdiff in precipitation forecasting and provided new insights for improving rainfall forecasting. Our project is open source and available on GitHub at: https://github.com/ybu-lxd/RNDiff.
Keyword:
Diffusion models
Rainfall prediction
Condition encoder
期刊
IF:
7.6
论文数:
1.3W
被引数:
4.5W
机构
引用论文
Sensorless Control of Z Source Inverter fed BLDC Motor Drive by FOC - DTC Hybrid Control Strategy Using Fuzzy Logic Controller采用模糊逻辑控制器的foc-dtc混合控制策略的Z源逆变器馈电BLDC电机驱动的无传感器控制
U2-Net: Going deeper with nested U-structure for salient object detectionU2-Net: 基于嵌套U结构的显著性目标检测
PATTERN RECOGNITION
IF7.6
Energy-Efficient Robot Configuration and Motion Planning Using Genetic Algorithm and Particle Swarm Optimization基于遗传算法和粒子群算法的节能机器人配置和运动规划
Energies
IF0
Recurrent flow networks: A recurrent latent variable model for density estimation of urban mobility
PATTERN RECOGNITION
IF7.6

