atlas

Research map / Image generation / Fast diffusion sampling

The Emergence of Reproducibility and Consistency in Diffusion Models

ICML 2024 · Image generation › Fast diffusion sampling · approach: Diffusion theory and design space

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.

Shows that diffusion models trained on the same data reproduce the same noise-to-image mapping under deterministic samplers, with this reproducibility arising from memorization (empirical score) or generalization (true distribution) regimes depending on training data size.

Problem

Why different diffusion models given the same initial noise and a deterministic sampler produce remarkably similar outputs, and what distribution they learn in different training data size regimes.

Approach

Introduces "consistent model reproducibility" with RP and GL scores, and shows it arises in two regimes: memorization (models converge to the empirical multi-delta score) and generalization (models converge to the underlying data score), with extensions to conditional models, inverse problems and fine-tuning.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train a second diffusion model on samples generated by a pretrained modelEDM on CIFAR-10 · CIFAR-10RP score: RP Score=0.80Helped
Compare different unconditional diffusion models using the same initial noiseUnconditional diffusion models · CIFAR-10RP score: 0.7Helped
Compare different unconditional diffusion models using the same initial noiseUnconditional diffusion models · CIFAR-10MAE score: 0.68Helped
Train on the same LAION-5B subset (v1-3 and v1-4 share the subset)Stable Diffusion · Prompt: “a photograph of an astronaut riding a horse”Reproducibility score: 0.63Helped
Use sequentially trained Stable Diffusion versions with overlapping datasetsStable Diffusion v1-1, v1-2, and v1-3 · Prompt: “a photograph of an astronaut riding a horse”Reproducibility score: below 0.21Helped
Change the initial noise while keeping the prompt fixedStable Diffusion · Prompt: “a photograph of an astronaut riding a horse”Reproducibility scoreHurt
Compare models with different network architecture typesDiffusion models solving inverse problems with DPS · CIFAR-10 image inpaintingRP scoreHurt
Make the inpainting task more challengingDiffusion models solving inverse problems with DPS · CIFAR-10 image inpaintingReproducibility scoreHurt

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 CIFAR-100 ImageNet MNIST LAION-5B

Benchmarks: CIFAR-10 ImageNet

Metrics: RP score GL score MAE score score matching loss L_score

Baselines: GAN (wGAN, SNGAN) VAE (VAMP)

Builds on: DDPM score-based SDE EDM DPS Progressive distillation

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