atlas

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

Data Filtering Networks

ICLR 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Self-distillation for vision-language pretraining · 15 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.

Shows that data filtering networks (DFNs), CLIP-style models trained on small high-quality data, can filter large web image-text pools into state-of-the-art training sets (DFN-2B, DFN-5B) for CLIP models.

Problem

Filtering a large uncurated web image-text pool into a high-quality pre-training set; it is unclear which properties of a model make it a good filter, since ImageNet accuracy does not predict filtering quality.

Approach

Treat a filtering model as a distinct object: a CLIP model trained on high-quality data (fine-tuned with weight ensembling and ImageNet/COCO/Flickr data) applied pointwise to a raw pool; its filtering quality is measured by the performance of models trained on the induced dataset rather than its own downstream accuracy.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train the filtering network on high-quality CC12M rather than ImageNet-labeled dataCLIP ViT-B/32 data filtering network · DataComp medium induced datasetImageNet top-1: 0.266 -> 0.289Helped
Use CLIP filtering rather than an ImageNet binary filterData filtering network · DataComp medium induced datasetAverage: 0.295 -> 0.335Helped
Add augmentation (ablation)DFN training · DataComp largeAverage: 0.536 -> 0.542Helped
Increase samples seen and batch size from 2.56B / 4096 to 5.12B / 8192 (ablation)DFN training · DataComp largeAverage: 0.545 -> 0.550Helped
Add fine-tuning (ablation)DFN training · DataComp largeAverage: 0.550 -> 0.560Helped
Use OpenAI initialization (ablation)DFN training · DataComp largeAverage: 0.548 -> 0.560Helped
Train the filtering network on HQITP-350M and use its induced data rather than adding HQITP-350M to the OpenAI-filtered datasetViT-L/14 CLIP training · DataComp evaluationAverage: 0.654 -> 0.669Helped
Add HQITP-350M directly to DFN-2B training data (ablation)ViT-L/14 CLIP training · DataComp evaluationAverage: 0.669 -> 0.670No clear effect
Fine-tune the DFN on ImageNet instead of directly adding ImageNet to training dataBaseline DFN · ImageNet distribution shiftsObjectNet: 0.509 -> 0.536Helped
Fine-tune the DFN on ImageNet rather than directly training on ImageNetBaseline DFN · ImageNet distribution shiftsIN-Sketch: 0.512 -> 0.536Helped
Fine-tune the DFN on ImageNet rather than directly training on ImageNetBaseline DFN · ImageNet distribution shiftsIN-R: 0.703 -> 0.743Helped
Train the CLIP filtering network on publicly available CC12M + CC3M + SS15M instead of OpenAI WIT-400MViT-B/32 DFN · DataComp mediumAverage: 0.338 -> 0.349Helped

Surprising findings

Limitations

Experimental setup

Datasets: HQITP-350M MS COCO Flickr30k ImageNet-1k DataComp CommonPool Conceptual 12M Conceptual Captions 3M Shutterstock 15M LAION-2B

Benchmarks: DataComp 38-task suite ImageNet ImageNet-V2 ObjectNet ImageNet-Sketch ImageNet-R ImageNet-A VQAv2 GQA OKVQA

Metrics: zero-shot top-1 accuracy average accuracy over 38 tasks retrieval recall

Baselines: LAION-2B DataComp-1B OpenAI WIT-400M MetaCLIP WebLI CLIPA-v2

Builds on: CLIP DataComp OpenCLIP BLIP-2 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