atlas

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

Classification Done Right for Vision-Language Pre-Training

NeurIPS 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Recaptioning and text-supervised pretraining · 2 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 SuperClass, a CLIP-free vision-language pretraining method that trains a ViT with a softmax classification loss over raw subword tokens (IDF-weighted, no filtering), pretrained on Datacomp-1B.

Problem

Contrastive vision-language pretraining like CLIP needs very large batch sizes and a text encoder; prior classification-based alternatives rely on hand-crafted text filtering and small-scale experiments, leaving their scalability unclear.

Approach

Use all subword tokens of each caption (from the CLIP BPE tokenizer) as a K-hot multi-label target, trained with a softmax cross-entropy over the vocabulary weighted by online IDF, so no text encoder, no contrastive batch, and no vocabulary curation are needed.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace contrastive CLIP pretraining with SuperClass classification pretrainingViT-Base · ImageNet-1K linear probingtop-1 accuracy (%): 78.5 -> 80.2Helped
Replace contrastive CLIP pretraining with SuperClass classification pretrainingViT-Large · ImageNet-1K linear probingtop-1 accuracy (%): 82.7 -> 85.0Helped
Use a subword-level tokenizer instead of a word-level tokenizer (ablation)SuperClass, ViT-L/16 · ImageNet-1K linear probingaccuracy: 80.3 -> 80.5Helped
Use a word-level tokenizer instead of a subword-level tokenizer (ablation)SuperClass, ViT-S/16 · ImageNet-1K linear probingaccuracy: 67.9 -> 68.4Helped
Replace the OpenAI CLIP tokenizer with WordPiece (ablation)SuperClass, ViT-L/16 · ImageNet-1Klinear probing accuracy: 80.5 -> 80.5No clear effect
Replace the OpenAI CLIP tokenizer with SentencePiece (ablation)SuperClass, ViT-L/16 · ImageNet-1Klinear probing accuracy: 80.5 -> 80.2Hurt
Use Softmax loss instead of BCE, ASL, SoftMargin, or Two-way loss (ablation)SuperClass, ViT-B/16 · ImageNet-1Klinear probing accuracy: 75.6 vs 73.6, 73.8, 73.5, 74.8Helped
Add IDF class weights (ablation)SuperClass, ViT-B/16 · ImageNet-1Klinear probing accuracy: 75.6 -> 76.0Helped
Add IDF class weights (ablation)SuperClass, ViT-B/16 · ImageNet-1Kzero-shot accuracy: 60.8 -> 61.7Helped
Keep stopwords in text labels instead of removing them (ablation)SuperClass, ViT-B/16 · ImageNet-1Klinear probing accuracy: 75.7 -> 76.0Helped
Replace CLIP pretraining with SuperClass pretrainingViT-L/16 · ImageNet-1K fine-tuningtop-1 accuracy: 86.5 -> 87.8Helped

Surprising findings

Limitations

Experimental setup

Datasets: Datacomp-1B ImageNet-1k COCO captions LLaVA training data

Benchmarks: ImageNet-1k Pets Cars COCO captions VQAv2 GQA VizWiz T-VQA SQA MMBench MME POPE

Metrics: top-1 accuracy linear probing accuracy zero-shot accuracy 10-shot accuracy CIDEr VQA accuracy

Baselines: OpenAI CLIP OpenCLIP CatLIP DINOv2 MAE CapPa BEiT iBOT

Builds on: CLIP LiT CatLIP ClipCap LLaVA

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