中文

ProxRouter:面向鲁棒性的近似加权 LLM 查询路由

机器学习 2025-10-14 v1 人工智能

摘要

大型语言模型(LLM)查询路由器是现代人工智能平台的关键组件,旨在通过将推理查询分配给准确且低成本的模型来提高效率。参数化路由器通常使用训练好的神经网络进行 LLM 选择,但 suffer from 重训练和维护开销。非参数化路由器无需训练,改为通过输入查询与训练集查询编码之间的相似性来估计 LLM 的准确性和成本。然而,像它们的参数化对应项一样,非参数化路由器在处理离群查询时难以泛化,这一问题因训练集的多样性受限而加剧——扩充训练集昂贵且难以维持与不断演变的用例的同步。我们提出 ProxRouter,通过应用指数倾斜聚合机制,在非参数化路由器中平衡偏差和方差,从而提高其对离群查询的鲁棒性。实验表明,ProxRouter 在保持内在性能的同时,以最小开销提升了离群路由性能。

关键词

引用

@article{arxiv.2510.09852,
  title  = {ProxRouter: Proximity-Weighted LLM Query Routing for Improved Robustness to Outliers},
  author = {Shivam Patel and Neharika Jali and Ankur Mallick and Gauri Joshi},
  journal= {arXiv preprint arXiv:2510.09852},
  year   = {2025}
}