atlas

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

AP-Adapter: Improving Generalization of Automatic Prompts on Unseen Text-to-Image Diffusion Models

NeurIPS 2024 · Image generation › Text-to-image diffusion · approach: Inference-time T2I improvement · 0 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 MGAPO, a task of training automatic prompt optimization on known Stable Diffusion checkpoints to generalize to unseen ones, and AP-Adapter, a two-stage method (GPT-3.5 in-context keyword rewriting plus a prototype-anchored adapter over CLIP text embeddings) evaluated on a newly collected CivitAI dataset.

Problem

Automatic prompt optimization methods for text-to-image generation are typically trained for a fixed diffusion model and generalize poorly to newly emerging fine-tuned checkpoints with different parameters.

Approach

Treat each SD checkpoint as a domain, build domain prototypes from channel-dropped CLIP image features that capture inter-model differences, and use similarity-weighted prototypes as anchors to adapt prompt text embeddings toward manual-prompt representations so they transfer to unseen checkpoints.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataColor: 0.472 -> 0.477Helped
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataShape: 0.413 -> 0.422Helped
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataTexture: 0.449 -> 0.452Helped
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataBlipscore: 0.325 -> 0.332Helped
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataAesthetic Score: 6.249 -> 6.384Helped
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataImageReward: 0.286 -> 0.427Helped
Use AP-Adapter instead of the strongest reported baselineAutomatic prompt optimization on unseen diffusion models · Target domain dataHPS: +0.005 relative to AKP (0.213 -> 0.218)Helped
Add denoising loss Ld to Lr (ablation)AP-Adapter · Target domain dataBlipScore: 0.303 -> 0.313Helped
Add denoising loss Ld to Lr (ablation)AP-Adapter · Target domain dataAesScore: 6.110 -> 6.139Helped
Add domain style distillation loss La to Lr and Ld (ablation)AP-Adapter · Target domain dataImageReward: 0.258 -> 0.384Helped
Add domain discrimination loss Lc to Lr, Ld, and La (ablation)AP-Adapter · Target domain dataImageReward: 0.384 -> 0.427Helped
Add learned domain prototypes (ablation)AP-Adapter · Target domain dataImageReward: 0.258 -> 0.427Helped

Surprising findings

Limitations

Experimental setup

Datasets: CivitAI community images and SD checkpoints (47,695 collected, 25,395 filtered) LLaVA-generated natural language descriptions

Benchmarks: Custom CivitAI-based target domain (3,064 samples, checkpoints ranked 41-100) T2I-CompBench attribute binding prompts (color, shape, texture)

Metrics: BLIPScore color/shape/texture attribute binding score LAION aesthetic score ImageReward HPS human win-rate for aesthetics and text-image alignment

Baselines: MagicPrompt PromptPerfect Promptist SUR-adapter AKP + SUR-adapter Manual prompts

Builds on: SUR-adapter DomainDrop Promptist Stable Diffusion 1.5

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