用过程评分提升推理模型路由效率与准确率
Rubric-Guided Process Reward for Stepwise Model Routing

- 基于评分标准生成中间步骤评价,替代仅看最终答案的奖励
- 在5个基准上实现更优准确率与成本平衡,跨家族设置也稳定领先
- 适合需要高效推理决策的AI系统开发者
逐步模型路由通过将每个推理步骤分配给合适模型来提升大推理模型(LRMs)的效率。现有方法将路由建模为序列决策过程,并用强化学习训练路由器,但其仍以最终答案正确性作为监督信号,无法评估中间路由决策质量,削弱了性能与泛化能力。为此,我们提出RoRo——一种基于评分标准的过程奖励框架。RoRo首先收集多样路由轨迹,依据结果、成本与过程质量构建偏好对;接着通过交替优化训练一个评分器(Rubricor)生成特定查询的评价标准,以及一个裁判(Judge)在该标准下对路由轨迹打分。最终,过程奖励与结果奖励结合,通过GRPO优化路由策略。在五个推理基准上,无论同族还是跨族设置,RoRo均持续优于强基线,取得更优的准确率与成本权衡。
原文摘要 · Abstract (English)
Stepwise model routing improves the efficiency of Large Reasoning Models (LRMs) by assigning each reasoning step to a suitable model. Recent methods formulate routing as a sequential decision process and train the router with reinforcement learning. However, although they model routing as a process, they still supervise the router with outcome rewards. Such rewards only reflect final answer correctness and fail to evaluate intermediate routing decisions, which can weaken performance and generalization. To address this gap, we propose RoRo, a rubric-guided process reward framework for stepwise model routing. RoRo first collects diverse routing trajectories and constructs preference pairs based on outcome, cost, and process quality. It then trains a Rubricor to generate a query-specific evaluation rubric and a Judge to score routing trajectories under this rubric through alternating optimization. The resulting process rewards are combined with outcome rewards to optimize the routing policy via GRPO. Experiments on five reasoning benchmarks under both same-family and cross-family settings show that RoRo consistently outperforms strong baselines and achieves better accuracy and cost trade-offs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。