atlas

Research map / Image generation / Fast diffusion sampling

On the Stability of Iterative Retraining of Generative Models on their own Data

ICLR 2024 · Image generation › Fast diffusion sampling · approach: Distributional control of diffusion · 2 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 maximum-likelihood theory of iterative retraining of generative models on mixtures of real and self-generated data, proving stability when real data proportion is large enough and the initial model is close to the data distribution, and validates it with OTCFM, DDPM and EDM on CIFAR-10 and FFHQ-64.

Problem

Self-consuming generative models retrained on web data containing their own synthetic outputs may collapse; the paper asks whether training on mixed finite real and self-generated data changes performance and stability.

Approach

Frames iterative retraining as a maximum-likelihood fixed-point map over parameters with a real-data term and a synthetic-data regularizer weighted by lambda; shows local stability (Jacobian norm < 1) when lambda(1+L*eps/alpha) < 1/2 and the model is close to the data distribution, and extends to finite-sample error with a neighborhood bound.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Retrain exclusively on generated data instead of mixing in real dataIterative generative models · CIFAR-10 and FFHQ-64FIDHurt
Use a sufficiently small fraction of generated data mixed with real data at each retraining stepEDM · CIFAR-10 and FFHQ-64FIDNo clear effect
Retrain with equal amounts of generated and original real data rather than only real dataEDM · FFHQ-64sample qualityNo clear effect
Iteratively retrain using real and generated dataEDM · CIFAR-10 and FFHQ-64precisionHurt
Iteratively retrain using real and generated dataEDM · CIFAR-10 and FFHQ-64recallHelped
Increase the fraction of generated data in the retraining mixtureOTCFM and EDM · CIFAR-10 and FFHQ-64FIDHurt
Iteratively retrain using real and generated dataOTCFM · CIFAR-10precisionMixed

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 FFHQ-64 8 Gaussians (synthetic) two moons (synthetic)

Benchmarks: CIFAR-10 FFHQ-64

Metrics: FID precision recall

Baselines: OTCFM DDPM EDM

Builds on: denoising diffusion probabilistic models elucidating diffusion models conditional flow matching

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