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

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


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 has been explored as an effective strategy, we find that pruning in shallow layers consistently degrades performance. 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 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 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 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.9× speed-up in the prefilling stage. Our code will be released at https://github.com/Twilight03/STD.
Anthology ID:
2026.emnlp-main.1226
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
27977–27993
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1226/
DOI:
Bibkey:
Cite (ACL):
Shuo Zhang, Jintao Tong, Yixiong Zou, Yuhua Li, and Ruixuan Li. 2026. Shallow to Deep: Aligning Token Pruning with Stage-wise Roles in LVLMs. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 27977–27993, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Shallow to Deep: Aligning Token Pruning with Stage-wise Roles in LVLMs (Zhang et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1226.pdf