arXiv:2506.05901cs.CLcs.AI2025-06被引 12

用强化学习路由器调度多模型协作,大幅降低复杂推理成本。

Route-and-Reason: Scaling Large Language Model Reasoning with Reinforced Model Router

  • 通过分解问题并分配给最优模型,实现跨异构大模型的细粒度协作。
  • 在六项基准测试中,推理成本降低84.46%,准确率仍具竞争力。
  • 适合需要高效推理、低成本部署的AI系统开发者使用。

链式思维已被证明能显著提升大语言模型的复杂推理能力,但同时带来高昂的计算开销。近期研究探索了在多个模型间路由查询的方法,显示出良好前景。然而,以往工作仅在任务层面进行分配,无法实现模型在中间推理步骤上的精细协同。为解决此问题,我们提出R2-Reasoner框架,核心是一个强化学习驱动的模型路由器,可协调九个参数量从不足10亿到数百亿不等的异构模型。该框架先由分解器将复杂问题拆分为子任务,再由子任务分配器将每个子任务分配至最合适的模型,兼顾性能与成本。路由器训练采用两阶段交替优化,结合监督微调与强化学习,实现自监督精炼。在六项挑战性推理基准上的实验表明,相比当前最优基线,R2-Reasoner将API成本降低84.46%,同时保持具有竞争力的推理准确率。本框架为构建更高效、可扩展的推理系统开辟了新路径。代码已开源:https://anonymous.4open.science/r/R2_Reasoner。

原文摘要 · Abstract (English)

Chain-of-thought has been proven essential for enhancing the complex reasoning abilities of Large Language Models (LLMs), but it also leads to high computational costs. Recent advances have explored the method to route queries among multiple models and proved it as a promising approach. However, previous works directly operate at the task level, i.e., assigning user queries to suitable LLMs, which does not allow hybrid LLMs to truly collaborate on finer-grained sub-tasks. Collaboration at the level of intermediate reasoning steps (thoughts) could enable more efficient coordination, but it also poses significant challenges for router scheduling, placing immense demands on the quality of task decomposition and the precision of the router. To address this, we propose R2-Reasoner, a novel framework centered around a Reinforced Model Router designed to efficiently scale LLM reasoning. This router orchestrates collaboration across nine heterogeneous models, whose parameter scales range from less than 1B to hundreds of billions, by first breaking down a complex query into subtasks with a decomposer, and then assigning each subtask to the optimal model with a subtask allocator, balancing performance with cost. Training this router involves a two-stage alternating process for the decomposer and the allocator, integrating supervised fine-tuning with reinforcement learning to enable effective self-supervised refinement. Extensive experiments across six challenging reasoning benchmarks demonstrate that R2-Reasoner reduces API costs by 84.46% compared with state-of-the-art baselines while maintaining competitive reasoning accuracy. Our framework paves the way for the development of more scalable and efficient reasoning systems. Our code is open-source at https://anonymous.4open.science/r/R2_Reasoner.

大模型推理强化学习路由优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。