atlas

Research map / Vision-language and multimodal / Multimodal LLMs

Efficient Large Multi-modal Models via Visual Context Compression

NeurIPS 2024 · Vision-language and multimodal › Multimodal LLMs · approach: Training-free token reduction for MLLMs · 8 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 visual tokens in LLaVA are redundant (average pooling inside the LLM drops 70% of tokens for ~3% GQA loss) and proposes LLaVolta, a staged training scheme that progressively compresses visual tokens during training to cut training time ~16-19% while matching or exceeding LLaVA-1.5-7B.

Problem

Visual tokens in multi-modal LLMs are numerous and redundant, but efficient training (not just inference) of MLLMs with compressed visual tokens has been largely unexplored; naive compression during training can lose information.

Approach

A nonparametric average-pooling compressor inserted at a deep LLM layer serves as the visual context compressor; LLaVolta trains in stages that progressively reduce compression (heavy to none/light), so the model is trained efficiently yet matches inference without compression.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
training-free average pooling of visual tokens at test time (stride 4 from layer 16, ~70% tokens removed) (ablation)LLaVA-1.5-7B · GQAGQA accuracy: ~3% dropHurt
add average pooling compressor during training (stride 2 to 16 staged, last stage compression)LLaVA-1.5-7B · 13 benchmarks averageAvg. score: 61.8 to 62.1Helped
staged training with compression (last stage compression) instead of full-token trainingLLaVA-1.5-7B · Training timetrain time per epoch: 15.3h to 12.4hHelped
use average pooler as compressor instead of attention-based, random, K-Means, Q-Former during training (ablation)MLLM training with compression · Compressor comparison, train with compressionAvg. score: 60.4 vs 58.5 (VCC), 57.9 (K-Means), 57.0 (FastV), 50.8 (random)Helped
attention-based compressors (FastV, VCC) used during training instead of inference-only (ablation)MLLM training with compression · Compressor comparison, train with compressionAvg. score: 57.0 (FastV), 58.5 (VCC) vs 60.4 average poolingHurt
replace Q-Former with nonparametric average pooling (stride 64 at layer 1) (ablation)MLLM training · Parametric vs nonparametric compressorAvg. score: 57.8 vs 56.2 Q-Former; training 9.2h vs 10.4hHelped
heavier compression during training (stride 8 at layer 2, CR 557%) vs lighter (stride 2 at layer 16) (ablation)MLLM training with average pooling · Compression levelAvg. score: 60.3 at CR 557% vs 61.8 at CR 133-188% light compressionMixed
multi-stage layer-wise progressive compression instead of single-stage compressor (ablation)LLaVA-1.5-7B · 13 benchmarks average (direct inference vs retrained)Avg. score: direct inference 60.0 vs 57.0 single; retrained 60.2 vs 61.3 singleMixed
apply LLaVolta compression schedule to 13B LLaVALLaVA-13B · 13 benchmarks averageAvg. score: 63.9 to 64.0; training 21.1h to 17.6hNo clear effect
apply LLaVolta schedule to Mini-Gemini 2BMGM-2B · 13 benchmarks averageAvg. score: 58.3 to 58.1; training 18.1h to 14.8hNo clear effect
apply LLaVolta schedule to VideoLLaVAVideoLLaVA · MSVD/MSRVTT/ActivityNet-QA average accuracyAverage accuracy: 57.8 to 58.2 (deeper then wider); training 40.7h to 37.1hHelped

Surprising findings

Limitations

Experimental setup

Datasets: LLaVA-1.5 instruction tuning data LLaVA-1.5 pretraining data Video-LLaVA training data

Benchmarks: GQA MM-Vet ScienceQA MME TextVQA POPE MMBench MMBench-CN VQA-v2 LLaVA-Bench-in-the-Wild VisWiz SEED-Image

Metrics: accuracy score latency (ms) TFLOPs training time (h) compression ratio

Baselines: LLaVA-1.5-7B FastV VCC K-Means Random Dropping Q-Former (MiniGPT-4) Mini-Gemini-2B LLaVA-13B Video-LLaVA

Builds on: LLaVA-1.5 FastV VCC BLIP-2 Q-Former Video-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