跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

UCCI:面向成本最优LLM级联路由的校准不确定性

文章摘要

UCCI提出一种“校准优先”的LLM级联路由方法:通过等渗回归将token级margin不确定性映射为每条查询的出错概率,并以受限成本最小化选择升级阈值。在生产级命名实体识别负载上,该方法在micro-F1=0.91时将推理成本降低31%,并将ECE从0.12降至0.03,优于熵阈值、split-conformal路由和FrugalGPT式学习阈值。

来源arXiv Machine Learning作者: Varun Kotte
UCCI:面向成本最优LLM级联路由的校准不确定性
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

LLM 级联与模型路由的核心思路,是让简单查询交给小模型处理、把困难查询升级给大模型,从而降低推理成本。然而论文指出,目前大多数已部署的路由器依赖未校准的置信度分数,并且需要针对每个工作负载单独调整阈值,这限制了它们在真实生产环境中的可靠性与可移植性。针对这一问题,作者 Varun Kotte 提出了 UCCI,一种“校准优先”的路由器。该论文于 2026 年 5 月 11 日提交至 arXiv,编号 2605.18796,属于机器学习(cs.LG)与计算与语言(cs.CL)方向,全文共 9 页、2 张图、4 张表,并附有代码链接。

UCCI 的做法是:首先把 token 级别的 margin 不确定性,通过等渗回归(isotonic regression)映射为每个查询的误差概率,然后通过带约束的成本最小化来选择升级阈值。作者给出了理论层面的论证:在三个明确假设下,基于校准分数的阈值策略具有成本最优性;同时,等渗校准在期望校准误差(ECE)上达到 O(n^{-1/3}) 的样本复杂度。这意味着该方法不仅有工程上的收益,也在统计学习理论上有相应的保证。

在实验方面,作者采用了一个生产级的命名实体识别(NER)工作负载,共包含 75,000 条查询,由运行在 H100 GPU 上的 4B 与 12B 指令微调模型共同服务。结果显示,在 micro-F1 达到 0.91 的情况下,UCCI 将推理成本降低了 31%(95% 置信区间为 [27%, 35%]),同时把期望校准误差从 0.12 降至 0.03。在相同的工作点上,UCCI 的表现优于熵阈值方法、split-conformal 路由,以及 FrugalGPT 风格的习得阈值方案。

论文特别强调了实验设置的严谨性:所有级联结果都基于真实模型输出的端到端路由,并使用实测的 H100 延迟,而不是依据全局准确率或名义 API 价格进行的模拟路由。这一细节对于评估路由方法的实际价值尤为关键,因为模拟环境往往会高估或低估真实部署中的成本收益。总体而言,UCCI 将不确定性校准与成本约束优化结合起来,为 LLM 级联路由提供了一条兼顾理论保证与生产可用性的路径。

展开要点与分析

文章情报

工程师进阶

要点

  • 将token级margin不确定性经等渗回归校准为每条查询的错误概率,ECE从0.12降至0.03
  • 通过受限成本最小化选择升级阈值,在三项明确假设下阈值策略达到成本最优
  • 在75,000条查询的生产NER负载上,以4B与12B模型和实测H100延迟实现31%成本下降

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。