HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

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

David Nukrai; Ron Mokady; Amir Globerson

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

Abstract

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.

Code Repositories

zelaki/wsac
pytorch
Mentioned in GitHub
uriberger/re_cap
pytorch
Mentioned in GitHub
davidhuji/capdec
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-captioning-on-coco-captionsCapDec
BLEU-4: 26.4
CIDER: 91.8
METEOR: 25.1
image-captioning-on-flickrstyle10kCapDec
BLEU-1 (Romantic): 29.4
image-captioning-on-mscoco-1CapDec
BLEU-4: 26.4
semi-supervised-learning-for-image-captioning-2CapDec
CIDEr: 39.1
semi-supervised-learning-for-image-captioning-3CapDec
CIDEr: 30.0

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Text-Only Training for Image Captioning using Noise-Injected CLIP | Papers | HyperAI