用强化学习直接优化模型路由,不依赖质量预测器。
RLCascadeRouter: Quality-Estimator-Free Cascade Routing via Reinforcement Learning

- 将路由过程建模为马尔可夫决策过程,直接优化性能与成本权衡。
- 在10个基准上用13个大模型测试,效果优于现有方法。
- 无需重新训练即可适配新模型,适合部署于异构LLM系统。
大型语言模型(LLM)生态日益丰富,但其能力与推理成本的异质性使得高效查询路由成为挑战。现有方法存在灵活性不足:一次性路由在看到响应前就决定,传统级联路由虽能自适应停止,但遵循固定模型顺序。级联路由通过在每次响应后重新判断是否停止或调用下一模型,突破了上述限制。当前方法采用先预测再优化的流程,依赖对响应质量与未来模型效用的估计,但预测误差与路由决策损失并不等价;小的预测偏差可能导致“停止”或模型选择决策反转。为此,本文提出 RLCascadeRouter,一种无需质量估计器的级联路由框架,将级联路由建模为包含“停止”与模型选择动作的马尔可夫决策过程,利用轨迹回报与优势值直接优化性能-成本目标。其级联策略网络同时建模候选模型互补性与剩余动作价值,省去独立的事后质量估计器。在包含13个大模型的10个LLMRouterBench基准上评估,RLCascadeRouter显著优于强基线,实现更优的性能-成本权衡。该方法无需重训练即可集成新模型,消融实验验证了策略组件的有效性。
原文摘要 · Abstract (English)
The growing ecosystem of large language models (LLMs) offers huge potential to optimize performance-cost trade-offs. However, their heterogeneous capabilities and inference costs make efficiently routing queries a significant challenge. Existing paradigms are inflexible: one-shot routers commit before observing responses, whereas conventional cascades stop adaptively but follow a fixed model order. Cascade routing removes both restrictions by reconsidering whether to stop or invoke another model after each response. Current methods use a predict-then-optimize pipeline estimating response quality and future model utility. However, prediction loss for quality or utility is not equivalent to routing-decision loss. A lower prediction error does not necessarily yield a better action; a small boundary-crossing error can reverse a ``stop'' or model-selection decision. Therefore, we propose RLCascadeRouter, a quality-estimator-free framework that formulates cascade routing as a Markov decision process with actions comprising ``stop'' and model selection. It uses trajectory returns and advantages to directly optimize the performance-cost objective. Its Cascade Policy Network models candidate complementarity for model selection and remaining-action value for stopping, eliminating independent post-hoc response-quality estimators. Evaluated across ten LLMRouterBench benchmarks with thirteen LLMs, RLCascadeRouter outperforms strong baselines and achieves superior performance-cost trade-offs. It incorporates unseen models without retraining, and ablation studies validate both policy components.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。