Moving Beyond Diversity: Visual Token Pruning as Subspace Reconstruction for Efficient VLMs

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

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

Method

SPARE iteratively selects visual tokens by minimizing reconstruction error via large projection residuals, then integrates anti-relevance for context-aware selection.

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.