Overview
This post introduces and explains the paper "Rethinking Classifier-Free Guidance in On-Policy Diffusion Distillation" by Bingnan Li, Haozhe Wang, Haozhong Xiong, et al. (arXiv:2607.24731, cs.CV / cs.AI / cs.LG).
Background
- Diffusion models generate images/videos by learning to iteratively denoise, predicting a velocity field at each step. They power tools from Stable Diffusion to Sora, but controllability is limited.
- Classifier-Free Guidance (CFG) fixes this by training both a conditional and an unconditional mode, then extrapolating:
- On-policy distillation (OPD) trains a small student to imitate a large teacher along the student's own noise trajectories, usually via velocity matching.
- Naive guided matching is highly sensitive to inference-time guidance scale w, with quality fluctuating sharply.
- PDM remains stable and high-quality across a range of guidance scales, confirming that it resolves the NBA issue.
- "Looking right" is not "being right": matching a combined output can hide entirely wrong internal branches.
- Even default configurations like CFG can harbor hidden pitfalls—only branch-level analysis reveals them.
- Similar issues may arise in any distillation over composite predictions (multi-task learning, multimodal fusion, etc.).
v̂ = v_uncond + w · (v_cond − v_uncond)
With guidance strength w > 1, generation is pushed further from the unconditional output toward the conditioning signal. CFG is the default recipe for high-quality, prompt-consistent generation.
The problem: Negative Branch Asymmetry (NBA)
In naive guided matching, the student imitates only the combined CFG prediction v̂, not the two branches separately. This combined target is under-identified at the branch level: the same v̂ can arise from many different (v_cond, v_uncond) pairs, so the student can converge to solutions where positive and negative branch errors cancel each other out.
The paper names the observed failure mode Negative Branch Asymmetry (NBA): when the teacher's unconditional branch embeds privileged information invisible to the student, branch errors become antagonistic—the student's positive branch improves while its negative branch worsens (a seesaw dynamic), degrading overall fidelity of the combined output.
The fix: Positive-Direction Matching (PDM)
PDM decouples the supervision into two explicit targets:
1. Directly match the teacher's positive branch (v_cond). 2. Match the conditional direction (v_cond − v_uncond).
Each component now has an unambiguous learning target, preventing error-cancellation illusions.
Experiments
Applied to dense-to-sparse video control, a task sensitive to guidance strength:
Why it matters
Reference: Li, B., Wang, H., Xiong, H., et al. (2026). Rethinking Classifier-Free Guidance in On-Policy Diffusion Distillation. arXiv:2607.24731.