atlas

Research map / Image generation / Text-to-image diffusion

The Hidden Language of Diffusion Models

ICLR 2024 · Image generation › Text-to-image diffusion · approach: Multi-concept composition · 4 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.

Conceptor decomposes a textual concept's representation in Stable Diffusion into a sparse weighted combination of interpretable vocabulary tokens learned by an MLP trained with a denoising objective.

Problem

Text-to-image diffusion models learn internal concept representations that are opaque; existing concept-based interpretability methods (CNN-based, prompt tuning, captioning) do not produce faithful, meaningful decompositions for generative diffusion models.

Approach

Learn a pseudo-token as a linear combination of top vocabulary token embeddings, with coefficients produced by a 2-layer MLP applied to each word embedding, trained with the diffusion denoising loss on concept images plus a sparsity loss that keeps the top-50 tokens representative; per-image decomposition is done by iterative token removal checked with CLIP similarity.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace the MLP with directly learned coefficient weights (ablation)Conceptor · ProfessionsCLIP pairwise: 87.0 ± 5.5 -> 78.0 ± 6.7Hurt
Remove the sparsity loss (ablation)Conceptor · ProfessionsCLIP pairwise: 87.0 ± 5.5 -> 80.3 ± 11.6Hurt
Reduce the number of decomposition tokens to 10 (ablation)Conceptor · ProfessionsCLIP pairwise: 87.0 ± 5.5 -> 82.9 ± 7.8Hurt
Increase the number of decomposition tokens to 100 (ablation)Conceptor · ProfessionsCLIP pairwise: 87.0 ± 5.5 -> 85.6 ± 6.9No clear effect
Select the top 50 tokens by CLIP similarity to the mean image (ablation)Conceptor · ProfessionsCLIP pairwise: 87.0 ± 5.5 -> 80.1 ± 9.9Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 classes Bias in Bios professions ConceptNet concepts Stable Diffusion-generated concept images Best 30 Stable Diffusion Prompts website prompts

Benchmarks: 188-concept diffusion interpretability suite (CIFAR-10, Bias in Bios, emotions, actions, complex prompts, ConceptNet)

Metrics: CLIP pairwise similarity LPIPS FID per concept FID entire set token diversity (SentenceBERT) user study preference

Baselines: PEZ BLIP-2 sentence BLIP-2 token PCA NMF k-means

Builds on: Textual Inversion PEZ TCAV CRAFT

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