Moving Beyond Diversity: Visual Token Pruning as Subspace Reconstruction for Efficient VLMs
Summary
SPARE is a novel subspace reconstruction method designed to efficiently prune visual tokens in Vision Language Models (VLMs), addressing their substantial computational overhead. Unlike existing diversity maximization techniques that rely on cosine similarity and discard crucial magnitude information, SPARE reformulates token pruning as a column subset selection problem, explicitly minimizing reconstruction error. The method iteratively selects tokens based on large projection residuals, moving beyond simple angular diversity. Furthermore, SPARE integrates an "anti-relevance" criterion, leveraging the counterintuitive finding that tokens with lower image-text relevance can better preserve contextual information. Extensive experiments show SPARE consistently achieves strong performance, particularly on compositional multi-skill reasoning tasks. For instance, when applied to LLaVA, SPARE removes up to 94% of visual tokens while maintaining 95% of the baseline performance, all without requiring additional training.
Key takeaway
For Machine Learning Engineers optimizing Vision Language Model inference, especially on compositional multi-skill reasoning tasks, you should evaluate SPARE. This training-free subspace reconstruction method offers superior token pruning by minimizing reconstruction error and integrating anti-relevance, outperforming traditional diversity-based approaches. You can achieve up to 94% visual token reduction while retaining 95% baseline performance, significantly improving VLM efficiency without complex retraining.
Key insights
SPARE prunes VLM visual tokens by minimizing reconstruction error and incorporating anti-relevance for context preservation.
Principles
- Cosine similarity-based pruning discards magnitude.
- Token pruning can be a column subset selection.
- Lower image-text relevance can preserve context.
Method
SPARE iteratively selects visual tokens by minimizing reconstruction error via large projection residuals, then integrates anti-relevance for context-aware selection.
In practice
- Apply SPARE to LLaVA for 94% token reduction.
- Improve VLM efficiency on compositional tasks.
- Use training-free token pruning for VLMs.
Topics
- Vision Language Models
- Token Pruning
- Subspace Reconstruction
- Computational Efficiency
- LLaVA
- Compositional Reasoning
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, Computer Vision Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.