arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2505.03203cs.CV

PiCo:在扩散模型中通过改进噪声选择与精确掩码控制增强文本-图像对齐

PiCo: Enhancing Text-Image Alignment with Improved Noise Selection and Precise Mask Control in Diffusion Models

Nanjing University of Aeronautics and Astronautics(南京航空航天大学)

Chang Xie, Chenyi Zhuang, Pan Gao

AI总结:

PiCo 提出一种免训练方法,通过噪声选择模块筛选适合目标文本的随机噪声,并利用参考掩码模块精确调制交叉注意力,从而提升复杂文本提示下的文本-图像对齐效果。

摘要:

先进的扩散模型在文本到图像组合生成方面取得了显著进展。然而,现有模型在面对复杂文本提示时仍难以实现文本-图像对齐。本文重点指出影响该对齐的两个因素:随机初始化噪声的质量以及所生成控制掩码的可靠性。为此,我们提出 PiCo(Pick-and-Control),一种无需训练的新方法,包含两个关键组件以应对上述两个因素。首先,我们开发了一个噪声选择模块,用于评估随机噪声的质量并判断该噪声是否适合目标文本。在噪声选择阶段采用快速采样策略以确保效率。其次,我们引入了一个参考掩码模块,用于生成像素级掩码并精确调制交叉注意力图。该参考掩码被应用于标准扩散过程,以引导文本与图像特征之间的合理交互。大量实验验证了 PiCo 在将用户从繁琐的随机生成过程中解放出来以及增强多样化文本描述的文本-图像对齐方面的有效性。

英文摘要:

Advanced diffusion models have made notable progress in text-to-image compositional generation. However, it is still a challenge for existing models to achieve text-image alignment when confronted with complex text prompts. In this work, we highlight two factors that affect this alignment: the quality of the randomly initialized noise and the reliability of the generated controlling mask. We then propose PiCo (Pick-and-Control), a novel training-free approach with two key components to tackle these two factors. First, we develop a noise selection module to assess the quality of the random noise and determine whether the noise is suitable for the target text. A fast sampling strategy is utilized to ensure efficiency in the noise selection stage. Second, we introduce a referring mask module to generate pixel-level masks and to precisely modulate the cross-attention maps. The referring mask is applied to the standard diffusion process to guide the reasonable interaction between text and image features. Extensive experiments have been conducted to verify the effectiveness of PiCo in liberating users from the tedious process of random generation and in enhancing the text-image alignment for diverse text descriptions.

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

↑