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

CATP: Contextually Adaptive Token Pruning for Efficient and Enhanced Multimodal In-Context Learning

About

Modern large vision-language models (LVLMs) convert each input image into a large set of tokens that far outnumber the text tokens. Although this improves visual perception, it also introduces severe image token redundancy. Because image tokens contain sparse information, many contribute little to reasoning but greatly increase inference cost. Recent image token pruning methods address this issue by identifying important tokens and removing the rest. These methods improve efficiency with only small performance drops. However, most of them focus on single-image tasks and overlook multimodal in-context learning (ICL), where redundancy is higher and efficiency is more important. Redundant tokens weaken the advantage of multimodal ICL for rapid domain adaptation and lead to unstable performance. When existing pruning methods are applied in this setting, they cause large accuracy drops, which exposes a clear gap and the need for new approaches. To address this, we propose Contextually Adaptive Token Pruning (CATP), a training-free pruning method designed for multimodal ICL. CATP uses two stages of progressive pruning that fully reflect the complex cross-modal interactions in the input sequence. After removing 77.8% of the image tokens, CATP achieves an average performance gain of 0.6% over the vanilla model on four LVLMs and eight benchmarks, clearly outperforming all baselines. At the same time, it improves efficiency by reducing inference latency by an average of 10.78%. CATP strengthens the practical value of multimodal ICL and lays the foundation for future progress in interleaved image-text settings.

Yanshu Li, Jianjiang Yang, Zhennan Shen, Ligong Han, Haoyan Xu, Ruixiang Tang• 2025

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringOCRVQA
Accuracy12
47
Object Shuffling PredictionObjShuf
Accuracy40
36
Image EditingIEdit
Accuracy10.12
36
Object Existence PredictionObjExist
Accuracy51.5
36
Instruction FollowingALFRED
Accuracy17.82
36
Counterfactual InferenceCFInfer
Accuracy34
36
Multimodal Question AnsweringMMQA
Accuracy66.5
36
Moving Direction PredictionMoveDir
Accuracy32.5
36
Action PredictionActPred
Accuracy0.515
36
OCR-based Visual Question AnsweringOCR-VQA
Accuracy45.5
35
Showing 10 of 11 rows

Other info

Follow for update