Classifier-free Guidance with Adaptive Scaling
This work addresses a specific bottleneck in text-to-image generation for AI practitioners, offering an incremental improvement over existing guidance methods.
The paper tackles the trade-off between image quality and prompt correspondence in classifier-free guidance for diffusion models by introducing β-CFG, which uses adaptive scaling and time-dependent curves to dynamically adjust guidance during generation, achieving better FID scores while maintaining similar CLIP similarity scores.
Classifier-free guidance (CFG) is an essential mechanism in contemporary text-driven diffusion models. In practice, in controlling the impact of guidance we can see the trade-off between the quality of the generated images and correspondence to the prompt. When we use strong guidance, generated images fit the conditioned text perfectly but at the cost of their quality. Dually, we can use small guidance to generate high-quality results, but the generated images do not suit our prompt. In this paper, we present $β$-CFG ($β$-adaptive scaling in Classifier-Free Guidance), which controls the impact of guidance during generation to solve the above trade-off. First, $β$-CFG stabilizes the effects of guiding by gradient-based adaptive normalization. Second, $β$-CFG uses the family of single-modal ($β$-distribution), time-dependent curves to dynamically adapt the trade-off between prompt matching and the quality of samples during the diffusion denoising process. Our model obtained better FID scores, maintaining the text-to-image CLIP similarity scores at a level similar to that of the reference CFG.