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

One Layer's Trash is Another Layer's Treasure: Adaptive Layer-wise Visual Token Selection in LVLMs

About

Large Vision-Language Models (LVLMs) have achieved remarkable success across diverse multimodal tasks, yet their practical deployment remains constrained by the computational burden arising from lengthy visual tokens. While visual token pruning has emerged as a promising solution, existing methods suffer from a fundamental limitation: once tokens are pruned at a specific layer, they become inaccessible to all subsequent layers, leading to premature information loss that can compromise model performance. Through empirical studies, we observe that different layers exhibit distinct visual region focus, indicating a varying optimal token subset across layers. Motivated by this insight, we propose Adaptive Layer-wise Visual Token Selection (ALVTS), a novel framework that breaks away from the conventional static token pruning paradigm. ALVTS incorporates a lightweight token selector to identify and route important tokens for further processing, while allowing less important tokens to skip the layer, thus minimizing computational redundancy. These two streams of tokens are seamlessly reintegrated before being fed into subsequent layers, facilitating adaptive compression across the entire model. Grounded in our importance consistency constrained low-rank approximation, the proposed token selection module closely emulates the full attention mechanism, effectively capturing its essential patterns without requiring model retraining. Extensive experiments on LLaVA-1.5, LLaVA-NeXT, and Qwen2.5-VL validate the effectiveness of our method. With an 89% token compression ratio, ALVTS retains 96.7% of the original model's accuracy, achieving a superior efficiency-accuracy trade-off for LVLM inference.

Yongru Chen, Kai Zhang, Zeliang Zong, Yuchen Lu, Wenming Tan, Ye Ren, Jilin Hu• 2026

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringVizWiz
Accuracy59.36
1863
Diagram Question AnsweringAI2D--
509
Visual Question AnsweringAI2D
Accuracy68.39
402
Object Hallucination EvaluationPOPE
Accuracy87.09
259
Visual Question AnsweringVizWiz (test)
Accuracy55.65
136
Object Hallucination EvaluationPOPE (test)
Accuracy85.99
123
Knowledge-based Visual Question AnsweringOKVQA
Accuracy0.417
101
Text-based Visual Question AnsweringVQAText
Accuracy58.37
89
Image CaptioningCOCO
CIDEr78.23
64
Image CaptioningNoCaps (test)
CIDEr104.5
63
Showing 10 of 19 rows

Other info

Follow for update