English

STAR: Stage-Wise Attention-Guided Token Reduction for Efficient Large Vision-Language Models Inference

Machine Learning 2025-05-20 v1 Computer Vision and Pattern Recognition

Abstract

Although large vision-language models (LVLMs) leverage rich visual token representations to achieve strong performance on multimodal tasks, these tokens also introduce significant computational overhead during inference. Existing training-free token pruning methods typically adopt a single-stage strategy, focusing either on visual self-attention or visual-textual cross-attention. However, such localized perspectives often overlook the broader information flow across the model, leading to substantial performance degradation, especially under high pruning ratios. In this work, we propose STAR (Stage-wise Attention-guided token Reduction), a training-free, plug-and-play framework that approaches token pruning from a global perspective. Instead of pruning at a single point, STAR performs attention-guided reduction in two complementary stages: an early-stage pruning based on visual self-attention to remove redundant low-level features, and a later-stage pruning guided by cross-modal attention to discard task-irrelevant tokens. This holistic approach allows STAR to significantly reduce computational cost while better preserving task-critical information. Extensive experiments across multiple LVLM architectures and benchmarks show that STAR achieves strong acceleration while maintaining comparable, and in some cases even improved performance.

Keywords

Cite

@article{arxiv.2505.12359,
  title  = {STAR: Stage-Wise Attention-Guided Token Reduction for Efficient Large Vision-Language Models Inference},
  author = {Yichen Guo and Hanze Li and Zonghao Zhang and Jinhao You and Kai Tang and Xiande Huang},
  journal= {arXiv preprint arXiv:2505.12359},
  year   = {2025}
}
R2 v1 2026-07-01T02:19:33.507Z