atlas

Research map / Vision-language and multimodal / Multimodal LLMs

ECoFLaP: Efficient Coarse-to-Fine Layer-Wise Pruning for Vision-Language Models

ICLR 2024 · Vision-language and multimodal › Multimodal LLMs · approach: Attention-based visual token pruning · 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.

Proposes ECoFLaP, a two-stage coarse-to-fine pruning method for vision-language models that allocates per-layer sparsity using zeroth-order global importance scores and then prunes each layer with Wanda.

Problem

Layer-wise one-shot pruning of large vision-language models uses uniform per-layer sparsity and lacks global weight-importance information, while global pruning needs costly Hessian or gradient computation that is memory-prohibitive for LVLMs.

Approach

Estimate global importance scores via zeroth-order gradient approximation (forward-only, two perturbed forward passes with Gaussian noise), convert them into adaptive per-layer or per-block sparsity ratios, then apply Wanda layer-wise unstructured pruning at those ratios.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use globally informed adaptive layer sparsity with zeroth-order gradients instead of Wanda's fixed layer sparsityBLIP-2 pruning · visual question answering, image captioning, and image retrievalperformance: 1.1%, 2.3%, and 1.0%Helped
use zeroth-order rather than first-order gradients for global importance estimationECoFLaP · BLIP-2 pruningGPU memory usage: 40%Helped
use ECoFLaP instead of UPopBLIP pruning at 0.5 sparsity · NLVR² and COCO captionsperformance: 1.8% and 2.6%Helped
use ECoFLaP with SparseGPT rather than SparseGPT aloneBLIP-2 · zero-shot multimodal tasks at 60% sparsitymacro average: 51.8 -> 53.1Helped
use 0.6 rather than 1.0 maximum per-layer sparsity (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 51.6 -> 58.4Helped
remove the maximum per-layer sparsity cap (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.4 -> 51.6Hurt
increase calibration samples from 16 to 32 (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.2 -> 58.4Helped
increase calibration samples from 32 to 64 (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.4 -> 58.3No clear effect
increase calibration samples from 32 to 128 (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.4 -> 58.4No clear effect
increase noises per sample from 1 to 4 at fixed forward budget (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.4 -> 58.1Hurt
increase noises per sample from 1 to 8 at fixed forward budget (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.4 -> 58.0Hurt
increase noises per sample from 1 to 32 at fixed forward budget (ablation)ECoFLaP · BLIP-2 VQAv2 val at 0.5 total sparsityaccuracy: 58.4 -> 58.1Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CC3M ImageNet C4 NLVR2 COCO captions VQAv2 OK-VQA GQA NoCaps Flickr30k MMLU WikiText

Benchmarks: VQAv2 OK-VQA GQA NoCaps Flickr30k NLVR2 COCO captions MMLU ImageNet-1k WikiText 11 CLIP zero-shot classification datasets

Metrics: accuracy CIDEr SPICE TR@1 IR@1 perplexity GPU memory usage

Baselines: Global Magnitude Pruning Gradient-based Pruning SparseGPT Wanda UPop

Builds on: Wanda SparseGPT WoodFisher UPop forward-forward algorithm

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