atlas

Research map / Vision-language and multimodal / CLIP and zero-shot

Modeling Caption Diversity in Contrastive Vision-Language Pretraining

ICML 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: CLIP training recipes and data curation · 0 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 Llip, a contrastive vision-language pretraining method whose image embedding is a text-conditioned mixture of K learnable visual mixture tokens, trained with a SigLIP-style loss on MetaCLIP data.

Problem

CLIP maps each image to a single vector and pushes all caption representations toward it, which ignores that an image admits many valid, diverse captions and limits the richness of visual representations.

Approach

The visual encoder outputs K learnable mixture tokens; a multi-head cross-attention whose queries come from the caption text feature produces caption-dependent mixing weights, yielding a contextualized image embedding contrasted with the text embedding under a SigLIP-style sigmoid loss, so the image representation is one-to-many rather than invariant.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
condition visual mixture weights on text features instead of using a learned shared query (ablation)Llip, ViT-B/32 · ImageNet zero-shottop-1 accuracyHelped
add 63 extra register tokens while using only the first learned token (ablation)SigLIP, ViT-B/32 · ImageNet zero-shottop-1 accuracyNo clear effect
add 63 extra learned tokens and uniformly average the 64 tokens (ablation)SigLIP, ViT-B/32 · ImageNet zero-shottop-1 accuracyHelped
use a learned shared query for non-uniform token mixing instead of uniform averaging (ablation)Llip token aggregation, ViT-B/32 · ImageNet zero-shottop-1 accuracyNo clear effect
increase the number of mixture tokens (ablation)Llip · ImageNet zero-shottop-1 accuracyHelped
increase cross-attention softmax temperature to large values (ablation)Llip · ImageNet zero-shottop-1 accuracyHurt
include image patch token outputs alongside mixture tokens in cross-attention (ablation)Llip, ViT-B/32 · ImageNet zero-shottop-1 accuracyMixed
use 32 mixture tokens rather than 1 mixture token with image patch tokens in cross-attention (ablation)Llip, ViT-B/32 · ImageNet zero-shottop-1 accuracyHelped
use caption-conditioned visual representations instead of SigLIP contrastive representationsLlip64, ViT-B/32 · ImageNet zero-shottop-1 accuracy: 67.3 -> 70.4Helped
use caption-conditioned visual representations instead of MetaCLIP representationsLlip64, ViT-G/14 · ImageNet zero-shottop-1 accuracy: 82.1 -> 83.5Helped
use Llip instead of MetaCLIPViT-L/14 · Geographic diversity, Overall Top5accuracy: 85.26 -> 83.55Hurt
use Llip instead of MetaCLIPViT-L/14 · Scene understanding, CLEVRaccuracy: 25.9 -> 25.5Hurt

Surprising findings

Limitations

Experimental setup

Datasets: MetaCLIP-2.5B Flickr30k MSCOCO ImageNet CIFAR10 CIFAR100 CUB Food-101 STL-10 Caltech-101 MNIST SUN397

Benchmarks: ImageNet CIFAR10 CIFAR100 CUB Food-101 STL-10 Caltech-101 MNIST SUN397 Cars Aircraft Pets

Metrics: top-1 accuracy average zero-shot accuracy recall@1 recall@5 recall@10

Baselines: CLIP SigLIP MetaCLIP OpenCLIP EVA-CLIP DFN CLIPA-v2

Builds on: CLIP SigLIP Vision Transformer Registers MetaCLIP

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