为图像生成定制双文本嵌入,提升真实感与图文对齐
End-to-end Training for Text-to-Image Synthesis using Dual-Text Embeddings
- 端到端训练双文本嵌入,分别优化图像真实感与图文对齐
- 在三个数据集上优于共享嵌入,接近甚至超过预训练模型
- 嵌入可迁移至图像编辑等下游任务,通用性强
文本到图像(T2I)合成需建模文本与图像间的复杂交互。现有方法通常使用预训练的图像对齐文本嵌入,通过对比损失进行训练,并在多个生成模型间复用。本文提出一种端到端学习方式,为T2I网络量身定制文本嵌入。结合生成与对比训练,采用两个独立嵌入:一个优化生成图像的真实感,另一个强化文本-图像对齐。在Oxford-102、Caltech-UCSD和MS-COCO三个基准数据集上的实验表明,双嵌入方案优于共享嵌入,性能优于使用判别式预训练文本编码器的方法。此外,所学嵌入还可用于文本到图像编辑等其他场景。
原文摘要 · Abstract (English)
Text-to-Image (T2I) synthesis is a challenging task that requires modeling complex interactions between two modalities ( i.e., text and image). A common framework adopted in recent state-of-the-art approaches to achieving such multimodal interactions is to bootstrap the learning process with pre-trained image-aligned text embeddings trained using contrastive loss. Furthermore, these embeddings are typically trained generically and reused across various synthesis models. In contrast, we explore an approach to learning text embeddings specifically tailored to the T2I synthesis network, trained in an end-to-end fashion. Further, we combine generative and contrastive training and use two embeddings, one optimized to enhance the photo-realism of the generated images, and the other seeking to capture text-to-image alignment. A comprehensive set of experiments on three text-to-image benchmark datasets (Oxford-102, Caltech-UCSD, and MS-COCO) reveal that having two separate embeddings gives better results than using a shared one and that such an approach performs favourably in comparison with methods that use text representations from a pre-trained text encoder trained using a discriminative approach. Finally, we demonstrate that such learned embeddings can be used in other contexts as well, such as text-to-image manipulation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。