StackTok: Accelerating VLMs Inference with Budget-Adaptive Visual Token Selection
On the map: Multimodal LLMs / vision-language models → Visual token pruning and efficiency
Abstract (arXiv)
Increasing image resolution produces ever-longer visual-token sequences in vision-language models (VLMs), substantially raising their inference cost. To reduce this overhead without retraining, existing methods select compact token subsets that prioritize query relevance, visual coverage, or a fixed trade-off between them. The appropriate balance, however, varies across queries and token budgets: localized questions favor relevance, whereas holistic questions demand broader visual coverage. We introduce StackTok, a training-free selector that treats query relevance as the objective and visual coverage as budget-calibrated support. StackTok builds a size-indexed coverage reference from a coverage-only greedy sequence and adjusts its support target using query--vision affinity entropy. A reference-gated interleaved selection policy then switches between relevance- and coverage-oriented additions according to the current subset's support deficit. For high-resolution inputs, StackTok allocates one shared token budget across crops according to the combined marginal gain of locally nominated tokens. Evaluated with five VLMs over ten distinct image-understanding benchmarks, StackTok ranks first among training-free selectors in every tested model--budget setting. On high-resolution LLaVA-NeXT-7B, it retains 95.26% of full-token performance with only 160 of 2{,}880 (5.6%) visual tokens.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Visual token pruning and efficiency, one of 16 directions in Multimodal LLMs / vision-language models. Builds on 12 of its 26 references that are in the map. First posted on arXiv 15 Sep 2026.
Builds on
12 of 26 references.
Closest ancestors · 1
Same direction, newest first
Other references · 11
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2023Improved Baselines with Visual Instruction Tuningcited by 4,418
- 2016Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answeringcited by 2,516
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2023MMMU: A Massive Multi-discipline Multimodal Understanding and Reasoning Benchmark for Expert AGIcited by 2,080
- 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
- 2019Towards VQA Models That Can Readcited by 1,718
- 2023Video-LLaVA: Learning United Visual Representation by Alignment Before Projectioncited by 1,402
- 2022Token Merging: Your ViT But Fastercited by 1,021
- 2023SEED-Bench: Benchmarking Multimodal LLMs with Generative Comprehensioncited by 940
- 2023OCRBench: On the Hidden Mystery of OCR in Large Multimodal Modelscited by 476
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Inference Optimal VLMs Need Fewer Visual Tokens and More Parameters
- 2025AdaFV: Rethinking of Visual-Language alignment for VLM acceleration
- 2026OccamToken: Efficient VLM Inference with Training-Free and Budget-Adaptive Token Pruning
- 2026GOTS: Greedy Orthogonal Token Selection for High-Resolution Vision-Language Models
- 2026Adaptive Visual Token Reduction for Accelerated Image Understanding
- 2025AdaptVision: Efficient Vision-Language Models via Adaptive Visual Acquisition
- 2025SparseVILA: Decoupling Visual Sparsity for Efficient VLM Inference
- 2024SparseVLM: Visual Token Sparsification for Efficient Vision-Language Model Inference