GeoPrune: A Robust and Engine-Friendly Importance Signal Beyond Attention for Vision Token Pruning in VLMs

Kang Lou, Shijin Zhang, Ming Chen, Junzhe Cao


Abstract
Vision token pruning can accelerate vision language model inference by removing redundant visual tokens. However, we observe that ViT attention based pruning drops sharply on Qwen3-VL and Qwen3.5. Empirically, we identify the root cause: the last-block MLP of the vision encoder significantly alters the spectral structure of hidden states and induces spectral collapse. Thus, pre-MLP attention rankings no longer reflect downstream token importance. Through further analysis of vision token selection when ViT attention based pruning still succeeds, we find that selected tokens strongly anti-correlate with raw L2 norm in the vision encoder output space Sout , while favoring tokens with larger residual energy. Motivated by shared components mask token-specific importance, we propose GeoPrune, a new importance-based pruning method that operates in Sout , applies deflation to remove dominant components, and ranks tokens by residual L2 norm. At aggressive pruning rates, GeoPrune is more robust and surpasses ViT attention based pruning by +4.4/+6.3 recovery points on Qwen3-VL and Qwen3.5. We further integrate GeoPrune into the production-grade inference engine vLLM, delivering 5.4%-10.5% mean E2E latency reductions with only slight accuracy drops, advancing vision token pruning toward practical deployment.
Anthology ID:
2026.findings-emnlp.1913
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
40759–40776
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1913/
DOI:
Bibkey:
Cite (ACL):
Kang Lou, Shijin Zhang, Ming Chen, and Junzhe Cao. 2026. GeoPrune: A Robust and Engine-Friendly Importance Signal Beyond Attention for Vision Token Pruning in VLMs. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 40759–40776, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
GeoPrune: A Robust and Engine-Friendly Importance Signal Beyond Attention for Vision Token Pruning in VLMs (Lou et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1913.pdf