Not All Prediction Targets Keep Training-Free Diffusion Guidance on the Manifold

2026-07-01Computer Vision and Pattern Recognition

Computer Vision and Pattern Recognition
AI summary

The authors study ways to guide pretrained diffusion models to create images with specific traits without extra training. They show that predicting the clean image directly (x-prediction) during the noisy sampling process is more accurate than estimating it indirectly, which helps keep the generated images realistic. They support this with new theoretical insights and a special metric called guided-class FID to detect quality issues missed by usual tests. Experiments on bird images and style transfers confirm that x-prediction best maintains image quality during guidance.

diffusion modelstraining-free guidancex-predictionepsilon-predictionv-predictionimage manifoldFID scorefine-grained benchmarkstyle transfersampling steps
Authors
Yunsung Lee, Hyeongmin Lee
Abstract
Training-free guidance (TFG) steers a pretrained diffusion model toward a desired attribute at inference. To be effective, this guidance must be applied from the earliest, high-noise steps of sampling. Because its objective (a classifier or energy) is defined on clean images, $ε$- and $v$-prediction models must first estimate the clean image $\hat{x}$ from the noisy state at each step, and the accuracy of that estimate determines how easily guidance drifts off the data manifold. $x$-prediction, a recent alternative, outputs the clean image directly, removing this source of error even at high noise. This is our motivation. We provide a theoretical analysis of how each prediction target shapes this accuracy, and introduce guided-class FID (Child FID), a metric that exposes the manifold damage standard evaluation misses. Experiments on a new fine-grained bird benchmark and on style transfer confirm that $x$-prediction keeps guided samples on the manifold most reliably, making it the strongest foundation for training-free guidance. Code is available at https://github.com/ManLuML/on-manifold-tfg