arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.08048cs.CV

通过交换令牌引导扩散模型

Guiding a Diffusion Model by Swapping Its Tokens

MoE Key Lab of Artificial Intelligence, AI Institute, Shanghai Jiao Tong University(上海交通大学人工智能研究院教育部人工智能重点实验室);vivo Mobile Communication Co., Ltd.(维沃移动通信有限公司)

Weijia Zhang, Yuehao Liu, Shanyan Guan, Wu Ran, Yanhao Ge, Wei Li, Chao Ma

AI总结:

本文提出一种简单方法,通过交换令牌实现条件和无条件生成的CFG引导,提升图像质量和提示对齐。

摘要:

分类器自由引导(CFG)是一种广泛使用的推理技术,用于提升扩散模型的图像质量。然而,其依赖文本条件限制了无条件生成的应用。我们提出了一种简单的方法,使CFG引导适用于条件和无条件生成。核心思想是通过简单的令牌交换操作生成扰动预测,并利用其与干净预测之间的方向,引导采样向更高保真度的分布发展。实践中,我们交换空间或通道维度中语义差异最大的令牌潜在表示。与现有方法不同,我们的方法选择性地交换和重组令牌潜在表示,允许更精细地控制扰动及其对生成样本的影响。在MS-COCO 2014、MS-COCO 2017和ImageNet数据集上的实验表明,所提出的自交换引导(SSG)在不同设置下优于先前的无条件方法,在图像保真度和提示对齐方面表现更佳。其细粒度扰动粒度也提高了鲁棒性,减少了扰动强度范围内的副作用。总体而言,SSG将CFG扩展到更广泛的应用,包括条件和无条件生成,并可作为插件轻松插入任何扩散模型以获得即时改进。

英文摘要:

Classifier-Free Guidance (CFG) is a widely used inference-time technique to boost the image quality of diffusion models. Yet, its reliance on text conditions prevents its use in unconditional generation. We propose a simple method to enable CFG-like guidance for both conditional and unconditional generation. The key idea is to generate a perturbed prediction via simple token swap operations, and use the direction between it and the clean prediction to steer sampling towards higher-fidelity distributions. In practice, we swap pairs of most semantically dissimilar token latents in either spatial or channel dimensions. Unlike existing methods that apply perturbation in a global or less constrained manner, our approach selectively exchanges and recomposes token latents, allowing finer control over perturbation and its influence on generated samples. Experiments on MS-COCO 2014, MS-COCO 2017, and ImageNet datasets demonstrate that the proposed Self-Swap Guidance (SSG), when applied to popular diffusion models, outperforms previous condition-free methods in image fidelity and prompt alignment under different set-ups. Its fine-grained perturbation granularity also improves robustness, reducing side-effects across a wider range of perturbation strengths. Overall, SSG extends CFG to a broader scope of applications including both conditional and unconditional generation, and can be readily inserted into any diffusion model as a plug-in to gain immediate improvements.

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

↑