arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.14101cs.CVcs.GRcs.LG

利用损失引导的扩散模型增强图像布局控制

Enhancing Image Layout Control with Loss-Guided Diffusion Models

University of Toronto(多伦多大学);Ecomtent

Zakaria Patel, Kirill Serkh

AI总结:

本研究针对扩散模型的图像空间布局控制问题,提出结合两类免训练注意力约束方法的策略,证明二者互补,联合使用可提升控制性能。

摘要:

扩散模型是一类强大的生成模型,能够通过简单的文本提示从纯噪声中生成高质量图像。大多数为生成图像引入额外空间约束(如边界框)的方法需要微调,而规模更小、更新的一类方法则利用模型的注意力机制,且无需训练。这类方法通常分为两类:第一类直接修改特定token的交叉注意力图,以增强图像特定区域的信号;第二类通过在交叉注意力图上定义损失函数,利用该损失的梯度引导隐变量。此前的研究将这两种方法视为替代策略,我们则提出了一种解读框架,凸显了二者的互补特性,并证明将两种方法结合使用可获得更优性能。

英文摘要:

Diffusion models are a powerful class of generative models capable of producing high-quality images from pure noise using a simple text prompt. While most methods which introduce additional spatial constraints into the generated images (e.g., bounding boxes) require fine-tuning, a smaller and more recent subset of these methods take advantage of the models' attention mechanism, and are training-free. These methods generally fall into one of two categories. The first entails modifying the cross-attention maps of specific tokens directly to enhance the signal in certain regions of the image. The second works by defining a loss function over the cross-attention maps, and using the gradient of this loss to guide the latent. While previous work explores these as alternative strategies, we provide an interpretation for these methods which highlights their complimentary features, and demonstrate that it is possible to obtain superior performance when both methods are used in concert.

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

↑