Spotlight Poster
TFG: Unified Training-Free Guidance for Diffusion Models
Haotian Ye · Haowei Lin · Jiaqi Han · Minkai Xu · Sheng Liu · Yitao Liang · Jianzhu Ma · James Zou · Stefano Ermon
Given an unconditional diffusion model and a predictor for a target property of interest (e.g., a classifier), the goal of training-free guidance is to generate samples with desirable target properties without additional training. Existing methods, though effective in various individual applications, often lack theoretical grounding and rigorous testing on extensive benchmarks. As a result, they could even fail on simple tasks, and applying them to a new problem becomes unavoidably difficult. This paper introduces a novel algorithmic framework that encompasses existing methods as special cases, unifying the study of training-free guidance into the analysis of an algorithm-agnostic design space. Via theoretical and empirical investigation, we propose an efficient and effective hyper-parameter searching strategy that can be readily applied to any downstream task. We systematically benchmark training-free guidance across 6 diffusion models on 14 tasks with 38 targets, and achieve a 7.4% performance improvement on average. We believe our framework and benchmark offer a solid foundation for future research in this area.
Live content is unavailable. Log in and register to view live content