arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.04428cs.CLcs.AIcs.LG

自信或寻求更强:从基准测试到泛化探索基于不确定性的端侧大语言模型路由

Confident or Seek Stronger: Exploring Uncertainty-Based On-device LLM Routing From Benchmarking to Generalization

Rice University(莱斯大学);Johns Hopkins University(约翰斯·霍普金斯大学);University of Minnesota(明尼苏达大学);Meta AI;University of California, San Diego(加利福尼亚大学圣迭戈分校)

Yu-Neng Chuang, Leisheng Yu, Guanchu Wang, Lizhe Zhang, Zirui Liu, Xuanting Cai, Yang Sui, Vladimir Braverman, Xia Hu

AI总结:

本文系统研究了端侧小语言模型到强大语言模型的不确定性路由策略,通过1500余种设置揭示不确定性量化方法与模型特性对路由性能的影响,并提出校准数据构建流水线以提升跨数据集泛化能力。

摘要:

大语言模型(LLMs)正越来越多地部署并普及到边缘设备上。为提高端侧部署效率,通常采用小语言模型(SLMs),因为它们具有高效的解码延迟和更低的能耗。然而,这些SLMs在处理复杂查询时往往会产生不准确的响应。一种有前景的解决方案是基于不确定性的SLM路由,即当SLM产生低置信度响应时,将高风险查询卸载到更强的LLMs。这遵循了“如果缺乏信心,就寻求更强支持”的原则以增强可靠性。依赖更强大的LLMs虽然有效,但会增加调用成本。因此,在效率和效果之间取得路由平衡仍然是一个关键挑战。此外,如何高效地将路由策略泛化到新数据集仍未得到充分探索。在本文中,我们对从SLMs到LLMs的不确定性驱动路由策略进行了全面的基准测试和泛化研究,覆盖1500多个设置。我们的发现表明:第一,不同不确定性量化(UQ)方法中的不确定性-正确性对齐显著影响路由性能。第二,不确定性分布更多地取决于特定的SLM和所选的UQ方法,而不是下游数据。基于这一见解,我们提出了一条校准数据构建指令流水线,并开源了一个构建的留出集,以增强路由在新下游场景中的泛化能力。实验结果表明,校准数据在没有任何新数据的情况下有效地引导了路由性能。

英文摘要:

Large language models (LLMs) are increasingly deployed and democratized on edge devices. To improve the efficiency of on-device deployment, small language models (SLMs) are often adopted due to their efficient decoding latency and reduced energy consumption. However, these SLMs often generate inaccurate responses when handling complex queries. One promising solution is uncertainty-based SLM routing, offloading high-stakes queries to stronger LLMs when resulting in low-confidence responses on SLM. This follows the principle of "If you lack confidence, seek stronger support" to enhance reliability. Relying on more powerful LLMs is yet effective but increases invocation costs. Therefore, striking a routing balance between efficiency and efficacy remains a critical challenge. Additionally, efficiently generalizing the routing strategy to new datasets remains under-explored. In this paper, we conduct a comprehensive investigation into benchmarking and generalization of uncertainty-driven routing strategies from SLMs to LLMs over 1500+ settings. Our findings highlight: First, uncertainty-correctness alignment in different uncertainty quantification (UQ) methods significantly impacts routing performance. Second, uncertainty distributions depend more on both the specific SLM and the chosen UQ method, rather than downstream data. Building on the insight, we propose a calibration data construction instruction pipeline and open-source a constructed hold-out set to enhance routing generalization on new downstream scenarios. The experimental results indicate calibration data effectively bootstraps routing performance without any new data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑