Cloud Removal Using Patch-Based Improved Denoising Diffusion Models and High Gray-Value Attention Mechanism

In recent years, diffusion-based methods have outperformed traditional models in many cloud removal tasks due to their strong generative capabilities. However, these methods face the challenges of long inference time and poor recovery effect in cloud regions. To address this issue, this paper proposes a Patch-Based Improved Denoising Diffusion Model with a high grey value attention for cloud removal in optical remote sensing images. We introduce an overlapping fixed-sized patches method in Improved Denoising Diffusion Model. The patch-based diffusion modeling approach enables size-agnostic image restoration by employing a guided denoising process with smoothed noise estimates across overlapping patches during inference. Additionally, we introduce a high grey value attention module, specifically designed to focus on thick cloud regions, enhancing attention on areas with relatively high grey values within the image. When compared with other existing cloud removal models on the RICE dataset, our model outperformed them in terms of both peak signal-to-noise ratio (PSNR) and structural similarity index (SSIM). Qualitative results demonstrate that the proposed method effectively removes clouds from images while preserving texture details. Ablation studies further confirm the effectiveness of the high grey value attention module. Overall, the proposed model delivers superior cloud removal performance compared to existing state-of-the-art methods.