arXiv:2410.20972cs.CV2024-10被引 10

降低注意力重叠可显著减少文本生成图像中实体缺失问题。

Attention Overlap Is Responsible for The Entity Missing Problem in Text-to-image Diffusion Models!

  • 通过设计四种无训练损失函数调控注意力重叠。
  • 在多个基准上提升视觉问答与人类评分,最高增9%。
  • 适合关注图像生成准确性与语义对齐的研究者。

文本到图像扩散模型(如 Stable Diffusion、DALL-E)虽能生成高质量图像,但在组合生成任务中常出现提示中特定实体缺失的问题。本文系统分析了交叉注意力动态的三大可能原因:(1)某些实体注意力强度不足,(2)注意力分布过宽,(3)不同实体间注意力图重叠过高。研究发现,减少实体间注意力重叠可有效降低实体缺失率。核心假设是:在去噪过程中,特定实体的词元争夺同一图像区域的注意力,导致注意力分散,影响准确表达。为此,提出 IoU、CoM 距离、KL 散度和聚类紧凑性(CC)四种无需重训练的损失函数,用于调节去噪过程中的注意力重叠。在多种基准测试中,所提方法显著提升组合生成准确率,优于以往方法,在视觉问答(VQA)、描述生成分数、CLIP 相似度及人工评估中表现优异,其中人工评估得分较最优基线提升 9%。

原文摘要 · Abstract (English)

Text-to-image diffusion models, such as Stable Diffusion and DALL-E, are capable of generating high-quality, diverse, and realistic images from textual prompts. However, they sometimes struggle to accurately depict specific entities described in prompts, a limitation known as the entity missing problem in compositional generation. While prior studies suggested that adjusting cross-attention maps during the denoising process could alleviate this problem, they did not systematically investigate which objective functions could best address it. This study examines three potential causes of the entity-missing problem, focusing on cross-attention dynamics: (1) insufficient attention intensity for certain entities, (2) overly broad attention spread, and (3) excessive overlap between attention maps of different entities. We found that reducing overlap in attention maps between entities can effectively minimize the rate of entity missing. Specifically, we hypothesize that tokens related to specific entities compete for attention on certain image regions during the denoising process, which can lead to divided attention across tokens and prevent accurate representation of each entity. To address this issue, we introduced four loss functions, Intersection over Union (IoU), center-of-mass (CoM) distance, Kullback-Leibler (KL) divergence, and clustering compactness (CC) to regulate attention overlap during denoising steps without the need for retraining. Experimental results across a wide variety of benchmarks reveal that these proposed training-free methods significantly improve compositional accuracy, outperforming previous approaches in visual question answering (VQA), captioning scores, CLIP similarity, and human evaluations. Notably, these methods improved human evaluation scores by 9% over the best baseline, demonstrating substantial improvements in compositional alignment.

图像生成注意力机制扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。