Alliance of Stochastic Routing and Reinforcement Fusion for Multi-Model Reasoning
Organizations: Georgia Institute of Technology
Abstract
This paper studies dynamic model routing under uncertainty in multimodal settings, where a system must select and potentially compose multiple models to answer a query. Standard reinforcement learning approaches for routing often suffer from premature convergence to suboptimal models due to noisy rewards and insufficient exploration. We propose a unified vision-language framework (VL-Scout) for efficient multimodal and multi-step reasoning with turn-based reinforcement fusion. First, VL-Scout combines Bayesian priors, policy gradient optimization, and sequential decision-making to enable structured exploration and adaptive inference. A context-dependent Dirichlet belief model allows for epistemic uncertainty over model quality and is updated online based on observed rewards and context-specific evidence, enabling Thompson-style exploration while serving as a prior for regularization. Second, a routing policy is trained via a KL-regularized REINFORCE objective by maximizing a reward-weighted evidence lower bound, where the policy acts as an amortized posterior, and the Dirichlet belief serves as a prior. Third, we introduce a fusion policy trained with Group Relative Policy Optimization (GRPO) or Reinforcement Learning with Verifiable Rewards (RLVR), which operates over intermediate model rationales and selects between two actions: CONTINUE or ANSWER, terminating with a generated output. We evaluate VL-Scout on four representative multi-modal benchmarks (MMMU, MMMU-Pro, OKVQA, and RealWorldQA), compared against 11 representative routing and policy optimization baselines. Our method provides consistent performance improvements while reducing premature policy collapse. These results highlight the importance of integrating uncertainty-aware priors with both amortized policy learning and sequential fusion for robust and efficient model selection in discrete action spaces
Figures & tables
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Category | Description |
| spatial_and_visual | Where, position, layout, scene understanding |
| counting_and_quantity | How many, amounts, numbers |
| text_and_document | OCR, reading, documents, charts |
| science_and_math | Physics, chemistry, biology, math, engineering |
| humanities_and_arts | History, literature, music, art, philosophy |
| social_and_applied | Economics, law, medicine, psychology, business |
| Prompt | Description | Content |
| VQA System Prompt | Base prompt. Instructs the model to output the final answer only, no reasoning. | You are a visual question answering assistant. |
| For every question, first write a short rationale | ||
| describing what you observe and how you reason, then give | ||
| your final answer. Always use this format: | ||
| Rationale: [your reasoning] | ||
| Answer: [your final answer]] |
| Question: Is the closest car closer to us than the pedestrian? | ||||
| Model | Rationale | Answer | Decision | Correct? |
| Qwen (Model 1) | The pedestrian is visible on the sidewalk to the right, standing near a tree and some snow piles. The closest car is further down the street, behind the pedestrian. Since the pedestrian is on the sidewalk and the car is on the road and further away, the pedestrian is closer to the viewer than the car. | Yes | CONTINUE | |
| InternVL (Model 2) | The closest car is on the left side of the road, while the pedestrian is on the right side. By comparing their relative positions to the camera’s viewpoint, the car appears closer to the viewer than the pedestrian. | Yes | CONTINUE | |
| Gemma (Model 3) | The image shows a street scene. The closest car is driving down the middle of the road, relatively far away. The pedestrian is walking on the sidewalk on the right side, and is much closer to the camera than the car. Therefore, the pedestrian is closer to us than the closest car. | 1 | ANSWER: No | ✓ |
| Ground Truth | No | ✓ | ||
| Question: Is there a stop sign? | ||||
| Model | Rationale | Answer | Decision | Correct? |
| InternVL (Model 1) | To determine if there is a stop sign, I examine the street for any octagonal signs with a red border and white background. Upon reviewing the image, I see several signs along the street, but none match the description of a stop sign. The signs visible are either rectangular or do not have the characteristic red border and white center. | No | CONTINUE | ✓ |
| Qwen (Model 2) | I can see a white SUV parked on the left side of the street, and behind it, there appears to be a stop sign mounted on a pole near the sidewalk. The sign is partially obscured by the SUV and the angle of the photo, but its shape and position are consistent with a stop sign. Therefore, based on the visual evidence, there is at least one stop sign present. | Yes | CONTINUE | |
| Gemma (Model 3) | I see a street with cars parked on the side and yellow crosswalk markings. I do not see any stop signs visible in the image. | 0 | ANSWER: No | ✓ |
| Ground Truth | No | ✓ | ||
Adaptive Model Routing
Adaptive model routing selects models or specialized experts per input to balance prediction quality, computational cost, latency, and resource constraints.
See all 87 papers