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

LD-RPS: Zero-Shot Unified Image Restoration via Latent Diffusion Recurrent Posterior Sampling

About

Unified image restoration is a significantly challenging task in low-level vision. Existing methods either make tailored designs for specific tasks, limiting their generalizability across various types of degradation, or rely on training with paired datasets, thereby suffering from closed-set constraints. To address these issues, we propose a novel, dataset-free, and unified approach through recurrent posterior sampling utilizing a pretrained latent diffusion model. Our method incorporates the multimodal understanding model to provide sematic priors for the generative model under a task-blind condition. Furthermore, it utilizes a lightweight module to align the degraded input with the generated preference of the diffusion model, and employs recurrent refinement for posterior sampling. Extensive experiments demonstrate that our method outperforms state-of-the-art methods, validating its effectiveness and robustness. Our code and data are available at https://github.com/AMAP-ML/LD-RPS.

Huaqiu Li, Yong Wang, Tongwen Huang, Hailang Huang, Haoqian Wang, Xiangxiang Chu• 2025

Related benchmarks

TaskDatasetResultRank
Image DenoisingKodak24 (test)
PSNR28.64
51
DehazingRESIDE (HSTS)
PSNR21.45
25
Low-light enhancementLOL v1
NIQE5.52
21
Low-light enhancementLOL v2
PSNR19.26
8
Low-light enhancementLOL v2 (test)
PSNR18.22
7
Image RestorationLOL Low-light + Noise v1
PSNR16.87
4
Image RestorationLOL Low-light + Haze + Noise v1
PSNR16.77
4
Showing 7 of 7 rows

Other info

Follow for update