atlas

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

Bootstrapping SparseFormers from Vision Foundation Models

CVPR 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Recaptioning and text-supervised pretraining · 1 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.

Bootstraps SparseFormer vision transformers from pretrained ViT/CLIP weights by inheriting cortex blocks and aligning final token embeddings with cosine loss on ImageNet-1K images only.

Problem

Training SparseFormers from scratch is expensive and hard to scale, and SparseFormer's largest variant reaches only ~84.8% ImageNet top-1, limiting its use as an efficient vision foundation model.

Approach

Inherit the standard transformer (cortex) blocks from large pretrained ViTs, freeze most of them, randomly initialize only the lightweight focusing transformer, and train by aligning SparseFormer's final token embedding with the teacher's [CLS]/projection embedding via cosine loss on unlabeled ImageNet-1K images.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add classification loss to representation-alignment bootstrapping (ablation)SF-BAugReg, 49 tokens · ImageNet-1K valtop-1 accuracy: 82.5 -> 77.4Hurt
reuse pretrained weights in the cortex transformer (ablation)SF-BAugReg, 49 tokens · ImageNet-1K valtop-1 accuracy: 69.5 -> 82.5Helped
use representation alignment instead of KL distillation at τ=3 (ablation)SF-BAugReg, 49 tokens · ImageNet-1K valtop-1 accuracy: 81.0 -> 82.5Helped
use representation alignment instead of KL distillation at τ=5 (ablation)SF-BAugReg, 49 tokens · ImageNet-1K valtop-1 accuracy: 81.5 -> 82.5Helped
bootstrap from pretrained weights rather than train from scratch (ablation)SparseFormer, 49 tokens · ImageNet-1K valtop-1 accuracy: 80.0 -> 82.5Helped
increase token count from 49 to 80 (ablation)SF-BAugReg · ImageNet-1K valtop-1 accuracy: 82.5 -> 83.4Helped
increase token count from 49 to 80 (ablation)SF-LAugReg · ImageNet-1K valtop-1 accuracy: 84.5 -> 85.2Helped
increase token count from 49 to 196 (ablation)SF-LAugReg · ImageNet-1K valtop-1 accuracy: 84.9 -> 86.7Helped
increase token count from 64 to 128 (ablation)SF-LCLIP · ImageNet-1K zero-shotaccuracy: 73.6 -> 74.9Helped
increase vision encoder token count from 64 to 128 (ablation)LLaVA with SF-LCLIP · ScienceQA testmulti-modal only accuracy: 84.5 -> 85.8Helped
replace CLIP ViT-L/14-336 with a bootstrapped SF-LCLIP using 144 tokensCLIP visual encoder · ImageNet-1K zero-shotaccuracy: 76.6 -> 75.9Hurt
replace CLIP ViT-L/14-336 with a bootstrapped SF-LCLIP using 144 tokensCLIP visual encoder · MS COCO valI→T@1: 57.9 -> 57.0Hurt

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1K ScienceQA training images

Benchmarks: ImageNet-1K ADE20k MS COCO Flickr30k ScienceQA

Metrics: top-1 accuracy mIoU zero-shot accuracy image-to-text recall@1 text-to-image recall@1 throughput (img/s) FLOPs

Baselines: ViT-B/16-224AugReg ViT-L/16-384AugReg ToMe DiffRate TinyCLIP Swin-L + UPerNet ViT-Adapter CLIP ViT-L/14-336

Builds on: SparseFormer ViT AugReg CLIP 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