Before Thinking, Learn to Decide: Proactive Routing for Efficient Visual Reasoning
Authors: Yinan Zhou, Haokun Lin, Yichen Wu, Yuxin Chen, Teng Wang, Caifeng Shan, Zhenan Sun, Chen Ma, Li Zhu, Ying Shan
Abstract
Large multimodal models have achieved strong reasoning oncomplex visual tasks, but their inference efficiency is often restricted bylong chains of thought. A promising solution is to pair a small draftmodel with a large target model, enabling cooperative inference employ-ing a routing signal that adaptively routes queries to either the draft ortarget model based on their difficulties for optimal efficiency and accu-racy. Yet, the remaining bottleneck is to establish a reliable query dif-ficulty signal under multimodal settings. Existing approaches designedfor language models either rely on post-hoc token probabilities, whichfall short in multimodal scenarios, or depend on supervised fine-tuning,which is a data-sensitive strategy. Both paradigms perform routing onlyafter a complete output, and ignore whether the target model can ac-tually solve the routed instances. To address this, we propose PRP,a Proactive Routing Paradigm that enables early decision-making byjointly evaluating the competence of both the draft and target models.Our Draft Rating Learning (DRL) equips the draft model with an inter-nal confidence estimator, while Joint Rating Learning (JRL) predictshow well the target model can handle a given query, thereby prioritiz-ing the allocation of samples it excels at rather than the hardest ones.These ratings enable fine-grained, instance-level Proactive Routingand substantially accelerate inference without compromising overall per-formance. Extensive experiments across multiple multimodal reasoningbenchmarks validate our effectiveness and efficiency.