VLM-Pruner: Buffering for Spatial Sparsity in an Efficient VLM Centrifugal Token Pruning Paradigm
By: Zhenkai Wu , Xiaowen Ma , Zhenliang Ni and more
Potential Business Impact:
Makes AI understand pictures faster on phones.
Vision-language models (VLMs) excel at image understanding tasks, but the large number of visual tokens imposes significant computational costs, hindering deployment on mobile devices. Many pruning methods rely solely on token importance and thus overlook inter-token redundancy, retaining numerous duplicated tokens and wasting capacity. Although some redundancy-aware approaches have been proposed, they often ignore the spatial relationships among visual tokens. This can lead to overly sparse selections of retained tokens that fail to adequately cover the regions of target objects. To address these limitations, we propose VLM-Pruner, a training-free token pruning algorithm that explicitly balances redundancy and spatial sparsity. We introduce a centrifugal token pruning paradigm that enables near-to-far selection while prioritizing the preservation of fine-grained object details. Moreover, we design a Buffering for Spatial Sparsity (BSS) criterion that defers the selection of spatially distant tokens. We further adopt a parallel greedy strategy to conduct token selection efficiently. To mitigate information loss from pruning, we selectively fuse salient information from the discarded tokens into the retained ones. Comprehensive comparisons demonstrate that VLM-Pruner consistently outperforms strong baselines across five VLMs with an 88.9\% pruning rate, while delivering an end-to-end inference speedup.
Similar Papers
Object-Centric Vision Token Pruning for Vision Language Models
CV and Pattern Recognition
Makes AI understand pictures faster and better.
Index-Preserving Lightweight Token Pruning for Efficient Document Understanding in Vision-Language Models
CV and Pattern Recognition
Makes AI understand papers faster and cheaper.
PoRe: Position-Reweighted Visual Token Pruning for Vision Language Models
CV and Pattern Recognition
Helps AI focus on important parts of pictures.