cs.CVSep 22, 2026

Shallow to Deep: Aligning Token Pruning with Stage-wise Roles in LVLMs

Authors: Shuo Zhang, Jintao Tong, Yixiong Zou, Yuhua Li, Ruixuan Li

Organizations: School of Computer Science and Technology, Huazhong University of Science and Technology

Abstract

Large Vision-Language Models (LVLMs) incur high computational costs from redundant visual tokens. Although training-free attention-based multi-layer pruning in the vision encoder stage has been explored as an effective strategy, we find that pruning in shallow layers consistently degrades performance. In this paper, we aim to understand this problem and seek a solution. By analyzing attention patterns across network depth, we find that shallow layers primarily function as edge detectors with chaotic attention maps, while deeper layers transition through local subject recognition and unstable semantic aggregation. To address the misalignment between pruning strategies and network stages, we propose STD, a hierarchical token pruning framework that adapts token selection mechanisms to the functional role of each network stage. STD employs High-Frequency Spectral Analysis in shallow layers to deterministically preserve structural edges, uses Gaussian-Smoothed Attention in intermediate layers to maintain spatial coherence, and introduces a Stability-Adaptive Trigger in deep layers to execute pruning only during semantically stable phases. Extensive experiments show that STD outperforms state-of-the-art pruning methods by 1.1% on LLaVA-1.5-7B with 88.9% token reduction, while also being plug-and-play and highly effective when combined with other methods, and by 2.1% on LLaVA-NeXT-7B with 94.4% reduction, delivering a 3.9x speed-up in the prefilling stage. Our code will be released at https://github.com/Twilight03/STD.

Explore similar work

CardsList
  1. STAR-Pro: Stage-Wise Token Adaptive Reduction with Progressive Refinement for Efficient Large Vision-Language Models

    Sep 5, 2026Yichen Guo, Tinghao Wang, Qizhe Zhang +15Efficient VLM InferenceVisual Token Pruning

  2. Hierarchical and Query-Aware Visual Token Pruning for High-Resolution LVLMs

    Oct 16, 2026Luther Wei, Ling Zou, Kang-Yang Huang +2Visual Token PruningEfficient VLM Inference

  3. ASAP: Attention-Shift-Aware Pruning for Efficient LVLM Inference

    Mar 15, 2026Surendra Pathak, Bo HanEfficient VLM InferenceVisual Token Pruning