Search papers, labs, and topics across Lattice.
This paper introduces S$^2$Prune, a novel method for visual token pruning in multimodal large language models (MLLMs) that prioritizes spatial coverage and local image structure over traditional importance-based selection. By dividing images into regions and allocating tokens based on Laplacian variation, S$^2$Prune effectively maintains performance while significantly reducing the number of tokens used鈥攁chieving 79.3% of full-model accuracy with only 32 out of 576 tokens. The method outperforms existing training-free pruning techniques, demonstrating the importance of spatially structured token selection in enhancing MLLM efficiency.
Retaining 79.3% of full performance with just 32 visual tokens reveals that spatial coverage trumps traditional importance metrics in multimodal models.
Visual token pruning reduces the inference overhead of multimodal large language models (MLLMs) by retaining only a subset of visual tokens. Existing methods usually select tokens based on importance or redundancy. However, we observe that these criteria produce stable spatial biases across inputs and do not always outperform simple Uniform Grid sampling, highlighting the value of broad spatial coverage. Motivated by this, we propose S$^2$Prune, a training-free pruning method that preserves spatial coverage while adapting token density to local image structure. We first divide the image into regions and assign at least one token to each region to preserve coverage. The remaining token budget is then distributed according to Laplacian variation, giving more tokens to regions with richer structure. We then use Early Representation Change (ERC), computed from the first decoder block, to select representative tokens within each region. We evaluate S$^2$Prune across diverse settings and two MLLM architectures. On Qwen2.5-VL-7B-Instruct, it achieves the highest average accuracy among the evaluated training-free pruning methods. With only 32 of the original 576 visual tokens, it still retains 79.3% of the full-model performance. Code is available at https://github.com/yuanyuanjia71-spec/S2Prune.
