atlas

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

Non-confusing Generation of Customized Concepts in Diffusion Models

ICML 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Recaptioning and text-supervised pretraining · 1 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 CLIF, which fine-tunes the CLIP text encoder contrastively on over-segmented, augmented customized-concept data before fine-tuning the diffusion U-Net with a shared LoRA, to reduce inter-concept confusion in multi-concept customization.

Problem

Inter-concept visual confusion (identity loss, attribute leakage, concept missing) when composing multiple user-provided customized concepts in text-guided diffusion models, especially with few-shot concept examples.

Approach

Revisits the two TGDM stages and argues prior customization methods only fine-tune the decoder; CLIF instead contrastively fine-tunes the CLIP text encoder on augmented image-text pairs (global, region, and mixed augmentations built from SAM segmentation, generated contexts, and GPT-4 captions) so concept token embeddings are decoupled, then fine-tunes the U-Net with a shared LoRA.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply CLIF instead of Mix-of-ShowMulti-concept generation · Multi-concepttext alignment: 0.526 -> 0.599Helped
Apply CLIF instead of Mix-of-ShowMulti-concept generation · Multi-conceptimage alignment: 0.713 -> 0.761Helped
Add CLIF to TITI · Multi-concepttext alignment: 0.507 -> 0.528Helped
Add CLIF to CustomCustom · Multi-concepttext alignment: 0.511 -> 0.535Helped
Add CLIF to CustomCustom · Single-conceptimage alignment: 0.749 -> 0.774Helped
Add G-Aug (ablation)CLIF · Multi-conceptimage alignment: 0.729 -> 0.761Helped
Add R-Aug (ablation)CLIF · Multi-conceptimage alignment: 0.753 -> 0.761Helped
Add R-Aug (ablation)CLIF · Single-conceptimage alignment: 0.774 -> 0.787Helped
Add M-Aug (ablation)CLIF · Multi-conceptimage alignment: 0.740 -> 0.761Helped
Add M-Aug (ablation)CLIF · Multi-concepttext alignment: 0.579 -> 0.599Helped

Surprising findings

Limitations

Experimental setup

Datasets: 18 curated customized characters (real-world, 3D-animated, 2D-animated) images generated by pre-trained diffusion model for general contexts

Benchmarks: custom 18-character multi-concept set DreamBench

Metrics: text alignment (CLIP) image alignment (CLIP) attendance

Baselines: Text-Inversion Custom Diffusion DreamBooth Mix-of-Show PACGen

Builds on: CLIP Stable Diffusion Text Inversion Custom Diffusion DreamBooth Mix-of-Show

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