Exploiting Diffusion Prior for Real-World Image Super-Resolution
About
We present a novel approach to leverage prior knowledge encapsulated in pre-trained text-to-image diffusion models for blind super-resolution (SR). Specifically, by employing our time-aware encoder, we can achieve promising restoration results without altering the pre-trained synthesis model, thereby preserving the generative prior and minimizing training cost. To remedy the loss of fidelity caused by the inherent stochasticity of diffusion models, we employ a controllable feature wrapping module that allows users to balance quality and fidelity by simply adjusting a scalar value during the inference process. Moreover, we develop a progressive aggregation sampling strategy to overcome the fixed-size constraints of pre-trained diffusion models, enabling adaptation to resolutions of any size. A comprehensive evaluation of our method using both synthetic and real-world benchmarks demonstrates its superiority over current state-of-the-art approaches. Code and models are available at https://github.com/IceClear/StableSR.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Object Detection | COCO 2017 (val) | -- | 2930 | |
| Instance Segmentation | COCO 2017 (val) | APm0.146 | 1304 | |
| Semantic segmentation | ADE20K | mIoU19.6 | 1028 | |
| Super-Resolution | Set5 | PSNR20.35 | 854 | |
| Image Super-resolution | RealSR | LPIPS0.2604 | 257 | |
| Image Super-resolution | DIV2K (val) | LPIPS0.3016 | 215 | |
| Image Super-resolution | DRealSR | MUSIQ58.72 | 182 | |
| Super-Resolution | DIV2K | PSNR20.59 | 155 | |
| Video Super-Resolution | UDM10 | PSNR26.45 | 111 | |
| Super-Resolution | ImageNet (test) | LPIPS0.2327 | 110 |