atlas

Research map / Vision-language and multimodal / CLIP and zero-shot

Controlling Vision-Language Models for Multi-Task Image Restoration

ICLR 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Fine-grained improvements to CLIP

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 DA-CLIP, a degradation-aware fine-tuned CLIP that predicts degradation and clean-content embeddings from low-quality images to steer diffusion (IR-SDE) and MSE-based restoration models for unified and degradation-specific image restoration.

Problem

Pretrained vision-language models like CLIP degrade on low-level inputs, and existing all-in-one image restoration methods lack discriminative degradation context and semantic guidance for restoring images with diverse corruptions.

Approach

Fine-tune CLIP's image encoder with a degradation classification/contrastive objective against frozen text encoder embeddings of degradation labels and clean captions, then inject the predicted degradation and HQ content embeddings into restoration networks via prompt modules and cross-attention.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use DA-CLIP for unified restorationUnified restoration model · Rain100L (unseen light rain)LPIPS: 0.081 -> 0.025Helped
add DA-CLIP degradation embedding (ablation)NAFNet · Unified image restorationPSNR: 26.34 -> 27.02Helped
add DA-CLIP degradation embedding (ablation)NAFNet · Unified image restorationLPIPS: 0.159 -> 0.146Helped
add DA-CLIP content embedding to degradation embedding (ablation)NAFNet · Unified image restorationPSNR: 27.02 -> 27.22Helped
add DA-CLIP degradation and content embeddings (ablation)NAFNet · Unified image restorationFID: 55.68 -> 47.94Helped
add DA-CLIP content embedding (ablation)PromptIR · Unified image restorationPSNR: 27.14 -> 27.26Helped
add DA-CLIP content embedding (ablation)PromptIR · Unified image restorationFID: 48.26 -> 47.75Helped
add DA-CLIP HQ content embedding (ablation)NAFNet · Image derainingPSNR: 31.49 -> 31.68Helped
add DA-CLIP HQ content embedding (ablation)NAFNet · Low-light enhancementPSNR: 23.09 -> 23.72Helped
add DA-CLIP to NAFNetNAFNet · InpaintingRuntime: 0.08s -> 0.08sNo clear effect
add DA-CLIP to IR-SDEIR-SDE · InpaintingRuntime: 4.2s -> 4.53s + 0.06sHurt

Surprising findings

Limitations

Experimental setup

Datasets: Rain100H Rain100L GoPro mixed degradation dataset (ten degradation types)

Benchmarks: Rain100L GoPro mixed degradation dataset (ten degradation types)

Metrics: PSNR SSIM LPIPS FID

Baselines: AirNet PromptIR IR-SDE NAFNet MAXIM Restormer CLIP + finetune

Builds on: CLIP IR-SDE NAFNet PromptIR AirNet

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