arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04321cs.CVcs.AIcs.CLcs.MM
收藏

判别式探测与调优用于文本到图像生成

Discriminative Probing and Tuning for Text-to-Image Generation

National University of Singapore(新加坡国立大学);Hong Kong Polytechnic University(香港理工大学);Nanyang Technological University(南洋理工大学);Skywork AI(天工AI);Harbin Institute of Technology (Shenzhen)(哈尔滨工业大学(深圳))

Leigang Qu, Wenjie Wang, Yongqi Li, Hanwang Zhang, Liqiang Nie, Tat-Seng Chua

AI总结:

本文提出通过判别式适配器探测并微调T2I模型的判别能力,以提升文本-图像对齐,并利用自校正机制优化生成,在三个基准上取得优越性能。

摘要:

尽管文本到图像生成(T2I)取得了进展,但先前的方法常常面临生成图像中的关系混淆等文本-图像错位问题。现有解决方案涉及通过交叉注意力操作以更好地理解组成,或集成大型语言模型以改进布局规划。然而,T2I模型固有的对齐能力仍然不足。通过审视生成式建模与判别式建模之间的联系,我们假设T2I模型的判别能力可能反映其在生成过程中的文本-图像对齐熟练度。基于此,我们主张增强T2I模型的判别能力,以实现生成过程中更精确的文本到图像对齐。我们构建了一个基于T2I模型的判别式适配器,在两个代表性任务上探测其判别能力,并利用判别式微调来改善其文本-图像对齐。作为判别式适配器的附带优势,一种自校正机制可以在推理过程中利用判别式梯度,使生成的图像更好地与文本提示对齐。在三个基准数据集上的全面评估,包括分布内和分布外场景,证明了我们方法的优越生成性能。同时,与其它生成模型相比,它在两个判别任务上达到了最先进的判别性能。

英文摘要:

Despite advancements in text-to-image generation (T2I), prior methods often face text-image misalignment problems such as relation confusion in generated images. Existing solutions involve cross-attention manipulation for better compositional understanding or integrating large language models for improved layout planning. However, the inherent alignment capabilities of T2I models are still inadequate. By reviewing the link between generative and discriminative modeling, we posit that T2I models' discriminative abilities may reflect their text-image alignment proficiency during generation. In this light, we advocate bolstering the discriminative abilities of T2I models to achieve more precise text-to-image alignment for generation. We present a discriminative adapter built on T2I models to probe their discriminative abilities on two representative tasks and leverage discriminative fine-tuning to improve their text-image alignment. As a bonus of the discriminative adapter, a self-correction mechanism can leverage discriminative gradients to better align generated images to text prompts during inference. Comprehensive evaluations across three benchmark datasets, including both in-distribution and out-of-distribution scenarios, demonstrate our method's superior generation performance. Meanwhile, it achieves state-of-the-art discriminative performance on the two discriminative tasks compared to other generative models.

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

↑