Cheng-Yu Yang, Shao-Yuan Lo, Yu-Lun Liu
4 min
Abstract
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üwa 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/
Sam: That's one of the more practical aspects of the design. The model already produces ranking scores as part of its normal operation—it's part of how the attention mechanism works, which is the process by which the model figures out which parts of the image to focus on. The authors just reuse those scores to decide what gets rerouted. There's no extra training required. You can take an existing model—they tested it on systems like LLaVA and Qwen—and plug this method straight in.
Alex: And the results hold up?
Sam: The results are consistent across the models they tested. The area where the improvement was most visible was in what's called "grounding"—the model's ability to correctly locate and point to a specific object in an image when asked about it. That task is quite sensitive to losing visual information, so it's a good stress test. Under the most aggressive efficiency settings, the standard deletion methods would often fail badly, while the Reroute approach stayed stable.
Alex: So it's less about raw speed and more about preserving the quality of understanding even as you push the model harder.
Sam: That's a good way to put it. The deeper point the paper is making is that information in these networks doesn't have a fixed value. Something that seems irrelevant at step three might be exactly what the model needs at step seven. Permanent deletion assumes you can know the future importance of a token from an early snapshot—and that assumption, the authors suggest, is often wrong. Rerouting is a way of staying humble about what you don't yet know.
Alex: That's a useful principle well beyond AI, honestly. Thanks for walking us through it, Sam, and thanks to everyone listening to ResearchPod.