atlas

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

MoPE-CLIP: Structured Pruning for Efficient Vision-Language Models with Module-wise Pruning Error Metric

CVPR 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Fine-grained improvements to CLIP · 21 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 MoPE-CLIP, a mask-free structured pruning framework for CLIP that scores heads, FFN neuron groups and layers by the zero-shot/task retrieval drop when removed (MoPE metric), supporting both pre-training and fine-tuning compression.

Problem

Large vision-language pre-trained models such as CLIP are too big for resource-constrained platforms, and existing VLP compression methods either use uni-modal pruning metrics that transfer poorly to cross-modal tasks or rely on costly mask-search or multi-stage distillation.

Approach

A Module-wise Pruning Error metric measures each head, FFN neuron group, or Transformer layer by the recall drop on cross-modal retrieval when it is removed; the metric builds cost tables that guide one-shot width and depth pruning, followed by distillation of cross-modal similarity, uni-modal features and hidden states from the teacher CLIP.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use MoPE-based structured pruning for the vision encoderFine-tuned CLIP-ViT-L/14 · MSCOCO image-to-text retrievalTR@1: 70.3 -> 72.1Helped
Use MoPE-based text-encoder pruningFine-tuned CLIP-ViT-L/14 · MSCOCO text-to-image retrievalIR@1: 59.3 -> 59.6Helped
Prune and pre-train the compact model using MoPE-CLIPCLIP-ViT-L/14 · MSCOCO zero-shot image-to-text retrievalTR@1: 46.8 -> 52.8Helped
Prune and pre-train the compact model using MoPE-CLIPCLIP-ViT-L/14 · MSCOCO zero-shot text-to-image retrievalIR@1: 29.3 -> 37.3Helped
Use MoPE-based module selectionMoPE-CLIPbase · 11 zero-shot image classification datasets, YFCC15Maverage top-1 accuracy: 40.2 -> 58.8Helped
Use MoPE rather than top-layer selection (ablation)MoPE-CLIPV · MSCOCO image-to-text retrieval, prune 6 layers at 0.375 widthTR@1: 57.6 -> 69.7Helped
Use MoPE rather than bottom-layer selection (ablation)MoPE-CLIPV · MSCOCO image-to-text retrieval, prune 6 layers at 0.375 widthTR@1: 63.9 -> 69.7Helped
Use MoPE rather than every-other-layer pruning (ablation)MoPE-CLIPV · MSCOCO image-to-text retrieval, prune 6 layers at 0.375 widthTR@1: 66.6 -> 69.7Helped
Use MoPE rather than loss-gradient layer selection (ablation)MoPE-CLIPV · MSCOCO image-to-text retrieval, prune 6 layers at 0.375 widthTR@1: 66.3 -> 69.7Helped
Prune width before depth (ablation)MoPE-CLIPV · MSCOCO image-to-text retrieval, 86M modelTR@1: 64.0 -> 69.7Helped
Prune width before depth rather than pruning both simultaneously (ablation)MoPE-CLIPV · MSCOCO image-to-text retrieval, 86M modelTR@1: 61.4 -> 69.7Helped
Add cross-modal similarity distillation loss Lsim (ablation)MoPE-CLIPV · MSCOCO image-to-text retrievalTR@1: 68.8 -> 69.7Helped

Surprising findings

Limitations

Experimental setup

Datasets: CC3M CC12M YFCC15M MSCOCO Flickr30K

Benchmarks: MSCOCO Flickr30K CIFAR10 CIFAR100 Caltech101 Flowers102 OxfordPets DTD StanfordCars FGVC-Aircraft SUN397 Food101

Metrics: TR@1 TR@5 TR@10 IR@1 IR@5 IR@10 top-1 accuracy latency (ms)

Baselines: CLIP-ViT-B/32 SE-CLIP MagnCLIP DynaCLIP UPop-CLIP TinyCLIP EfficientVLM SLIP DeCLIP UniCLIP MCD ALIP

Builds on: CLIP DynaBERT UPop TinyCLIP MagnitudePruning

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