atlas

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

HyperDreamBooth: HyperNetworks for Fast Personalization of Text-to-Image Models

CVPR 2024 · Image generation › Text-to-image diffusion · approach: Multi-subject personalization · 117 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 HyperDreamBooth, a hypernetwork (ViT-H encoder + transformer decoder) that predicts compact LoRA-style personalization weights (LiDB) from one face image, followed by ~40 steps of rank-relaxed fast finetuning, for fast face personalization of Stable Diffusion.

Problem

DreamBooth-style text-to-image personalization is slow (minutes per subject) and produces large per-subject models (over 1GB), limiting practical use.

Approach

A hypernetwork with a ViT image encoder and an iterative transformer decoder predicts Lightweight DreamBooth weights (a random orthogonal incomplete basis inside rank-1 LoRA space, ~30K variables, ~120KB) from a single face image; these are then refined with rank-relaxed LoRA finetuning for about 40 iterations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use HyperDreamBooth instead of DreamBoothStable Diffusion 1.5 face personalization · 100 CelebAHQ identities, 30 promptsDINO: 0.441 -> 0.473Helped
Use HyperDreamBooth instead of DreamBoothStable Diffusion 1.5 face personalization · 100 CelebAHQ identities, 30 promptsCLIP-T: 0.282 -> 0.286Helped
Remove the HyperNetwork at test time (ablation)HyperDreamBooth · HyperDreamBooth ablationDINO: 0.473 -> 0.392Hurt
Remove the HyperNetwork at test time (ablation)HyperDreamBooth · HyperDreamBooth ablationCLIP-T: 0.286 -> 0.299Helped
Use HyperNetwork prediction without fast finetuning (ablation)HyperDreamBooth · HyperDreamBooth ablationDINO: 0.473 -> 0.414Hurt
Disable iterative prediction (k=1) (ablation)HyperDreamBooth · HyperDreamBooth ablationDINO: 0.473 -> 0.464Hurt
Reduce DreamBooth training to 400 iterationsDreamBooth · DreamBooth comparisonDINO: 0.441 -> 0.323Hurt
Reduce DreamBooth training to 40 iterationsDreamBooth · DreamBooth comparisonDINO: 0.441 -> 0.360Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CelebA-HQ SFHQ

Benchmarks: CelebA-HQ identities with 30 prompts (30,000 samples)

Metrics: Face Recognition similarity DINO CLIP-I CLIP-T user preference

Baselines: DreamBooth Textual Inversion E4T LoRA DreamBooth InstantBooth

Builds on: DreamBooth LoRA Stable Diffusion Textual Inversion E4T

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