Research map / Image generation / Fast diffusion sampling
EmerDiff: Emerging Pixel-level Semantic Knowledge in Diffusion Models
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
- Unsupervised segmentation with a modified evaluation protocol reaches 33.1 mIoU on ADE20K (AD150) and 45.7 on PASCAL-Context PC59, outperforming SD naive upsampling (29.1 / 41.5) and DINO/STEGO baselines.
- Under the traditional Hungarian-matching protocol on COCO-Stuff-27, the method reaches 26.6 mIoU, on par with STEGO (26.8).
- Plugging the masks into MaskCLIP raises ADE20K mIoU from 11.5 to 15.9 and Cityscapes from 20.9 to 26.5 in annotation-free open-vocabulary segmentation.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| add pixel-level correspondence (modulation-based image-resolution maps) vs naive bilinear upsampling of low-res masks | SD v1.4 unsupervised segmentation · ADE20K (AD150) | mIoU: 29.1 -> 33.1 | Helped |
| add pixel-level correspondence (modulation-based image-resolution maps) vs naive bilinear upsampling | SD v1.4 unsupervised segmentation · PC59, PC459, CS27, City19 | mIoU: e.g. CS27 42.1 -> 45.8; PC59 41.5 -> 45.7 | Helped |
| add Ours masks to MaskCLIP (annotation-free open-vocab) | MaskCLIP · ADE20K (AD150) | mIoU: 11.5 -> 15.9 | Helped |
| add Ours masks to TCL | TCL · PASCAL-Context PC59 / ADE20K / CS171 / City19 | mIoU: PC59 30.9->35.4; AD150 14.6->17.4; PC459 5.22->6.49; CS171 18.7->21.8; City19 20.9->23.4 | Mixed |
| add Ours masks to CLIPpy | CLIPpy · Cityscapes City19 open-vocab | mIoU: 9.76 -> 9.24 | Hurt |
| remove attention injection in modulated denoising (ablation) | Ours unsupervised segmentation · ADE20K unsupervised seg | mIoU: 31.8 -> 33.1 with injection | Helped |
| attention injection in open-vocab pipeline (ablation) | MaskCLIP + Ours · ADE20K open-vocab | mIoU: 15.6 -> 15.9 | Helped |
| extract low-res feature maps at later timesteps (vs t=1) (ablation) | Ours unsupervised segmentation · ADE20K unsupervised seg | mIoU: 33.1 at t=1 -> 26.9 at t=801 | Hurt |
| modulation timestep choice (early-middle vs very early or very late) (ablation) | Ours unsupervised segmentation · ADE20K unsupervised seg | mIoU: t_m=1: 29.0; 281: 33.1; 481: 33.1; 881: 27.4 | Mixed |
| modulation strength lambda 1 to 1000 (ablation) | Ours unsupervised segmentation · ADE20K unsupervised seg | mIoU: 32.7 / 33.1 / 32.8 / 32.6 | No clear effect |
| choose first vs second vs third 16x16 cross-attention layer for feature extraction (ablation) | Ours unsupervised segmentation · ADE20K unsupervised seg | mIoU: 33.1 / 33.1 / 32.8 | No clear effect |
| number of masks per image (10 to 40) (ablation) | Ours and MaskCLIP+Ours, TCL+Ours · ADE20K / CS171 / City19 | mIoU: Ours AD150 30.9 to 33.4 across 10-40 masks | No clear effect |
Surprising findings
- Under the traditional Hungarian-matching evaluation, the method only performs on par with DINO-based STEGO, which the authors attribute to limitations of that protocol rather than the masks themselves.
- Combining with CLIPpy gives only marginal gains, which the authors attribute to over-smoothed pixel embeddings.
Limitations
- Struggles to distinguish extremely small objects such as animal legs and human faces.
- Feature representations mix object semantics with attributes like location and color, causing over-segmentation of sky and ground.
- Only evaluated with Stable Diffusion; generalization to other generative models is left to future work.
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
- Seg4Diff: Unveiling Open-Vocabulary Semantic Segmentation in Text-to-Image Diffusion Transformers (NeurIPS 2025 · similarity 0.90)
- Diffusion-Driven Two-Stage Active Learning for Low-Budget Semantic Segmentation (NeurIPS 2025 · similarity 0.89)
- VidSeg: Training-free Video Semantic Segmentation based on Diffusion Models (CVPR 2025 · similarity 0.89)
- High-Precision Dichotomous Image Segmentation via Probing Diffusion Capacity (ICLR 2025 · similarity 0.89)
- $\mathbf{D^{3}S^{2}}$: Diffusion-Guided Dataset Distillation for Semantic Segmentation (NeurIPS 2026 · similarity 0.89)
- Unleashing the Potential of the Diffusion Model in Few-shot Semantic Segmentation (NeurIPS 2024 · similarity 0.88)
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.
