Skip to main content
Article
Improving Text-To-Image Synthesis Using Contrastive Learning
arXiv
  • Hui Ye, Georgia State University, Atlanta, GA, United States
  • Xiulong Yang, Georgia State University, Atlanta, GA, United States
  • Martin Takáč, Mohamed bin Zayed University of Artificial Intelligence
  • Raj Sunderraman, Georgia State University, Atlanta, GA, United States
  • Shihao Ji, Georgia State University, Atlanta, GA, United States
Document Type
Article
Abstract

The goal of text-to-image synthesis is to generate a visually realistic image that matches a given text description. In practice, the captions annotated by humans for the same image have large variance in terms of contents and the choice of words. The linguistic discrepancy between the captions of the identical image leads to the synthetic images deviating from the ground truth. To address this issue, we propose a contrastive learning approach to improve the quality and enhance the semantic consistency of synthetic images. In the pretraining stage, we utilize the contrastive learning approach to learn the consistent textual representations for the captions corresponding to the same image. Furthermore, in the following stage of GAN training, we employ the contrastive learning method to enhance the consistency between the generated images from the captions related to the same image. We evaluate our approach over two popular text-to-image synthesis models, AttnGAN and DM-GAN, on datasets CUB and COCO, respectively. Experimental results have shown that our approach can effectively improve the quality of synthetic images in terms of three metrics: IS, FID and R-precision. Especially, on the challenging COCO dataset, our approach boosts the FID significantly by 29.60% over AttnGAN and by 21.96% over DM-GAN. © 2021, CC BY.

DOI
10.48550/arXiv.2107.02423
Publication Date
11-28-2021
Disciplines
Comments

Preprint: arXiv

  • Archived with thanks to arXiv
  • Preprint license: CC by
  • Uploaded 24 March 2022
Citation Information
H. Ye, "Improving text-to-image synthesis using contrastive learning," 2021, arXiv:2107.02423