cs.LGOct 5, 2026

CACFG: Curvature-Aware Classifier-Free Guidance and Optimal Control

Authors: Max Collins, Dasith de Silva Edirimuni, Jordan Vice, Tim French, Ajmal Mian

Organizations: School of Physics, Maths and Computing The University of Western Australia Perth, WA 6009

Abstract

Diffusion models generate samples by learning to reverse a fixed corruption process, and classifier-free guidance (CFG) is the standard mechanism for conditioning this process on a desired class or prompt. CFG can be applied at varying guidance strengths, and while higher strengths improve image quality and conditional alignment, too high a guidance strength can degrade image quality and diversity. Furthermore, CFG violates principled diffusion sampling dynamics, and existing explanations for why it works despite the violation disagree on the underlying theory or do not extend to deterministic samplers used in practice. We address both these issues. We first frame CFG sampling as a continuous-time optimal control problem, treating the sampling trajectory as a sequence of controls chosen to maximise the probability of the desired condition. Solving the resulting Hamilton--Jacobi--Bellman equation shows that CFG is recovered under specific path costs when using an unconstrained control set. We argue this lack of constraint is responsible for CFG's failure at high guidance strengths, since it permits the sampling path to move arbitrarily far from the current image estimate. To fix this, we propose curvature-aware CFG (CACFG), which constrains the control set to a hypersphere informed by the Gaussian regularisation used when training variational autoencoders. We show that the control inputs produced by CFG sampling routinely violate this bound, and that across diffusion models, datasets, and guidance schedules, CACFG achieves superior generative quality at mid-to-high guidance strengths with a less severe quality-diversity tradeoff than regular CFG.

Figures & tables

Appendix figures & tables19 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Analytic Distribution of Classifier-Free Guidance for Schedule Design

    Jul 22, 2026Enze Jiang, Zheng MaClassifier-Free GuidanceDistributions

  2. ContrastiveCFG: Guiding Diffusion Sampling by Contrasting Positive and Negative Concepts

    Nov 26, 2024Jinho Chang, Changsun Lee, Hyungjin Chung +1Classifier-Free GuidanceText-Conditioned Diffusion Model