cs.CVApr 13, 2026

Decoupled Similarity for Task-Aware Token Pruning in Large Vision-Language Models

Authors: Kexin Ma, Jing Xiao, Chaofeng Chen, Geyong Min, Guibo Zhu, Jinqiao Wang, Liang Liao

Organizations: School of Artificial Intelligence, Wuhan University, Wuhan, China · Department of Computer Science, University of Exeter, Exeter, United Kingdom · Chinese Academy of Sciences, Beijing, China; Wuhan AI Research, Wuhan, China · Hangzhou Institute of Technology, Xidian University, Hangzhou, China

Abstract

Token pruning has emerged as an effective approach to reduce the substantial computational overhead of Large Vision-Language Models (LVLMs) by discarding less informative visual tokens while preserving performance. However, existing methods typically rely on individual attention sources from different LVLM components, resulting in incomplete and suboptimal pruning decisions due to biased attention distributions. To address this problem, we propose DeSAP, a novel Decoupled Similarity-Aware Pruning method for precise, task-aware token pruning within the visual encoder. Specifically, DeSAP introduces a decoupled similarity to capture fine-grained cross-modal relevance between visual features and text tokens, providing explicit task-related guidance for pruning. By integrating decoupled similarity with visual saliency signals derived from visual attention, DeSAP performs token pruning under the guidance of both task-related and visual cues, enabling robust pruning even under aggressive pruning ratios. Extensive experiments across diverse benchmarks and architectures show that DeSAP consistently outperforms SOTA methods in both accuracy and efficiency. On LLaVA-1.5-7B, DeSAP achieves a 10 times FLOPs reduction and a 2.3 times prefill speedup by retaining only 11.1% of visual tokens, while maintaining 98.1% of the original performance.

Figures & tables

Appendix figures & tables4 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. LearnPrunerV2: Learnable Progressive Token Pruning for Vision-Language Models

    Oct 27, 2026Rinyoichi Takezoe, Yaqian Li, Weiren Wang +6Visual Token PruningLVLM Pruning

  2. When Text Matters: Design Principles for Visual Token Pruning in Vision-Language Model

    Sep 28, 2026Minchan Kang, Kyeonghye Park, Seoyoung Cho +2Efficient VLM InferenceVisual Token Pruning