Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

TrimTokenator: Towards Adaptive Visual Token Pruning for Large Multimodal Models

About

Large Multimodal Models (LMMs) have achieved significant success across various tasks. These models usually encode visual inputs into dense token sequences, which are then concatenated with textual tokens and jointly processed by a language model. However, the increased token count substantially raises computational and memory costs during inference. Token pruning has emerged as a promising approach to address this issue. Existing token pruning methods often rely on costly calibration or suboptimal importance metrics, leading to redundant retained tokens. In this paper, we analyze the redundancy differences between visual and textual tokens and propose pruning exclusively on visual tokens. Based on this, we propose a visual token pruning strategy that explicitly preserves both cross-modal alignment and intra-modal informational diversity. We introduce a mutual information-based token pruning strategy that removes visual tokens semantically misaligned with textual tokens, effectively preserving the alignment between the visual and textual modalities. To further improve the representational quality of the retained tokens, we additionally prune redundant visual tokens by maximizing the expected pairwise distances in the embedding space, which is solved efficiently with a greedy algorithm. Extensive experiments demonstrate that our method maintains strong performance while reducing tokens by 88.9% on models such as LLaVA-1.5-7B and LLaVA-NEXT-7B, resulting in a 56.7% improvement in inference speed.

Hao Zhang, Mengsi Lyu, Chenrui He, Yulong Ao, Yonghua Lin• 2025

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringOCRVQA
Accuracy11.5
47
Instruction FollowingALFRED
Accuracy18.67
36
Action PredictionActPred
Accuracy0.54
36
Image EditingIEdit
Accuracy10.61
36
Object Existence PredictionObjExist
Accuracy53.5
36
Counterfactual InferenceCFInfer
Accuracy36.5
36
Multimodal Question AnsweringMMQA
Accuracy67
36
Moving Direction PredictionMoveDir
Accuracy33.5
36
Object Shuffling PredictionObjShuf
Accuracy39.5
36
OCR-based Visual Question AnsweringOCR-VQA
Accuracy47
35
Showing 10 of 10 rows

Other info

Follow for update