atlas

Research map / Image generation / Fast diffusion sampling

Don't Play Favorites: Minority Guidance for Diffusion Models

ICLR 2024 · Image generation › Fast diffusion sampling · approach: Diffusion theory and design space · 3 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 minority score (Tweedie-based reconstruction discrepancy) and minority guidance, a classifier-guidance sampler that steers diffusion models toward low-density minority samples without labels.

Problem

Diffusion models are majority-oriented: ancestral sampling mostly yields high-likelihood samples, making generation of minority (low-density) samples inefficient, and existing minority samplers need labels.

Approach

Define a minority score as the discrepancy between a clean sample and its Tweedie-denoised reconstruction (one-shot, cheap to compute, equivalent up to scaling to noise-prediction error), then condition the diffusion sampler on categorized minority-score levels via classifier guidance, enabling label-free minority generation with controllable uniqueness.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use minority guidance instead of the baseline samplerADM on CelebA 64×64 · CelebA 64×64cFID: 27.97 -> 26.98Helped
Use minority guidance instead of the baseline samplerADM on CelebA 64×64 · CelebA 64×64Rec: 0.42 -> 0.34Hurt
Use minority guidance instead of the baseline samplerADM on ImageNet 256×256 · ImageNet 256×256sFID: 9.88 -> 7.82Helped
Use minority guidance instead of the baseline samplerADM on ImageNet 256×256 · ImageNet 256×256cFID: 13.43 -> 13.83Hurt
Use minority guidance instead of the baseline samplerADM on ImageNet 256×256 · ImageNet 256×256Rec: 0.42 -> 0.45Helped
Use minority guidance instead of the baseline samplerADM on LSUN Bedrooms 256×256 · LSUN Bedrooms 256×256cFID: 56.45 -> 41.52Helped
Use minority guidance instead of the baseline samplerADM on LSUN Bedrooms 256×256 · LSUN Bedrooms 256×256sFID: 5.96 -> 6.67Hurt
Use minority guidance instead of the baseline samplerADM on Brain MRI 256×256 · Brain MRI 256×256cFID: 21.71 -> 19.73Helped
Increase the requested minority class score categoryMinority guidance · Generated samplesLOFHelped
Increase classifier scale w while conditioning on minority class l̃ = 0Minority guidance · Generated samplesLOFHurt

Limitations

Experimental setup

Datasets: CelebA CIFAR-10 LSUN-Bedrooms ImageNet CIFAR-10-LT brain MRI (in-house)

Benchmarks: CelebA 64x64 LSUN-Bedrooms 256x256 Brain MRI 256x256 ImageNet 64x64 ImageNet 256x256 CIFAR-10 CIFAR-10-LT

Metrics: AvgkNN LOF Rarity Score cFID sFID Improved Precision Improved Recall

Baselines: BigGAN StyleGAN StyleGAN2-ADA ADM ancestral sampling LDM Sehwag et al.

Builds on: DDPM classifier guidance Tweedie's formula ADM

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