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

ERA: Entropy-Guided Visual Token Pruning with Rectified Attention for Efficient MLLMs

About

Multimodal Large Language Models (MLLMs) incur prohibitive inference costs due to long visual token sequences. Training-free visual token reduction provides an efficient solution. However, existing methods distort attention distributions, giving rise to a phenomenon we term Attention Logit Collapse. To address this issue, we propose ERA, an Entropy-guided visual token pruning framework with Rectified Attention for efficient MLLMs. Specifically, ERA comprises three crucial components: Dual-view Entropy Pruning (DEP), Bias-aware Token Recycling (BTR), and Logit-preserving Attention Rectification (LAR). First, DEP identifies representative anchor tokens by jointly modeling visual diversity and head-wise saliency. BTR then recycles pruned tokens into their corresponding anchors while estimating a cluster-level logit bias. Building upon this, LAR injects the estimated bias into attention logits, effectively rectifying the collapse induced by token reduction. Together, these components preserve visual evidence even under aggressive compression, enabling robust performance across single-image, multi-image, and video settings on a wide range of MLLMs. Beyond delivering practical acceleration, ERA establishes logit-preserving visual token pruning as a principled framework for efficient MLLMs, unifying theoretical foundation, algorithmic design, and practical deployment. The code is at https://github.com/924973292/ERA.

Yuhao Wang, Mu Qiao, Haiwen Diao, Yunzhi Zhuge, Pingping Zhang, Xindong Zhang, Lei Zhang, Huchuan Lu• 2026

Related benchmarks

TaskDatasetResultRank
Multimodal EvaluationMME
Score1.47e+3
902
Video UnderstandingMVBench
Accuracy60.5
635
Diagram Question AnsweringAI2D
AI2D Accuracy79.8
509
Video UnderstandingVideoMME
Score (Overall)60.3
369
Multimodal EvaluationMM-Vet
Score37.2
249
Multi-modal EvaluationMME
MME Score2.25e+3
240
Visual Question AnsweringVQAv2
Accuracy77.9
226
Science Question AnsweringScienceQA SQA-I
Accuracy73.4
149
Multimodal EvaluationMMBench
MMB^CN Score61.7
146
Object Hallucination EvaluationPOPE
Accuracy (POPE)84.9
137
Showing 10 of 31 rows

Other info

Follow for update