atlas

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

DreamIdentity: Enhanced Editability for Efficient Face-Identity Preserved Image Generation

AAAI 2024 · Image generation › Text-to-image diffusion · approach: Multi-subject personalization · 14 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 DreamIdentity, an optimization-free face-identity encoder (M^2 ID encoder with multi-scale ViT features and multi-word projection) plus self-augmented editability learning from celebrity images generated by Stable Diffusion, enabling identity-preserving text-to-image generation in 0.04 s per face.

Problem

Preserving a conditioned face's identity while following text prompts in text-to-image generation, without per-identity optimization and without harming the editability of the pre-trained T2I model.

Approach

Replace the CLIP last-layer feature with a face-recognition ViT encoder whose multi-layer CLS features are projected to multiple pseudo-word embeddings, and train the encoder with a self-generated celebrity editing dataset (generated by the T2I model itself) so that editing ability transfers to unseen faces, while the T2I model's weights stay frozen.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace the CLIP encoder with a face-specific ID encoder (ablation)DreamIdentity · Face identity re-contextualization test setFace-similarity: 0.266 to 0.302Helped
Add multi-scale features to the face-specific ID encoder (ablation)DreamIdentity · Face identity re-contextualization test setFace-similarity: 0.302 to 0.412Helped
Add multi-word embeddings (2 embeddings) (ablation)DreamIdentity with multi-scale ID features · Face identity re-contextualization test setFace-similarity: +12%Helped
Add multi-word embeddings (2 embeddings) (ablation)DreamIdentity with multi-scale ID features · Face identity re-contextualization test setText-alignment: 0.4%No clear effect
Increase the number of word embeddings beyond 2 (ablation)DreamIdentity · Face identity re-contextualization test setText-alignment: 17%Hurt
Train with self-augmented editability learning instead of reconstruction objective alone (ablation)DreamIdentity · Face identity re-contextualization test setText-alignmentHelped
Remove reconstruction learning on FFHQ and train only with the self-augmented editing dataset (ablation)DreamIdentity · Face identity re-contextualization test setFace-similarityHurt
Use DreamIdentity instead of E4TFace identity re-contextualization test set · Face identity re-contextualization test setText-alignment: 7%Helped

Surprising findings

Limitations

Experimental setup

Datasets: FFHQ self-augmented celebrity editing dataset

Benchmarks: FFHQ test faces (100 faces)

Metrics: text-alignment (CLIP cosine) face similarity (ArcFace cosine) encoding time

Baselines: Textual Inversion DreamBooth E4T ELITE InstructPix2Pix

Builds on: Textual Inversion ELITE Latent Diffusion Models

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