Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

UnfoldLDM: Deep Unfolding-based Blind Image Restoration with Latent Diffusion Priors

About

Deep unfolding networks (DUNs) combine the interpretability of model-based methods with the learning ability of deep networks, yet remain limited for blind image restoration (BIR). Existing DUNs suffer from: (1) \textbf{Degradation-specific dependency}, as their optimization frameworks are tied to a known degradation model, making them unsuitable for BIR tasks; and (2) \textbf{Over-smoothing bias}, resulting from the direct feeding of gradient descent outputs, dominated by low-frequency content, into the proximal term, suppressing fine textures. To overcome these issues, we propose UnfoldLDM to integrate DUNs with latent diffusion model (LDM) for BIR. In each stage, UnfoldLDM employs a multi-granularity degradation-aware (MGDA) module as the gradient descent step. MGDA models BIR as an unknown degradation estimation problem and estimates both the holistic degradation matrix and its decomposed forms, enabling robust degradation removal. For the proximal step, we design a degradation-resistant LDM (DR-LDM) to extract compact degradation-invariant priors from the MGDA output. Guided by this prior, an over-smoothing correction transformer (OCFormer) explicitly recovers high-frequency components and enhances texture details. This unique combination ensures the final result is degradation-free and visually rich. Experiments show that our UnfoldLDM achieves a leading place on various BIR tasks and benefits downstream tasks. Moreover, our design is compatible with existing DUN-based methods, serving as a plug-and-play framework. Code will be released.

Chunming He, Rihan Zhang, Zheng Chen, Bowen Yang, Chengyu Fang, Yunlong Lin, Yulun Zhang, Fengyang Xiao, Sina Farsiu• 2025

Related benchmarks

TaskDatasetResultRank
Image DeblurringGoPro
PSNR34.32
354
DerainingRain100L
PSNR39.56
196
Image DenoisingDND
PSNR40.15
135
Low-light Image EnhancementLOL v1
PSNR25.58
135
Low-light Image EnhancementLOL real v2
PSNR23.88
122
Image DenoisingSIDD
PSNR40.23
102
Object DetectionExDark
mAP (Mean Average Precision)81.1
58
Image DeblurringHIDE
PSNR31.85
56
Low-light Image EnhancementLIME
NIQE3.336
56
Low-light Image EnhancementMEF
NIQE3.152
46
Showing 10 of 24 rows

Other info

Follow for update