让多个客户端联合训练语言模型路由,提升效率与隐私保护。
Federate the Router: Learning Language Model Routers with Sparse and Decentralized Evaluations
- 通过联邦学习整合各客户端本地评估数据,共享路由策略。
- 在两个基准上,联合训练比单客户端提升准确率与成本平衡。
- 适合需隐私保护的边缘或企业级大模型服务场景。
大型语言模型(LLMs)正被边缘和企业客户端远程调用,因模型能力与价格差异大,合理路由查询以平衡质量与推理成本至关重要。现有路由方法依赖集中化的查询-模型评估数据,但此类数据常分散在终端用户和组织中,且涉及隐私,难以集中。同时,客户端独立训练效果差,因本地数据量少、覆盖查询分布有限且模型评估有偏差。本文提出首个面向LLM路由的联邦框架,使客户端能基于本地离线评估数据联合学习共享路由策略。该框架支持参数化多层感知机路由器与非参数K-means路由器,适应异构客户端查询分布与非均匀模型覆盖。在两个基准测试中,联邦协作显著改善了准确率-成本权衡,既提升了有效模型覆盖率,也增强了查询泛化能力。理论分析还证明,联邦训练可降低路由次优性。
原文摘要 · Abstract (English)
Large language models (LLMs) are increasingly accessed as remotely hosted services by edge and enterprise clients that cannot run frontier models locally. Since models vary widely in capability and price, routing queries to models that balance quality and inference cost is essential. Existing router approaches assume access to centralized query-model evaluation data. However, these data are often fragmented across clients, such as end users and organizations, and are privacy-sensitive, which makes centralizing data infeasible. Additionally, per-client router training is ineffective since local evaluation data is limited and covers only a restricted query distribution and a biased subset of model evaluations. We introduce the first federated framework for LLM routing, enabling clients to learn a shared routing policy from local offline query-model evaluation data. Our framework supports both parametric multilayer perceptron router and nonparametric K-means router under heterogeneous client query distributions and non-uniform model coverage. Across two benchmarks, federated collaboration improves the accuracy-cost frontier over client-local routers, both via increased effective model coverage and better query generalization. Our theoretical results also validate that federated training reduces routing suboptimality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。