atlas

Research map / Image generation / Fast diffusion sampling

Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Models

NeurIPS 2024 · Image generation › Fast diffusion sampling · approach: Guidance for conditional diffusion · 10 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.

Limits classifier-free guidance to a middle interval of noise levels (sigma_lo, sigma_hi] instead of all steps, improving FID on ImageNet-512 with EDM2 (1.81 to 1.40) and sample quality/speed on SD-XL.

Problem

Classifier-free guidance is applied with a constant weight across all sampling steps, but it is harmful at high noise levels (reducing diversity, mode dropping) and largely unnecessary at low noise levels, so constant guidance is suboptimal for image quality and sampling cost.

Approach

Apply guidance only within a continuous interval of noise levels in the middle of the sampling chain (disabling it elsewhere, with the weight set to 1), exposing the guidance interval as a sampler hyperparameter; the interval endpoints are aligned to step boundaries so the ODE remains smooth within each step.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
restrict guidance to a middle noise-level intervalEDM2-S · ImageNet-512FID: 2.23 to 1.68Helped
restrict guidance to a middle noise-level intervalEDM2-XXL · ImageNet-512FID: 1.81 to 1.40Helped
restrict guidance to a middle noise-level intervalEDM2-XXL · ImageNet-512FD-DINO: 33.09 to 29.16Helped
restrict guidance to a middle noise-level intervalEDM2-S · ImageNet, 16 sampling stepsFID: 2.49 to 1.84Helped
restrict guidance to a middle noise-level intervalEDM2-S · ImageNet, 64 sampling stepsFID: 2.27 to 1.70Helped
apply guidance at 50% of sampling stepsStable Diffusion XL · Stable Diffusion XL samplingspeed-up: over 20%Helped
include higher noise levels in the guidance interval (ablation)limited guidance interval · ImageNetFIDHurt
narrow the guidance interval by lowering its upper limit (ablation)limited guidance interval · ImageNetFIDHurt
apply guidance at low noise levels (ablation)limited guidance interval · ImageNetFIDNo clear effect
replace binary interval guidance with smooth weighting functions (ablation)limited guidance interval · ImageNetFIDNo clear effect
estimate guidance importance by toggling it at one sampling step at a time (ablation)limited guidance interval · ImageNetguidance-interval sensitivityHurt
disable guidance at high noise levels (ablation)Stable Diffusion XL · Stable Diffusion XL samplesimage composition and color saturationMixed

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet

Benchmarks: ImageNet 512x512

Metrics: FID FD_DINOv2 precision recall

Baselines: EDM2-S EDM2-XXL DiT-XL/2 SD-XL standard classifier-free guidance dynamic CFG

Builds on: classifier-free guidance EDM EDM2 Heun sampler DiT

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