atlas

Research map / Image generation / Fast diffusion sampling

Understanding and Improving Training-free Loss-based Diffusion Guidance

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

Analyzes training-free loss-based diffusion guidance from an optimization view, shows it is more sensitive to adversarial gradients and converges slower than classifier guidance, and proposes random augmentation plus Polyak step size to improve it on image and motion diffusion.

Problem

Training-free loss-based guidance for pretrained diffusion models uses off-the-shelf clean-image networks, but its mechanism is poorly understood and it suffers from adversarial gradients and slow convergence when the condition differs from unconditional samples.

Approach

Views training-free guidance as gradient descent on a guidance loss (PL-condition analysis), shows off-the-shelf time-independent networks are less smooth/robust than time-dependent classifiers, and mitigates this with differentiable data-augmentation averaging (random augmentation) and an adaptive Polyak step size.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add Polyak step size and random augmentation to FreeDoMFreeDoM · CelebA-HQ segmentation mapsDistance: 1657.0 -> 1575.7Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · CelebA-HQ segmentation mapsFID: 38.65 -> 33.31Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · CelebA-HQ sketchesDistance: 34.21 -> 30.41Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · CelebA-HQ sketchesFID: 47.20 -> 41.26Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · CelebA-HQ textDistance: 10.78 -> 10.72Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · CelebA-HQ textFID: 41.74 -> 41.25Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · ImageNet text guidanceCLIP score: 25.9 -> 27.7Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · Human motion: BackwardsLoss: 0.98+6.48 -> 0.68+1.32Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · Human motion: Balanced BeamLoss: 4.42+0.02 -> 1.13+0.30Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · Human motion: WalkingLoss: 1.30+0.39 -> 0.43+0.31Helped
add Polyak step size and random augmentation to FreeDoMFreeDoM · Human motion: JoggingLoss: 6.12+2.38 -> 2.93+1.15Helped
use random augmentation with |T|=10 rather than no random augmentation (ablation)FreeDoM · ImageNet text guidanceCLIP score: 0.541 -> 0.571Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet CelebA-HQ text-motion pairs (MDM training corpus)

Benchmarks: CelebA-HQ segmentation/sketch/text guidance ImageNet zero-shot CLIP text guidance MDM targeting and object avoidance

Metrics: FID distance CLIP score targeting loss object avoidance loss

Baselines: Universal Guidance (UG) LGD-MC FreeDoM MPGD-Z

Builds on: FreeDoM Universal Guidance Loss-Guided Diffusion MPGD classifier 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