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

TrimTokenator-LC: Towards Adaptive Visual Token Pruning for Large Multimodal Models with Long Contexts

About

Large Multimodal Models (LMMs) have proven effective on various tasks. They typically encode visual inputs into Original Model sequences of tokens, which are then concatenated with textual tokens and jointly processed by the language model. However, the growing number of visual tokens greatly increases inference cost. Visual token pruning has emerged as a promising solution. However, existing methods often overlook scenarios involving long context inputs with multiple images. In this paper, we analyze the challenges of visual token pruning in long context, multi-image settings and introduce an adaptive pruning method tailored for such scenarios. We decompose redundancy into intra-image and inter-image components and quantify them through intra-image diversity and inter-image variation, which jointly guide dynamic budget allocation. Our approach consists of two stages. The intra-image stage allocates each image a content-aware token budget and greedily selects its most representative tokens. The inter-image stage performs global diversity filtering to form a candidate pool and then applies a Pareto selection procedure that balances diversity with text alignment. Extensive experiments show that our approach can reduce up to 80% of visual tokens while maintaining performance in long context settings.

Hao Zhang, Mengsi Lyu, Bo Huang, Yulong Ao, Yonghua Lin• 2025

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringOCRVQA
Accuracy13
47
Action PredictionActPred
Accuracy0.56
36
Image EditingIEdit
Accuracy11.34
36
Instruction FollowingALFRED
Accuracy19.84
36
Object Existence PredictionObjExist
Accuracy55.5
36
Object Shuffling PredictionObjShuf
Accuracy42.5
36
Counterfactual InferenceCFInfer
Accuracy38
36
Moving Direction PredictionMoveDir
Accuracy35.5
36
Multimodal Question AnsweringMMQA
Accuracy69
36
OCR-based Visual Question AnsweringOCR-VQA
Accuracy49.5
35
Showing 10 of 11 rows

Other info

Follow for update