atlas

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

FairRAG: Fair Human Generation via Fair Retrieval Augmentation

CVPR 2024 · Image generation › Text-to-image diffusion · approach: Fairness and bias in generative models · 7 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.

Introduces FairRAG, which conditions a frozen Stable Diffusion on demographically balanced reference images retrieved from an external database via a trained linear projector, improving demographic diversity in human image generation.

Problem

Text-to-image diffusion models reflect and amplify demographic biases (age, gender, skin tone) in human image generation, and prior debiasing methods do not substantially improve diversity.

Approach

Condition a frozen pre-trained latent diffusion model on retrieved real human reference images via a single linear projector mapping CLIP image embeddings into the text token space, and debias retrieval at inference with a debiased query, balanced group sampling over retrieved candidates, and a transfer instruction.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add fair retrieval plus transfer instruction on top of linear-conditioned RAG (full FairRAG vs Base RAG)Stable Diffusion v2.1 RAG · Human generation promptsIntersectional diversity: 0.374 -> 0.438Helped
Remove debiased query from FairRAG (ablation)FairRAG · Human generation promptsIntersectional diversity: 0.438 -> 0.414Hurt
Remove balanced sampling from FairRAG (ablation)FairRAG · Human generation promptsIntersectional diversity: 0.438 -> 0.420Hurt
Remove text transfer instruction from FairRAG (ablation)FairRAG · Human generation promptsIntersectional diversity: 0.438 -> 0.407Hurt
Remove text transfer instruction from FairRAG (ablation)FairRAG · Human generation promptsSkin tone diversity: 0.416 -> 0.416No clear effect
Explicit demographic text in prompt (TextAug) vs SDv2.1Stable Diffusion v2.1 · Human generation promptsIntersectional diversity: 0.188 -> 0.341Helped
Ethical phrase prompt augmentation (Interven)Stable Diffusion v2.1 · Human generation promptsCLIP score: 0.142 -> 0.132Hurt
Semantic guidance with uniform group prior (FairDiff)Stable Diffusion v2.1 · Human generation promptsIntersectional diversity: 0.188 -> 0.196No clear effect
Linear-conditioned reference retrieval (Base RAG) vs SDv2.1Stable Diffusion v2.1 · Human generation promptsFID: 85.3 -> 49.4Helped

Surprising findings

Limitations

Experimental setup

Datasets: MSCOCO OpenImages-v6 FairFace (group definitions)

Benchmarks: custom profession prompt set (8 categories)

Metrics: normalized entropy demographic diversity (intersectional, ag CLIP score FID

Baselines: SDv2.1 Interven FairDiff TextAug Base RAG

Builds on: Stable Diffusion Re-Imagen Fair Diffusion Ethical Interventions CLIP

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