Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Reroute, Don't Remove: Recoverable Visual Token Routing for Vision-Language Models

About

Vision-language models (VLMs) project images into hundreds to thousands of visual tokens, making decoder inference expensive in both attention computation and KV-cache memory. Existing visual-token reduction methods largely follow a rank-and-remove paradigm: they score visual tokens, keep a compact subset, and permanently discard the rest. We show that this irreversible action is fragile because visual-token importance changes across decoder depth; tokens ranked low at one stage may become relevant in later layers, especially for grounding-sensitive queries. We propose Reroute, a training-free plug-in that replaces removal with recoverable routing. At each routing stage, selected vision tokens pass through decoder blocks, while deferred tokens bypass the stage and re-enter the candidate pool at the next routing decision. Reroute reuses existing attention-score ranking rules and stage-wise schedules, preserving the theoretical TFLOPs and KV-cache budget class of the pruning method it augments. Across FastV, PDrop, and N\"uwa variants on LLaVA-1.5 and Qwen backbones, reroute improves grounding under aggressive token reduction while maintaining general VQA performance. These results suggest that VLM token reduction should not be viewed only as irreversible pruning, but also as recoverable routing. The code can be found here: https://github.com/elmma/mllm-reroute/

Cheng-Yu Yang, Shao-Yuan Lo, Yu-Lun Liu• 2026

Related benchmarks

TaskDatasetResultRank
Science Question AnsweringScienceQA
Accuracy86.9
916
Multimodal EvaluationMME
Score2.04e+3
902
Multimodal UnderstandingSEED-Bench
Accuracy76.8
571
Multimodal UnderstandingMMStar
Accuracy48.4
511
Diagram Question AnsweringAI2D
AI2D Accuracy74.6
509
Multi-discipline Multimodal UnderstandingMMMU
Accuracy49.6
422
Visual Question AnsweringAI2D
Accuracy77.3
402
Visual GroundingRefCOCO+ (val)--
264
Object Hallucination EvaluationPOPE
Accuracy87.1
259
Visual GroundingRefCOCO+ (testA)--
256
Showing 10 of 37 rows

Other info

GitHub

Follow for update