Don't Just Chase "Highlighted Tokens" in MLLMs: Revisiting Visual Holistic Context Retention

Xin Zou, Di Lu, Yizhou Wang, Yibo Yan, Yuanhuiyi Lyu +3 more
10/3/2025
cs.CV

No official code found. Best implementation: obananas/HoloV (66 stars, Apache-2.0, last commit 2026-03-05).

Abstract

Despite their powerful capabilities, Multimodal Large Language Models (MLLMs) suffer from considerable computational overhead due to their reliance on massive visual tokens. Recent studies have explored token pruning to alleviate this problem, which typically uses text-vision cross-attention or [\texttt{CLS}] attention to assess and discard redundant visual tokens. In this work, we identify a critical limitation of such attention-first pruning approaches, i.e., they tend to preserve semantically similar tokens, resulting in pronounced performance drops under high pruning ratios. To this end, we propose {HoloV}, a simple yet effective, plug-and-play visual token pruning framework for efficient inference. Distinct from previous attention-first schemes, HoloV rethinks token retention from a holistic perspective. By adaptively distributing the pruning budget across different spatial crops, HoloV ensures that the retained tokens capture the global visual context rather than isolated salient features. This strategy minimizes representational collapse and maintains task-relevant information even under aggressive pruning. Experimental results demonstrate that our HoloV achieves superior performance across various tasks, MLLM architectures, and pruning ratios compared to SOTA methods. For instance, LLaVA1.5 equipped with HoloV preserves 95.8\% of the original performance after pruning 88.9\% of visual tokens, achieving superior efficiency-accuracy trade-offs.

View on arXivView PDF

Code Implementations(1)

obananas/HoloVCommunity62%
Apache-2.0

[NeurIPS 2025 đŸ”„] Official implementation for "Don't Just Chase “Highlighted Tokens” in MLLMs: Revisiting Visual Holistic Context Retention"

661CSS, HTMLSep 21, 20257 months agoApache-2.0

Cite this paper

@article{zou2025dont,
  title  = {Don't Just Chase "Highlighted Tokens" in MLLMs: Revisiting Visual Holistic Context Retention},
  author = {Xin Zou and Di Lu and Yizhou Wang and Yibo Yan and Yuanhuiyi Lyu and Xu Zheng and Linfeng Zhang and Xuming Hu},
  year   = {2025},
  eprint = {2510.02912},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2510.02912v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2510.02912.svg)](https://codeofpaper.com/paper/2510.02912)

Discussion