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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Generation | MS-COCO 10K (val) | CLIPScore31.9586 | 15 | |
| Text-to-Image Generation | MS-COCO prompts 2014 (test) | CLIPScore31.56 | 10 | |
| Text-to-Image Generation | Human Preference Study | Alignment & Quality Score33.55 | 6 | |
| Text-to-Image Generation | User Study 15 image sets | Text Alignment4.2145 | 5 | |
| Text-to-Image Generation | SD 1.5 | CS31.2724 | 3 | |
| Human Preference Evaluation | minority samples | ImageReward-0.026 | 3 | |
| Minority-oriented image generation | SD 1.5 | CLIPScore30.4351 | 3 |