Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TOPS: First-Principles Visual Token Pruning via Constructing Token Optimal Preservation Sets for Efficient MLLM Inference

About

Multimodal large language models (MLLMs) have achieved strong multimodal reasoning capabilities, but their efficiency is limited by the large number of visual tokens, which introduces substantial computational overhead. Visual token pruning offers a natural solution, yet existing methods are imperfect: attention-based criteria tend to retain redundant tokens, while diversity-based criteria are often agnostic to user instructions. Even methods that combine multiple criteria still lack a principled formulation of the intrinsic objective of token pruning. In this paper, we revisit visual token pruning from a first-principles perspective and formulate it as constructing Token Optimal Preservation Sets. Through a top-down information-theoretic analysis, we identify three fundamental principles for effective token selection: Task Relevance, Information Coverage, and Semantic Diversity. Based on these principles, we propose TOPS, a training-free and model-agnostic pruning module that can be applied to various MLLMs. Extensive experiments on 7 MLLM backbones and 14 benchmarks demonstrate that TOPS outperforms prior methods under diverse pruning settings. Notably, on LLaVA-NeXT, TOPS removes 77.8% of visual tokens while preserving 100.0% and 100.6% performance on its 7B and 13B models, respectively, suggesting that pruning redundant visual tokens can sometimes mitigate hallucination and inspire future lightweight MLLM design.

Tinghao Wang, Yichen Guo, Rui Huang, Zheng Lu, Qizhe Zhang, Chenxi Li, Yuan Zhang, Jiajun Cao, Zhirong Shen, Yaosong Du, Guangyan Gan, Wenya Wang, Lin William Cong, Shanghang Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Science Question AnsweringScienceQA
Accuracy72.4
916
Diagram Question AnsweringAI2D
AI2D Accuracy72.4
509
Long Video UnderstandingLongVideoBench (val)
Accuracy61.5
282
Long Video UnderstandingMLVU
Accuracy61.5
265
Multimodal UnderstandingLLaVA Evaluation Suite 1.5--
127
Hallucination and Visual Reasoning EvaluationHallusionBench--
61
Multimodal UnderstandingMLLM Evaluation Suite GQA, ScienceQA-IMG, TextVQA, POPE, MME, MMBench, MM-Vet, MMStar, AI2D, HallusionBench (test val)
Relative Accuracy (%)100.6
56
Multimodal Question AnsweringMMBench English
MMBen84.8
33
Multimodal Capability EvaluationMMStar
Overall Score62.8
31
Multimodal Understanding and Question AnsweringLLaVA-NeXT Evaluation Suite (GQA, SQA-IMG, VQA-Text, POPE, MME, MMB-EN, MMB-CN, MMVet) 13B (test)
GQA Score64.1
28
Showing 10 of 15 rows

Other info

Follow for update