atlas

Research map / Image generation / Text-to-image diffusion

Leveraging Unpaired Data for Vision-Language Generative Models via Cycle Consistency

ICLR 2024 · Image generation › Text-to-image diffusion · approach: Personalized and region-controlled generation · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes ITIT, a unified image-text encoder with separate image and text decoders trained with cycle-consistency losses (T2I2T and I2T2I) so that unpaired image-only and text-only data can supplement a small paired set for generative vision-language training.

Problem

Generative vision-language models (text-to-image and image-to-text) need huge paired image-text corpora, which are costly to annotate or noisy when web-scraped; unpaired images and texts go unused.

Approach

A joint image-text encoder with disjoint autoregressive text decoder and parallel masked-token image decoder, trained on unpaired data by synthesizing the missing modality and reconstructing the original input (text-image-text and image-text-image cycles), with straight-through/Gumbel gradient estimation through argmax outputs and gradient truncation to keep cycle training tractable.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add full T2I2T and I2T2I cycle losses using unpaired WebLI images and text (ablation)ITIT-B trained on paired CC3M · MS-COCOFID: 15.7 -> 14.3Helped
Add full T2I2T and I2T2I cycle losses using unpaired WebLI images and text (ablation)ITIT-B trained on paired CC3M · MS-COCOCIDEr: 23.5 -> 31.1Helped
Use low-quality WebLI as paired rather than unpaired data (ablation)ITIT-B trained on paired CC3M · MS-COCOCIDEr: 23.5 -> 20.7Hurt
Add unpaired WebLI through T2T and I2I training without cycle losses (ablation)ITIT-B trained on paired CC3M · MS-COCOCIDEr: 23.5 -> 28.5Helped
Replace full cycle losses with half-cycle losses (ablation)ITIT-B trained on paired CC3M and unpaired WebLI · MS-COCOFID: 14.3 -> 14.5Hurt
Replace full cycle losses with half-cycle losses (ablation)ITIT-B trained on paired CC3M and unpaired WebLI · MS-COCOCIDEr: 31.1 -> 30.5Hurt
Add full T2I2T and I2T2I cycle losses without additional data (ablation)ITIT-B trained with CC3M · MS-COCOFID: 15.7 -> 15.4Helped
Add full T2I2T and I2T2I cycle losses without additional data (ablation)ITIT-B trained with CC3M · MS-COCOCIDEr: 23.5 -> 24.4Helped
Use Gumbel softmax instead of straight-through softmax for T2I2T gradient estimation (ablation)ITIT T2I2T cycle · COCO Karpathy splitCIDEr (zs): +0.3Helped
Use Gumbel softmax for I2T2I gradient estimation (ablation)ITIT I2T2I cycle · MS-COCOgeneration performanceNo clear effect
Add the full T2I2T cycle (ablation)ITIT-B with unpaired WebLI text · MS-COCOFID and CIDErMixed
Add the full I2T2I cycle (ablation)ITIT-B with unpaired WebLI images · MS-COCOFID and CIDErMixed

Surprising findings

Limitations

Experimental setup

Datasets: CC3M WebLI Shutterstock MS-COCO Karpathy split COCO Captions

Benchmarks: MS-COCO COCO Karpathy split captioning COCO Captions 30K FID

Metrics: FID CIDEr

Baselines: Stable Diffusion GLIDE DALL-E 2 Parti Muse BLIP SimVLM GIT CoBIT CM3Leon

Builds on: GIT Muse MAGE VQGAN CoBIT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy