Oral
in
Workshop: Mathematics of Modern Machine Learning (M3L)
Classifier-Free Guidance is a Predictor-Corrector
Arwen Bradley · Preetum Nakkiran
Keywords: [ Langevin ] [ SDE ] [ guidance ] [ score ] [ diffusion ] [ theory ]
[
Abstract
]
[ Project Page ]
presentation:
Mathematics of Modern Machine Learning (M3L)
Sat 14 Dec 8:50 a.m. PST — 5 p.m. PST
[
OpenReview]
Sat 14 Dec 10:30 a.m. PST
— 10:45 a.m. PST
Sat 14 Dec 8:50 a.m. PST — 5 p.m. PST
Abstract:
We investigate the theoretical foundations of classifier-free guidance (CFG). CFG is the dominant method of conditional sampling for text-to-image diffusion models, yet unlike other aspects of diffusion, it remains on shaky theoretical footing. In this paper, we disprove common misconceptions, by showing that CFG interacts differently with DDPM and DDIM, and neither sampler with CFG generates the gamma-powered distribution $p(x|c)^\gamma p(x)^{1-\gamma}$. Then, we clarify the behavior of CFG by showing that it is a kind of predictor-corrector method (Song et al., 2020) that alternates between denoising and sharpening, which we call predictor-corrector guidance (PCG). We prove that in the SDE limit, CFG is actually equivalent to combining a DDIM predictor for the conditional distribution together with a Langevin dynamics corrector for a gamma-powered distribution (with a carefully chosen gamma). Our work thus provides a lens to theoretically understand CFG by embedding it in a broader design space of principled sampling methods.
Chat is not available.