Search papers, labs, and topics across Lattice.
The paper introduces CLASP, a plug-and-play token reduction framework for MLLMs that uses class-adaptive layer fusion to create category-specific visual representations. It then employs a dual-stage pruning strategy, allocating tokens to both attention-salient "pivot" tokens and redundancy-aware "completion" tokens. Experiments show CLASP consistently outperforms existing methods across various benchmarks, pruning ratios, and MLLM architectures by enabling prompt-conditioned feature fusion and budget allocation.
MLLMs can aggressively prune visual tokens without sacrificing performance by adapting token reduction strategies to specific classes and prompts.
Multimodal Large Language Models (MLLMs) suffer from substantial computational overhead due to the high redundancy in visual token sequences. Existing approaches typically address this issue using single-layer Vision Transformer (ViT) features and static pruning strategies. However, such fixed configurations are often brittle under diverse instructions. To overcome these limitations, we propose CLASP, a plug-and-play token reduction framework based on class-adaptive layer fusion and dual-stage pruning. Specifically, CLASP first constructs category-specific visual representations through multi-layer vision feature fusion. It then performs dual-stage pruning, allocating the token budget between attention-salient pivot tokens for relevance and redundancy-aware completion tokens for coverage. Through class-adaptive pruning, CLASP enables prompt-conditioned feature fusion and budget allocation, allowing aggressive yet robust visual token reduction. Extensive experiments demonstrate that CLASP consistently outperforms existing methods across a wide range of benchmarks, pruning ratios, and MLLM architectures. Code will be available at https://github.com/Yunkaidang/CLASP.