atlas

Research map / Image generation / Image restoration

Boosting Image Restoration via Priors from Pre-trained Models

CVPR 2024 · Image generation › Image restoration · approach: Priors for restoration networks · 19 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 PTG-RM, a lightweight (<1M params) plug-in refinement module that uses off-the-shelf features from frozen pre-trained models (CLIP, BLIP2, Stable Diffusion, restoration models) to refine outputs of existing image restoration networks across low-light enhancement, deraining, deblurring, and denoising.

Problem

Existing image restoration networks have hit performance upper bounds, and strong priors are hard to estimate; pre-trained large-scale models contain degradation-related information that has been unexplored for low-level restoration.

Approach

Use features from a frozen pre-trained model as a prior to learn spatially-varying operation ranges (mixing short-range CNN and long-range transformer features via learned spatial weights) and pre-trained-guided channel-spatial attention with spatially varying kernels, trained jointly with the restoration network and not needed at inference.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add PTG-RM with CLIP priorsSNR · LOL-realPSNR: 21.48 -> 25.50Helped
Add PTG-RM with CLIP priorsRestormer · Test2800PSNR: 34.18 -> 34.47Helped
Add PTG-RM with CLIP priorsRestormer · GoProPSNR: 32.92 -> 33.18Helped
Add PTG-RM with CLIP priorsRestormer · SIDDPSNR: 40.02 -> 40.22Helped
Add PTG-SVE (ablation)Full PTG-RM on SNR · LOL-realPSNR: 24.25 -> 25.50Helped
Add channel attention (ablation)Full PTG-RM on SNR · LOL-realPSNR: 24.05 -> 25.50Helped
Add spatial attention (ablation)Full PTG-RM on SNR · LOL-realPSNR: 23.86 -> 25.50Helped
Replace the stripped components with a fourfold-channel larger module (ablation)PTG-RM on SNR · LOL-realPSNR: 24.51 -> 25.50Hurt
Add learnable position embeddings for priors (ablation)Full PTG-RM on SNR · LOL-realPSNR: 24.13 -> 25.50Helped
Replace the SNR-value mask with the learned PTG-SVE mask (ablation)PTG-RM on SNR · LOL-realPSNR: 24.77 -> 25.50Helped
Replace concatenated 1D priors with the proposed prior-guidance modules (ablation)PTG-RM on SNR · LOL-realPSNR: 23.83 -> 25.50Helped
Replace concatenated 2D priors with the proposed prior-guidance modules (ablation)PTG-RM on SNR · LOL-realPSNR: 24.11 -> 25.50Helped

Surprising findings

Limitations

Experimental setup

Datasets: LOL-real SID Rain13K Rain100H Rain100L Test100 Test1200 Test2800 Set12 BSD68 CBSD68 Kodak24

Benchmarks: LOL-real SID Rain100H Rain100L Test100 Test1200 Test2800 GoPro HIDE RealBlur-R RealBlur-J DPDD

Metrics: PSNR SSIM LPIPS

Baselines: UHD URetinex SNR SKF SMG SPAIR Restormer MPRNet Uformer IFAN GRL DRUNet

Builds on: SNR-Net CLIP-IQA Restormer SKF SMG

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