Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Text-Only Training for Image Captioning using Noise-Injected CLIP

About

We consider the task of image-captioning using only the CLIP model and additional text data at training time, and no additional captioned images. Our approach relies on the fact that CLIP is trained to make visual and textual embeddings similar. Therefore, we only need to learn how to translate CLIP textual embeddings back into text, and we can learn how to do this by learning a decoder for the frozen CLIP text encoder using only text. We argue that this intuition is "almost correct" because of a gap between the embedding spaces, and propose to rectify this via noise injection during training. We demonstrate the effectiveness of our approach by showing SOTA zero-shot image captioning across four benchmarks, including style transfer. Code, data, and models are available on GitHub.

David Nukrai, Ron Mokady, Amir Globerson• 2022

Related benchmarks

TaskDatasetResultRank
Image CaptioningMS-COCO
CIDEr91.8
61
Image CaptioningNoCaps
CIDEr (in-domain)60.1
36
Image CaptioningMSCOCO
BLEU@426.4
27
Movie Audio Description generationMAD-eval-Named v2 (test)
C Score6.7
17
Audio DescriptionMAD-Eval (test)
CIDEr6.7
16
Image CaptioningFlickr30K zero-shot
CIDEr23.1
11
Image CaptioningFlickr30K
BLEU-417.7
8
Image CaptioningCOCO zero-shot
BLEU@49.3
7
Style-Guided Image CaptioningFlickrStyle10K Romantic
BLEU-127.9
5
Style-Guided Image CaptioningFlickrStyle10K Humorous
BLEU-129.4
5
Showing 10 of 11 rows

Other info

Code

Follow for update