Token-Budget-Aware Pool Routing for Cost-Efficient LLM Inference
该论文指出生产环境中 vLLM 集群按最大上下文长度配置每个实例,导致 80-95% 的短请求浪费了 4-8 倍并发能力,同时引发 KV-cache OOM 崩溃和请求拒绝。作者提出 token-budget-aware pool routing 方案:通过自校准的分类器估计每个请求的总 token 预算,将请求路由到配置匹配的实例池,从根本上解决配置-流量不匹配问题。
INSIGHT该方案直接解决了 LLM 搜索服务部署的核心痛点:不同查询的长度差异极大,统一配置造成严重资源浪费。对搜索系统的启示是:应该根据 query 复杂度和预期回答长度进行请求分级路由,而非所有请求共享相同的 GPU 资源配置。这种思路可直接应用于搜索场景中的 LLM reranker 或 LLM-based query 理解服务。