atlas

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

Sieve: Multimodal Dataset Pruning using Image Captioning Models

CVPR 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Self-distillation for vision-language pretraining · 14 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 Sieve, a multimodal data pruning signal that scores image-text alignment by comparing alt-text to nucleus-sampled captions from a BLIP captioner using a sentence transformer, and fuses it with CLIPScore for DataComp pruning.

Problem

CLIPScore-based pruning of noisy web-crawled image-text data suffers from false positives (misaligned pairs scored high) and false negatives (aligned hard pairs scored low), hurting CLIP pretraining quality.

Approach

Use captions generated by an image-captioning model pretrained on small, curated data (BLIP, 14M pairs) with nucleus sampling, mask medium phrases like "image of", and measure alt-text alignment via cosine similarity in the embedding space of a lightweight sentence transformer (all-MiniLM-L6-v2); fuse with CLIPScore.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fuse Sieve alignment scores with CLIPScoreSieve on DataComp medium · Average over 38 datasetsaverage score: 32.8 -> 35.4Helped
Use Sieve pruning instead of CLIPScore pruningCLIP pretraining on DataComp medium · Retrievalscore: 25.1 -> 28.9Helped
Fuse Sieve alignment scores with CLIPScoreSieve on DataComp large · Average over 38 datasetsaverage score: 52.9 -> 54.6Helped
Use Sieve pruning instead of CLIPScore pruningCLIP pretraining on DataComp large · Retrievalscore: 46.6 -> 52.0Helped
Use curated 14M BLIP pretraining data instead of BLIP-129M (ablation)Sieve on DataComp medium · Average over 38 datasetsaverage score: 33.05 -> 34.03Helped
Use BLIP instead of GIT as the caption generator (ablation)Sieve on DataComp medium · Average over 38 datasetsaverage score: 32.40 -> 34.03Helped
Replace the sentence transformer with the CLIP text encoder (ablation)Sieve on DataComp medium · Average over 38 datasetsaverage score: 34.03 -> 25.90Hurt
Replace the sentence transformer with the BLIP text encoder (ablation)Sieve on DataComp medium · Average over 38 datasetsaverage score: 34.03 -> 31.85Hurt
Increase generated captions per image from 1 to 8 (ablation)Sieve on DataComp medium · Average over 38 datasetsaverage score: 32.50 -> 34.03Helped
Set CLIPScore fusion weight to 0.5 rather than 0.3 (ablation)Sieve+CLIPScore on DataComp medium · Average over 38 datasetsaverage score: 34.40 -> 35.15Helped
Set CLIPScore fusion weight to 0.7 rather than 0.5 (ablation)Sieve+CLIPScore on DataComp medium · Average over 38 datasetsaverage score: 35.15 -> 34.35Hurt

Surprising findings

Limitations

Experimental setup

Datasets: DataComp medium pool (128M) DataComp large pool (1.28B) BLIP pretraining data (14M curated; 129M variant) COCO Visual Genome Conceptual Captions Conceptual 12M SBU captions

Benchmarks: DataComp 38 downstream tasks ImageNet ImageNet distribution shifts VTAB retrieval tasks

Metrics: zero-shot accuracy retrieval accuracy average over 38 tasks

Baselines: No Filtering Basic Filtering LAION Filtering CLIPScore GIT-based Sieve CLIP text encoder BLIP text encoder

Builds on: CLIP DataComp BLIP CLIPScore all-MiniLM-L6-v2

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