atlas

Research map / Image generation / Fast diffusion sampling

Rethinking Peculiar Images by Diffusion Models: Revealing Local Minima’s Role

AAAI 2024 · Image generation › Fast diffusion sampling · approach: Training-free diffusion acceleration · 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.

Proposes adding naive and positive-negative momentum to diffusion sampling (DDIM) to escape local minima, reducing peculiar image generation (e.g., extra or missing limbs) at no extra compute cost.

Problem

Diffusion models frequently generate peculiar images with anatomical artifacts (extra fingers, wrong leg counts), and fast sampling with few steps worsens quality; existing fixes rely on heuristics, extra training, or post-processing.

Approach

Frames diffusion sampling and SGD as instances of generalized EM, hypothesizes peculiar images correspond to local minima, and applies momentum (including positive-negative momentum with negative beta) to the full DDIM update term to escape them without extra network evaluations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add momentum to PLMS sampling at T=50PLMS on CelebA · CelebA 64×64FID: 4.52 -> 2.20Helped
reduce sampling steps from T=250 to T=50 without momentumPLMS on CelebA · CelebA 64×64FID: 2.45 -> 4.52Hurt
add momentum to PLMS sampling at T=50PLMS on CelebA · CelebA 64×64GFLOPS/img: 50×46.74 -> 50×46.74No clear effect
add positive-negative momentum (β=-0.8) to DDIM (ablation)DDIM on CIFAR10 · CIFAR10, linear skip, 50 NFEFID: 7.19 -> 13.24Hurt
add positive-negative momentum (β=-0.8) to DDIM (ablation)DDIM on CIFAR10 · CIFAR10, quadratic skip, 50 NFEFID: 4.53 -> 4.25Helped
add positive-negative momentum (β=-0.8) to DDIM (ablation)DDIM on CelebA · CelebA, linear skip, 50 NFEFID: 5.99 -> 13.75Hurt
add naive momentum (β=0.2) to DDIM (ablation)DDIM on CelebA · CelebA, linear skip, 50 NFEFID: 5.99 -> 3.91Helped
add naive momentum (β=0.2) to PLMS (ablation)PLMS on CelebA · CelebA, linear skip, 50 NFEFID: 4.52 -> 2.20Helped
add positive-negative momentum (β=-0.3) to DPM++ (ablation)DPM++ on CIFAR10 · CIFAR10, quadratic skip, 50 NFEFID: 3.99 -> 3.45Helped
add positive-negative momentum (β=-0.3) to DDIM sampling (ablation)Stable Diffusion · single-person human imagesaverage joint confidence: 0.4351 -> 0.4383Helped
add positive-negative momentum (β=-0.3) to DDIM sampling (ablation)Stable Diffusion · group human imagesaverage joint confidence: 0.1880 -> 0.1991Helped
add positive-negative momentum (β=-0.3) to DDIM sampling (ablation)Stable Diffusion · single-person human images, 5–50 clustersaverage RMSE to nearest cluster center: 0.2823 -> 0.2782; 0.2589 -> 0.2515; 0.2461 -> 0.2416; 0.2407 -> 0.2378; 0.2221 -> 0.2219Mixed

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 CelebA MS COCO

Benchmarks: CIFAR-10 FID CelebA FID

Metrics: FID NFE average joint confidence (OpenPose) RMSE to nearest pose cluster center GFLOPS per image

Baselines: DDPM DDIM PLMS DPM-Solver++

Builds on: DDIM DDPM PLMS DPM-Solver++ SGD with momentum

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