Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Minority-Focused Text-to-Image Generation via Prompt Optimization

About

We investigate the generation of minority samples using pretrained text-to-image (T2I) latent diffusion models. Minority instances, in the context of T2I generation, can be defined as ones living on low-density regions of text-conditional data distributions. They are valuable for various applications of modern T2I generators, such as data augmentation and creative AI. Unfortunately, existing pretrained T2I diffusion models primarily focus on high-density regions, largely due to the influence of guided samplers (like CFG) that are essential for high-quality generation. To address this, we present a novel framework to counter the high-density-focus of T2I diffusion models. Specifically, we first develop an online prompt optimization framework that encourages emergence of desired properties during inference while preserving semantic contents of user-provided prompts. We subsequently tailor this generic prompt optimizer into a specialized solver that promotes generation of minority features by incorporating a carefully-crafted likelihood objective. Extensive experiments conducted across various types of T2I models demonstrate that our approach significantly enhances the capability to produce high-quality minority instances compared to existing samplers. Code is available at https://github.com/soobin-um/MinorityPrompt.

Soobin Um, Jong Chul Ye• 2024

Related benchmarks

TaskDatasetResultRank
Text-to-Image GenerationMS-COCO 10K (val)
CLIPScore31.9586
15
Text-to-Image GenerationMS-COCO prompts 2014 (test)
CLIPScore31.56
10
Text-to-Image GenerationHuman Preference Study
Alignment & Quality Score33.55
6
Text-to-Image GenerationUser Study 15 image sets
Text Alignment4.2145
5
Text-to-Image GenerationSD 1.5
CS31.2724
3
Human Preference Evaluationminority samples
ImageReward-0.026
3
Minority-oriented image generationSD 1.5
CLIPScore30.4351
3
Showing 7 of 7 rows

Other info

Code

Follow for update