atlas

Research map / Image generation / Fast diffusion sampling

EmerDiff: Emerging Pixel-level Semantic Knowledge in Diffusion Models

ICLR 2024 · Image generation › Fast diffusion sampling · approach: Diffusion models as perception tools · 0 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.

Builds an unsupervised, fine-grained image segmentor from Stable Diffusion v1.4 alone by perturbing low-resolution cross-attention query features and measuring pixel-level changes in generated images, then plugs the masks into annotation-free open-vocabulary models.

Problem

Pre-trained diffusion models' semantic features live in low-resolution layers, so k-means segmentation from them is coarse; it is unclear how much pixel-level semantic knowledge diffusion models hold without extra annotations or priors.

Approach

Modulate a sub-region of 16x16 cross-attention features during SD's denoising, run with +/- offsets, and use the RGB difference map as the pixel-to-mask semantic correspondence strength; each pixel takes the mask with the strongest correspondence, yielding image-resolution masks without training.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add pixel-level correspondence (modulation-based image-resolution maps) vs naive bilinear upsampling of low-res masksSD v1.4 unsupervised segmentation · ADE20K (AD150)mIoU: 29.1 -> 33.1Helped
add pixel-level correspondence (modulation-based image-resolution maps) vs naive bilinear upsamplingSD v1.4 unsupervised segmentation · PC59, PC459, CS27, City19mIoU: e.g. CS27 42.1 -> 45.8; PC59 41.5 -> 45.7Helped
add Ours masks to MaskCLIP (annotation-free open-vocab)MaskCLIP · ADE20K (AD150)mIoU: 11.5 -> 15.9Helped
add Ours masks to TCLTCL · PASCAL-Context PC59 / ADE20K / CS171 / City19mIoU: PC59 30.9->35.4; AD150 14.6->17.4; PC459 5.22->6.49; CS171 18.7->21.8; City19 20.9->23.4Mixed
add Ours masks to CLIPpyCLIPpy · Cityscapes City19 open-vocabmIoU: 9.76 -> 9.24Hurt
remove attention injection in modulated denoising (ablation)Ours unsupervised segmentation · ADE20K unsupervised segmIoU: 31.8 -> 33.1 with injectionHelped
attention injection in open-vocab pipeline (ablation)MaskCLIP + Ours · ADE20K open-vocabmIoU: 15.6 -> 15.9Helped
extract low-res feature maps at later timesteps (vs t=1) (ablation)Ours unsupervised segmentation · ADE20K unsupervised segmIoU: 33.1 at t=1 -> 26.9 at t=801Hurt
modulation timestep choice (early-middle vs very early or very late) (ablation)Ours unsupervised segmentation · ADE20K unsupervised segmIoU: t_m=1: 29.0; 281: 33.1; 481: 33.1; 881: 27.4Mixed
modulation strength lambda 1 to 1000 (ablation)Ours unsupervised segmentation · ADE20K unsupervised segmIoU: 32.7 / 33.1 / 32.8 / 32.6No clear effect
choose first vs second vs third 16x16 cross-attention layer for feature extraction (ablation)Ours unsupervised segmentation · ADE20K unsupervised segmIoU: 33.1 / 33.1 / 32.8No clear effect
number of masks per image (10 to 40) (ablation)Ours and MaskCLIP+Ours, TCL+Ours · ADE20K / CS171 / City19mIoU: Ours AD150 30.9 to 33.4 across 10-40 masksNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: COCO-Stuff PASCAL-Context ADE20K Cityscapes

Benchmarks: ADE20K (AD150) PASCAL-Context (PC59, PC459) COCO-Stuff (CS171, CS27) Cityscapes (City19)

Metrics: mIoU

Baselines: IIC PiCIE DINO STEGO DINOSAUR SegDiscover ACSeg CLIP TCL CLIPpy MaskCLIP GroupViT

Builds on: Stable Diffusion DDPM-based inversion MaskCLIP STEGO Prompt-to-prompt attention injection

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