atlas

Research map / Image generation / Super-resolution

Taming Diffusion Prior for Image Super-Resolution with Domain Shift SDEs

NeurIPS 2024 · Image generation › Super-resolution · approach: Diffusion-based super-resolution · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes DoSSR, a domain-shift diffusion SR model that starts sampling from LR images, fine-tunes SD 2.1 via LAControlNet, and derives continuous DoS-SDEs with fast customized solvers reaching strong quality in 5 steps.

Problem

Diffusion-based super-resolution either leverages pretrained diffusion priors but needs many steps starting from Gaussian noise, or starts from LR images but trains from scratch and cannot use pretrained diffusion priors.

Approach

A domain shift equation linearly interpolates between LR and HR with a shifting sequence that is compatible with the DDPM noise schedule, so pretrained SD can be fine-tuned and sampling starts at an intermediate timestep from LR. The shift is extended to continuous DoS-SDEs with an exact solution and customized first-to-third-order solvers that include a domain shift guidance term.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DoSSR's domain-shift diffusion process and DoSG instead of the original DDPM diffusion equation (ablation)Diffusion-based SR model · SR benchmarks in the DoSG ablationMetrics reported in the ablation tableHelped
Decrease the starting point t1 from T to T/2 (ablation)DoSSR · SR performance across t1 settingsSR performanceHelped
Decrease the starting point t1 further from T/2 to 3/T (ablation)DoSSR · SR performance across t1 settingsSR performanceHurt
Increase the number of inference steps (ablation)DoSSR · RealSRReference-based metricsHurt
Increase the number of inference steps (ablation)DoSSR · RealSRNon-reference metricsHelped
Use one inference step instead of more sampling steps (ablation)DoSSR · RealSRMANIQA: 0.5115 MANIQA; 0.6258 CLIPIQAMixed
Use a higher-order DoS-SDE solver rather than the first-order solver (ablation)DoSSR · SR benchmarks under limited inference stepsNon-reference metricsHelped
Use DoSSR rather than SeeSRStable Diffusion-based image SR · DIV2K-Val, RealSR, DRealSR, and Real200CLIPIQA: 0.6935 -> 0.7014; 0.6699 -> 0.7025; 0.6708 -> 0.6776; 0.7167 -> 0.7437Helped
Use DoSSR rather than SeeSRStable Diffusion-based image SR · DIV2K-Val, RealSR, DRealSR, and Real200MANIQA: 0.5041 -> 0.5294; 0.5406 -> 0.5781; 0.5115 -> 0.5214; 0.5588 -> 0.5794Helped
Use DoSSR rather than SeeSRStable Diffusion-based image SR · x4 SR, 128×128 LR imagesNFE: 50 -> 5Helped
Use DoSSR rather than SeeSRStable Diffusion-based image SR · x4 SR, 128×128 LR imagesLatency/Image: 7.24s -> 1.03sHelped

Surprising findings

Limitations

Experimental setup

Datasets: DIV2K DIV8K Flickr2K OST RealSR DRealSR RealLR200 ImageNet

Benchmarks: DIV2K-Val RealSR DRealSR RealLR200

Metrics: PSNR SSIM LPIPS CLIPIQA MUSIQ MANIQA TOPIQ latency

Baselines: StableSR ResShift DiffBIR SeeSR Real-ESRGAN BSRGAN LDL DASR

Builds on: DDPM ResShift ControlNet StableSR Stable Diffusion

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy