atlas

Research map / Vision-language and multimodal / Multimodal LLMs

Inference Optimal VLMs Need Fewer Visual Tokens and More Parameters

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

Derives inference-compute scaling laws for VLMs trading off LLM size against visual token count, finds that visual reasoning favors the largest affordable LLM with very few (often one) visual tokens, and proposes a query-based convolutional cross-attention compressor (QueCC) for extreme compression.

Problem

Under a fixed inference FLOPs budget, it is unclear how to trade off the number of visual tokens passed to the LLM against the LLM's parameter count, and existing token compression work only studies modest compression ratios.

Approach

Fits a power-law scaling law for VLM error as a function of LLM parameters N and total inference tokens T, using projector-based token compression (TokenPacker variant) to vary visual tokens, and finds that for visual reasoning the optimal configuration is the largest LLM with very few visual tokens; it then adds user-query injection and learnable depthwise-convolution downsampling with local cros

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use QueCC instead of TokenPacker at 1 visual tokenVicuna-1.5 7B VLM · GQAaccuracy: 53.4* -> 53.5Helped
Use QueCC instead of TokenPacker at 1 visual tokenVicuna-1.5 7B VLM · VizWizaccuracy: 41.1* -> 44.1Helped
Use QueCC instead of TokenPacker at 4 visual tokensVicuna-1.5 7B VLM · MMBenchscore: 61.5* -> 62.1Helped
Replace the single visual token with 4 tokens under the same compression methodQueCC visual reasoning VLM · visual reasoning benchmarksperformanceMixed
Allocate fixed inference compute to a larger LLM and fewer visual tokensVLMs on visual reasoning tasks · visual reasoning benchmarksaverage downstream errorHelped
Reduce visual tokens while increasing LLM parametersVLMs on OCR tasks · DocVQA and TextVQAaverage downstream performanceHurt
Repeat 4 compressed visual tokens multiple timesVLMs · visual reasoning tasksperformanceNo clear effect
Add query injection to convolution-only compression (ablation)QueCC at 1 visual token · MMBenchscore: 57.5 -> 59.4Helped
Add convolutional downsampling to query-only compression (ablation)QueCC at 1 visual token · TextVQAaccuracy: 46.3 -> 46.9Helped
Add query injection to convolution-only compression (ablation)QueCC at 1 visual token · MMEscore: 1215.5 -> 1269.1Helped
Remove convolutional downsampling (ablation)QueCC at 1 visual token · GQAaccuracy: 53.5 -> 53.3No clear effect
Remove query injection (ablation)QueCC at 1 visual token · VizWizaccuracy: 44.1 -> 45.6Hurt

Surprising findings

Limitations

Experimental setup

Datasets: LLaVA-Next pretraining and finetuning data LLaVA-1.5 training data DocVQA TextVQA

Benchmarks: MME GQA AI2D MMBench MMMU ScienceQA MathVista POPE ChartQA DocVQA TextVQA VizWiz

Metrics: normalized average error across benchmarks accuracy F1 (POPE)

Baselines: TokenPacker LLaVA-PruMerge Matryoshka Multimodal Models Matryoshka Query Transformer LLaMA-VID LLaVA-1.5

Builds on: TokenPacker LLaVA-1.5 LLaVA-PruMerge Chinchilla scaling laws Matryoshka Query Transformer

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