atlas

Research map / Image generation / Fast diffusion sampling

On Error Propagation of Diffusion Models

ICLR 2024 · Image generation › Fast diffusion sampling · approach: Fast diffusion ODE solvers · 1 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.

Develops a theoretical framework for error propagation in diffusion models (modular error, cumulative error, propagation equation) and uses a bootstrapped upper bound of cumulative error (MMD-based) as a regularizer, improving DDPM FID on CIFAR-10, ImageNet and CelebA.

Problem

Diffusion models are sequential chains that may suffer error propagation (related to exposure bias), but it is unclear whether and why this affects them, and how to reduce it.

Approach

Formalizes modular and cumulative KL errors and a propagation inequality E_t^cumu >= E_{t+1}^cumu + E_t^mod showing amplification factor >= 1 in standard DMs; regularizes training with an MMD-based upper bound of the cumulative error, estimated efficiently via bootstrapped (TD-style) warm-started backward chains.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add cumulative-error regularization with bootstrappingDDPM · CIFAR-10FID: 3.61 -> 2.93Helped
Add cumulative-error regularization with bootstrappingDDPM · ImageNetFID: 3.62 -> 2.55Helped
Add cumulative-error regularization with bootstrappingDDPM · CelebAFID: 1.73 -> 1.22Helped
Add cumulative-error regularization instead of ADM-IP input perturbationDDPM · CIFAR-10FID: 3.25 -> 2.93Helped
Add cumulative-error regularization instead of ADM-IP input perturbationDDPM · ImageNetFID: 2.72 -> 2.55Helped
Add cumulative-error regularization instead of ADM-IP input perturbationDDPM · CelebAFID: 1.31 -> 1.22Helped
Add an exponential loss-weight schedule rather than equal weights (ablation)DDPM w/ proposed regularization · CIFAR-10FID: 3.12 -> 2.93Helped
Add an exponential loss-weight schedule rather than equal weights (ablation)DDPM w/ proposed regularization · CelebAFID: 1.35 -> 1.22Helped
Add warm-start initialization with L=5 (ablation)DDPM w/ proposed regularization · CIFAR-10FID: 3.38 -> 2.93Helped
Add warm-start initialization with L=5 (ablation)DDPM w/ proposed regularization · CelebAFID: 1.45 -> 1.22Helped
Add warm-start initialization with L=7 (ablation)DDPM w/ proposed regularization · CIFAR-10FID: 3.15 -> 2.93Helped
Add warm-start initialization with L=7 (ablation)DDPM w/ proposed regularization · CelebAFID: 1.27 -> 1.22Helped

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 ImageNet 32x32 CelebA 64x64

Benchmarks: CIFAR-10 ImageNet CelebA

Metrics: FID

Baselines: ADM-IP Consistent DM FP-Diffusion DDPM

Builds on: DDPM ADM exposure bias analysis

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