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

Smooth Diffusion:在扩散模型中构建平滑潜空间

Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models

Georgia Tech(佐治亚理工学院);UIUC(伊利诺伊大学厄巴纳-香槟分校);Tsinghua University(清华大学);Picsart AI Research (PAIR)(Picsart人工智能研究院(PAIR))

Jiayi Guo, Xingqian Xu, Yifan Pu, Zanlin Ni, Chaofei Wang, Manushree Vasu, Shiji Song, Gao Huang, Humphrey Shi

AI总结:

本文提出 Smooth Diffusion,通过逐步变化正则化和 ISTD 指标构建并评估平滑潜空间,以即插即用 Smooth-LoRA 提升 T2I 生成及下游任务表现。

摘要:

近来,扩散模型在文本到图像(T2I)生成方面取得了显著进展,能够合成具有高保真度和多样内容的图像。尽管有这一进展,扩散模型内部潜空间的平滑性仍在很大程度上未被探索。平滑的潜空间可确保对输入潜变量的扰动对应于输出图像的稳定变化。这一特性被证明有利于图像插值、反演和编辑等下游任务。在本工作中,我们通过观察微小潜变量变化所导致的明显视觉波动,揭示了扩散潜空间的非平滑性。为解决这一问题,我们提出了 Smooth Diffusion,这是一类新型扩散模型,能够同时具备高性能与平滑性。具体而言,我们引入了逐步变化正则化(Step-wise Variation Regularization),以强制任意输入潜变量的变化与输出图像变化之间的比例在任何扩散训练步骤中都保持为常数。此外,我们设计了插值标准差(ISTD)指标,用于有效评估扩散模型的潜空间平滑性。大量定量和定性实验表明,Smooth Diffusion 不仅在 T2I 生成中,而且在各种下游任务中,都是一种更理想的解决方案。Smooth Diffusion 被实现为即插即用的 Smooth-LoRA,可与各种社区模型配合使用。代码可在 https://github.com/SHI-Labs/Smooth-Diffusion 获取。

英文摘要:

Recently, diffusion models have made remarkable progress in text-to-image (T2I) generation, synthesizing images with high fidelity and diverse contents. Despite this advancement, latent space smoothness within diffusion models remains largely unexplored. Smooth latent spaces ensure that a perturbation on an input latent corresponds to a steady change in the output image. This property proves beneficial in downstream tasks, including image interpolation, inversion, and editing. In this work, we expose the non-smoothness of diffusion latent spaces by observing noticeable visual fluctuations resulting from minor latent variations. To tackle this issue, we propose Smooth Diffusion, a new category of diffusion models that can be simultaneously high-performing and smooth. Specifically, we introduce Step-wise Variation Regularization to enforce the proportion between the variations of an arbitrary input latent and that of the output image is a constant at any diffusion training step. In addition, we devise an interpolation standard deviation (ISTD) metric to effectively assess the latent space smoothness of a diffusion model. Extensive quantitative and qualitative experiments demonstrate that Smooth Diffusion stands out as a more desirable solution not only in T2I generation but also across various downstream tasks. Smooth Diffusion is implemented as a plug-and-play Smooth-LoRA to work with various community models. Code is available at https://github.com/SHI-Labs/Smooth-Diffusion.

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

↑