Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

DiffusionDB: A Large-scale Prompt Gallery Dataset for Text-to-Image Generative Models

About

With recent advancements in diffusion models, users can generate high-quality images by writing text prompts in natural language. However, generating images with desired details requires proper prompts, and it is often unclear how a model reacts to different prompts or what the best prompts are. To help researchers tackle these critical challenges, we introduce DiffusionDB, the first large-scale text-to-image prompt dataset totaling 6.5TB, containing 14 million images generated by Stable Diffusion, 1.8 million unique prompts, and hyperparameters specified by real users. We analyze the syntactic and semantic characteristics of prompts. We pinpoint specific hyperparameter values and prompt styles that can lead to model errors and present evidence of potentially harmful model usage, such as the generation of misinformation. The unprecedented scale and diversity of this human-actuated dataset provide exciting research opportunities in understanding the interplay between prompts and generative models, detecting deepfakes, and designing human-AI interaction tools to help users more easily use these models. DiffusionDB is publicly available at: https://poloclub.github.io/diffusiondb.

Zijie J. Wang, Evan Montoya, David Munechika, Haoyang Yang, Benjamin Hoover, Duen Horng Chau• 2022

Related benchmarks

TaskDatasetResultRank
Prompt RefinementSD 1.4 (In-distribution)
ImageReward (Anime)0.066
10
Prompt RefinementDeepFloyd IF unseen (test)
ImageReward61.9
10
Prompt RefinementSDXL unseen v1.0 (test)
ImageReward0.861
10
Prompt RefinementSUR-adapter unseen (test)
ImageReward0.573
10
Prompt RefinementReFL unseen (test)
ImageReward0.406
10
Prompt RefinementUnseen Systems Aggregated (test)
Relevance1.67
5
Showing 6 of 6 rows

Other info

Follow for update