The Diffusion Duet: Harmonizing Dual Channels with Wavelet Suppression for Image Separation

TL;DR

This paper introduces a novel dual-channel diffusion model with wavelet suppression for improved blind image separation, effectively handling complex mixtures and noise, and achieving state-of-the-art results in rain/snow removal and image restoration.

Contribution

It presents the first integration of diffusion models with a wavelet suppression module for dual-channel blind image separation, enhancing feature learning and detail preservation.

Findings

Achieves higher PSNR/SSIM in rain and snow removal compared to existing methods.

Outperforms state-of-the-art in complex mixture separation with significant PSNR and SSIM gains.

Demonstrates superior subjective and objective performance in image restoration tasks.

Abstract

Blind image separation (BIS) refers to the inverse problem of simultaneously estimating and restoring multiple independent source images from a single observation image under conditions of unknown mixing mode and without prior knowledge of the source images. Traditional methods relying on statistical independence assumptions or CNN/GAN variants struggle to characterize complex feature distributions in real scenes, leading to estimation bias, texture distortion, and artifact residue under strong noise and nonlinear mixing. This paper innovatively introduces diffusion models into dual-channel BIS, proposing an efficient Dual-Channel Diffusion Separation Model (DCDSM). DCDSM leverages diffusion models' powerful generative capability to learn source image feature distributions and reconstruct feature structures effectively. A novel Wavelet Suppression Module (WSM) is designed within the dual-branch reverse denoising process, forming an interactive separation network that enhances detail separation by exploiting the mutual coupling noise characteristic between source images. Extensive experiments on synthetic datasets containing rain/snow and complex mixtures demonstrate that DCDSM achieves state-of-the-art performance: 1) In image restoration tasks, it obtains PSNR/SSIM values of 35.0023 dB/0.9549 and 29.8108 dB/0.9243 for rain and snow removal respectively, outperforming Histoformer and LDRCNet by 1.2570 dB/0.9272 dB (PSNR) and 0.0262/0.0289 (SSIM) on average; 2) For complex mixture separation, the restored dual-source images achieve average PSNR and SSIM of 25.0049 dB and 0.7997, surpassing comparative methods by 4.1249 dB and 0.0926. Both subjective and objective evaluations confirm DCDSM's superiority in addressing rain/snow residue removal and detail preservation challenges.