预测大模型对提示的平均得分,实现高效推理路由。
Expected Reward Prediction, with Applications to Model Routing
- 通过重复采样预测模型在提示上的期望奖励得分。
- 在多个模型组合中,该方法优于基于平均表现的路由基线。
- 可无缝扩展新模型,适合动态模型池场景。
奖励模型是评估大语言模型生成响应的标准工具,通常用于对同一提示下生成的多个响应进行排序。本文研究将响应级奖励模型的评分推广至预测模型对特定提示的适用性,即在未生成任何响应前预测该模型的期望奖励。我们证明,通过重复采样即可简单而准确地预测期望奖励。该预测具备足够精度和区分度,可用于构建推理时的模型路由协议:根据提示分配最优模型以最大化奖励,同时控制计算成本。我们在 open-perfectblend 数据集上验证了该方法,模型池包含 Llama3.1-Instruct 8B/70B、Gemma2-IT 9B/27B 与 Gemma1-IT 7B。基于期望奖励预测的路由(ERP)显著优于按提示类别选择平均表现最佳模型的基线,并解释了复杂路由策略的成功机制。该方法在新增模型时可零成本扩展。
原文摘要 · Abstract (English)
Reward models are a standard tool to score responses from LLMs. Reward models are built to rank responses to a fixed prompt sampled from a single model, for example to choose the best of n sampled responses. In this paper, we study whether scores from response-level reward models lifted to score a model's suitability for a prompt, prior to seeing responses from that model. Specifically, we show that it is straightforward to predict the expected reward that an LLM would earn from the reward model under repeated sampling. Further, we show that these expected reward predictions are precise and discriminative enough to support an application to a model routing protocol that routes prompts to models at inference time to maximize reward while controlling computational cost. We demonstrate the performance of this routing procedure on the open-perfectblend dataset, using a model pool composed of Llama3.1-Instruct 8B/70B, Gemma2-IT 9B/27B, and Gemma1-IT 7B models. Our simple expected reward prediction--based routing (ERP) outperforms baselines that route prompts to models with the best average performance within each prompt's category, and explains the success of more complex routing protocols that implicitly estimate an expected reward. Our approach has the added advantage of being trivially extensible as new models are added to the pool.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。