Search papers, labs, and topics across Lattice.
This paper introduces a training-free token pruning framework, IWP, for large vision language models based on a dual-form perspective of attention, reformulating attention as an implicit linear layer. The core idea is to prune tokens by selecting the optimal subset of rank-1 updates (key-value pairs) that best approximates the original attention weight matrix. Experiments show IWP achieves a better performance-efficiency trade-off compared to existing token pruning methods.
Token pruning can be reframed as implicit weight pruning, offering a novel training-free approach that significantly improves the performance-efficiency trade-off in large vision language models.
Large Vision Language Models show impressive performance across image and video understanding tasks, yet their computational cost grows rapidly with the number of visual tokens. Existing token pruning methods mitigate this issue through empirical approaches while overlooking the internal mechanism of attention. In this paper, we propose a novel training free token pruning framework grounded in the dual form perspective of attention. We reformulate attention as an implicit linear layer whose weight matrix is the sum of rank 1 outer products, each generated by a single token's key value pair. Token pruning thus reduces to selecting an optimal subset of these rank 1 updates that best approximates the original dual weight matrix. Extending this perspective to standard softmax attention in LVLMs, we derive a novel metric quantifying both a token's information magnitude and information duplication. To efficiently select the subset with the proposed metric, we introduce Progressive Chunked Maximal Marginal Relevance. Extensive experiments demonstrate that our method achieves a better trade off between performance and efficiency, while providing another perspective on existing pruning approaches.
