arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.08070cs.CVcs.AIcs.LG
收藏

CFG++: 面向扩散模型的流形约束无分类器引导

CFG++: Manifold-constrained Classifier Free Guidance for Diffusion Models

KAIST(韩国科学技术院)

Hyungjin Chung, Jeongsol Kim, Geon Yeong Park, Hyelin Nam, Jong Chul Ye

AI总结:

针对传统无分类器引导(CFG)存在的流形外现象引发的可逆性差、模式崩溃等问题,提出流形约束的CFG++方法,将文本引导重构为带文本条件得分匹配损失的逆问题,在多类扩散任务中性能显著提升。

摘要:

无分类器引导(CFG)是现代扩散模型中用于文本引导生成的基础工具。尽管效果显著,但CFG存在明显缺陷。例如,搭配CFG的DDIM缺乏可逆性,使图像编辑变得复杂;此外,高质量输出所必需的高引导尺度,经常会导致模式崩溃等问题。与普遍认为这些是扩散模型固有局限的观点相反,本文揭示这些问题实际上源于与CFG相关的流形外现象,而非扩散模型本身。更具体地说,受基于扩散模型的逆问题求解器(DIS)最新进展的启发,我们将文本引导重新表述为带有文本条件得分匹配损失的逆问题,并开发了CFG++——一种解决传统CFG固有流形外挑战的新方法。CFG++对CFG的修正简单得出奇简单,但却带来了显著提升,包括文本到图像生成的样本质量更高、具备可逆性、引导尺度更小、模式崩溃减少等。此外,CFG++能够在较低引导尺度下实现无条件采样和条件采样之间实现无缝插值,在所有尺度下均始终优于传统CFG。而且,CFG++可以轻松集成到高阶扩散求解器中,并自然扩展到蒸馏扩散模型。实验结果证实,我们的方法在文本到图像生成、DDIM反转、编辑和求解逆问题方面显著提升了性能,表明其在各类利用文本引导的领域中具有广泛影响和潜在应用。项目页面:https://cfgpp-diffusion.github.io/。

英文摘要:

Classifier-free guidance (CFG) is a fundamental tool in modern diffusion models for text-guided generation. Although effective, CFG has notable drawbacks. For instance, DDIM with CFG lacks invertibility, complicating image editing; furthermore, high guidance scales, essential for high-quality outputs, frequently result in issues like mode collapse. Contrary to the widespread belief that these are inherent limitations of diffusion models, this paper reveals that the problems actually stem from the off-manifold phenomenon associated with CFG, rather than the diffusion models themselves. More specifically, inspired by the recent advancements of diffusion model-based inverse problem solvers (DIS), we reformulate text-guidance as an inverse problem with a text-conditioned score matching loss and develop CFG++, a novel approach that tackles the off-manifold challenges inherent in traditional CFG. CFG++ features a surprisingly simple fix to CFG, yet it offers significant improvements, including better sample quality for text-to-image generation, invertibility, smaller guidance scales, reduced mode collapse, etc. Furthermore, CFG++ enables seamless interpolation between unconditional and conditional sampling at lower guidance scales, consistently outperforming traditional CFG at all scales. Moreover, CFG++ can be easily integrated into high-order diffusion solvers and naturally extends to distilled diffusion models. Experimental results confirm that our method significantly enhances performance in text-to-image generation, DDIM inversion, editing, and solving inverse problems, suggesting a wide-ranging impact and potential applications in various fields that utilize text guidance. Project Page: https://cfgpp-diffusion.github.io/.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑