Research map / Image generation / Fast diffusion sampling
Understanding and Improving Training-free Loss-based Diffusion Guidance
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
- On CelebA-HQ, the proposed method reaches FID 33.31 on segmentation guidance versus 38.65 for FreeDoM, and 41.26 on sketch versus 52.18 for FreeDoM.
- On unconditional ImageNet with zero-shot CLIP text guidance, the method achieves CLIP score 27.7 versus 25.9 for FreeDoM and 25.7 for UG.
- On MDM targeting with object avoidance, the method reduces the 'Backwards' loss to 0.68+1.32 from 1.09+6.63 for FreeDoM.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · CelebA-HQ segmentation maps | Distance: 1657.0 -> 1575.7 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · CelebA-HQ segmentation maps | FID: 38.65 -> 33.31 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · CelebA-HQ sketches | Distance: 34.21 -> 30.41 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · CelebA-HQ sketches | FID: 47.20 -> 41.26 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · CelebA-HQ text | Distance: 10.78 -> 10.72 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · CelebA-HQ text | FID: 41.74 -> 41.25 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · ImageNet text guidance | CLIP score: 25.9 -> 27.7 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · Human motion: Backwards | Loss: 0.98+6.48 -> 0.68+1.32 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · Human motion: Balanced Beam | Loss: 4.42+0.02 -> 1.13+0.30 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · Human motion: Walking | Loss: 1.30+0.39 -> 0.43+0.31 | Helped |
| add Polyak step size and random augmentation to FreeDoM | FreeDoM · Human motion: Jogging | Loss: 6.12+2.38 -> 2.93+1.15 | Helped |
| use random augmentation with |T|=10 rather than no random augmentation (ablation) | FreeDoM · ImageNet text guidance | CLIP score: 0.541 -> 0.571 | Helped |
Surprising findings
- Off-the-shelf time-independent classifiers yield gradients that are visually unrecognizable, indicating adversarial gradients, whereas time-dependent classifiers produce recognizable gradients.
- Training-free guidance can reach low loss even when guidance fails to produce the target condition, attributed to adversarial gradients.
Limitations
- Refined training-free guidance still requires more NFEs than classifier-free guidance because adversarial gradients cannot be eliminated without training.
- MPGD cannot be applied to motion diffusion without an auto-encoder, and MPGD without projection fails on motion tasks.
- Positional text prompts (e.g., 'left to', 'below') are not satisfied by any method due to CLIP 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
- Adaptive Guidance: Training-free Acceleration of Conditional Diffusion Models (AAAI 2025 · similarity 0.91)
- REG: Rectified Gradient Guidance for Conditional Diffusion Models (ICML 2025 · similarity 0.90)
- Feedback Guidance of Diffusion Models (NeurIPS 2025 · similarity 0.90)
- Decoupling Training-Free Guided Diffusion by ADMM (CVPR 2025 · similarity 0.89)
- Universal Guidance for Diffusion Models (ICLR 2024 · similarity 0.89)
- Multilevel and Sequential Monte Carlo for Training-Free Diffusion Guidance (NeurIPS 2026 · similarity 0.89)
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.
