Not All Prediction Targets Keep Training-Free Diffusion Guidance on the Manifold
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 -prediction models must first estimate the clean image from the noisy state at each step, and the accuracy of that estimate determines how easily guidance drifts off the data manifold. -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 -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
Keywords
Cite
@article{arxiv.2607.00647,
title = {Not All Prediction Targets Keep Training-Free Diffusion Guidance on the Manifold},
author = {Yunsung Lee and Hyeongmin Lee},
journal= {arXiv preprint arXiv:2607.00647},
year = {2026}
}
Comments
Accepted to ECCV 2026. 15-page main paper with appendix (48 pages total, 14 figures). Project page: https://manluml.github.io/on-manifold-tfg