用小模型预测大模型增益,智能路由降低翻译成本
RouteLMT: Learned Sample Routing for Hybrid LLM Translation Deployment
- 基于边际收益构建路由机制,用小模型输出预测大模型改进程度
- 在同等预算下提升翻译质量,达成更优的性价比平衡
- 无需额外模型或解码,适合大规模部署场景
大型语言模型(LLMs)在机器翻译中表现优异,但大规模部署成本过高。现有混合系统通过小模型处理多数请求,仅将部分任务路由至大模型以平衡成本与质量。然而,当前路由策略多依赖启发式规则、外部预测器或绝对质量估计,难以判断大模型是否带来实质性提升。本文将路由问题建模为预算分配问题,提出以边际增益(即大模型相对于小模型的改进)作为最优决策信号。基于此,我们提出RouteLMT(用于基于LLM的MT路由),一种高效的模型内路由器,通过探测小模型的提示-标记表示来预测该预期增益,无需外部模型或假设解码。大量实验表明,RouteLMT优于启发式方法和质量/难度估计基线,在相同预算下实现更优的质量-预算帕累托前沿。此外,我们分析了回归风险,并证明简单防护变体可有效缓解严重质量损失。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have achieved remarkable performance in Machine Translation (MT), but deploying them at scale remains prohibitively expensive. A widely adopted remedy is the hybrid system paradigm, which balances cost and quality by serving most requests with a small model and selectively routing a fraction to a large model. However, existing routing strategies often rely on heuristics, external predictors, or absolute quality estimation, which fail to capture whether the large model actually provides a worthwhile improvement over the small one. In this paper, we formulate routing as a budget allocation problem and identify marginal gain, i.e., the large model's improvement over the small model, as the optimal signal for budgeted decisions. Building on this, we propose \textbf{RouteLMT} (routing for LLM-based MT), an efficient in-model router that predicts this expected gain by probing the small translators prompt-token representation, without requiring external models or hypothesis decoding. Extensive experiments demonstrate that our RouteLMT outperforms heuristics, quality/difficulty estimation baselines, achieving a superior quality-budget Pareto frontier. Furthermore, we analyze regression risks and show that a simple guarded variant can mitigate severe quality losses.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。