atlas

Research map / Image generation / Fast diffusion sampling

Data-free Distillation of Diffusion Models with Bootstrapping

ICML 2024 · Image generation › Fast diffusion sampling · approach: Fast diffusion ODE solvers

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 BOOT, a data-free distillation method that trains a time-conditioned single-step student of a diffusion model from pure Gaussian noise using bootstrapped Signal-ODE objectives, applied to image benchmarks and text-to-image models.

Problem

Distilling diffusion models into fast single-step generators usually needs either synthetic teacher-generated datasets or real training data, which is costly or unavailable (e.g., for large text-to-image models).

Approach

A student gθ(ϵ, t) maps fixed Gaussian noise to the teacher's intermediate signal state at any time t, trained via finite-difference bootstrapping from a derived Signal-ODE between adjacent timesteps, plus a boundary loss at t_max; no real or synthetic data is needed and no EMA is used.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Distill to a single step with BOOT instead of using 50-step DDIMDiffusion model on FFHQ 64 × 64 · FFHQ 64 × 64FID: 7.6 -> 9.0Hurt
Distill to a single step with BOOT instead of using 10-step DDIMDiffusion model on FFHQ 64 × 64 · FFHQ 64 × 64FID: 18.3 -> 9.0Helped
Distill to a single step with BOOT instead of using 50-step DDIMClass-conditional diffusion model · ImageNet 64 × 64FID: 13.7 -> 12.3Helped
Distill to a single step with BOOT instead of using 10-step DDIMClass-conditional diffusion model · ImageNet 64 × 64FID: 18.3 -> 12.3Helped
Use LPIPS loss instead of L2 loss (ablation)BOOT · CIFAR-10FID: 6.88 -> 4.38Helped
Use BOOT instead of CMSingle-step diffusion distillation · CIFAR-10FID: 3.55 -> 4.38Hurt
Use BOOT instead of PDSingle-step diffusion distillation · CIFAR-10FID: 8.34 -> 4.38Helped
Distill StableDiffusion to one step with BOOT instead of using 50-step DDIMStableDiffusion v2.1-Base · COCO2017 validationCLIP score: 0.262 -> 0.254Hurt
Distill StableDiffusion with BOOT instead of LCMStableDiffusion · DiffusionDBFID: 111.55 -> 30.21Helped
Add the boundary loss (ablation)BOOT · Text-to-image generationgeneration diversityHelped
Use uniform time sampling instead of progressive time training (ablation)BOOT · Text-to-image generationgeneration artifactsHelped

Surprising findings

Limitations

Experimental setup

Datasets: DiffusionDB CIFAR-10 FFHQ ImageNet COCO2017

Benchmarks: CIFAR-10 FFHQ 64x64 ImageNet 64x64 COCO2017 DiffusionDB

Metrics: FID precision recall CLIP score fps

Baselines: DDIM DDPM DPM-solver-2 DEIS Direct distillation DFNO ReFlow Progressive distillation Consistency Model LCM

Builds on: consistency models progressive distillation DDIM EDM classifier-free guidance

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