Search papers, labs, and topics across Lattice.
This paper addresses the inefficiencies of large vision-language models (LVLMs) by introducing a novel approach to diversity-based visual token pruning. The authors reveal that centering token features enhances the resolution of pairwise similarities but paradoxically degrades pruning performance when used in isolation due to the loss of globally distinctive token preferences. By developing the Centered Geometry Pruner (Cen-Prune), which combines centered cosine similarity with raw-space distinctiveness, they achieve significant performance improvements across various benchmarks while maintaining low computational costs.
Centering visual token features can enhance diversity detection but may paradoxically reduce the effectiveness of pruning鈥攗nless you leverage both centered and raw geometries together.
Large vision-language models (LVLMs) incur substantial inference costs due to their long and highly redundant visual-token sequences. Diversity-based pruning mitigates this cost by selecting token subsets based on pairwise cosine similarity. We find, however, that similarities between raw visual tokens are strongly concentrated in the positive range, limiting their ability to distinguish non-redundant tokens. A natural way to improve this resolution is to center token features before computing cosine similarity. Centering indeed reveals a substantially richer pairwise structure, yet unexpectedly degrades pruning performance when used alone. We show that this apparent contradiction arises because the raw geometry does more than represent pairwise diversity: it also implicitly favors globally distinctive tokens, which tend to contain semantically informative content. Centering better resolves subset diversity but loses this useful token-wise preference, revealing that diversity and distinctiveness are entangled in the raw geometry. Based on this analysis, we propose the \textbf{Cen}tered Geometry \textbf{Prune}r (Cen-Prune), which measures subset diversity using centered cosine similarity while retaining raw-space distinctiveness as a complementary token-wise preference. This lightweight, plug-and-play correction leaves the underlying selection mechanism unchanged and incurs negligible computational overhead. Extensive experiments across multiple image- and video-understanding benchmarks and LVLM architectures demonstrate that Cen-Prune provides robust improvements in overall performance across existing diversity-based pruners.
