arXiv:2608.22788cs.AIcs.LG2026-08

优化大模型生成的长尾任务调度,提升整体效率

TailSieve: Partial-Rollout-Guided Tail Routing for LLM Rollouts

论文配图:TailSieve: Partial-Rollout-Guided Tail Routing for LLM Rollouts
图 1 · 摘自论文原文
  • 通过部分生成结果识别长尾请求,动态分组调度
  • 相比均匀分配,最高提速1.67倍,长尾池支持专用解码加速2.59倍
  • 无需训练,适合大规模生成场景如强化学习后训练

大规模模型生成已成为现代大模型系统的核心,涵盖强化学习后训练、在线策略蒸馏和高采样量评估流水线。与注重请求级延迟和吞吐的在线服务不同,少数长尾生成会主导整个生成步骤的端到端完成时间。实践中,生成请求常被均匀分配至各副本,导致长生成进入高并发解码批次。为此,本文提出TailSieve,一种基于部分生成引导的框架,联合控制长尾路由与副本分配。在已知完成长度的理想情况下,我们证明最优调度需结合长尾隔离与负载均衡,且简单top-k策略可近似该离线最优。利用长尾提示在策略更新中保持长尾特性的观察,TailSieve使用部分生成作为无训练信号识别候选长尾组。层级控制器结合响应工作历史与测量的并发-吞吐模型,动态调整隔离组数及尾部与主体池间的副本分配。实验显示,其路由仅加速达1.67倍;进一步利用低并发尾部池,实现针对路由的推测解码(如MTP或DFlash),最高提速2.59倍。关键提示在当前策略下重新生成,维持在线生成特性,避免稳态下额外的路由引入长度偏差。

原文摘要 · Abstract (English)

Large-scale rollouts have become a core component of modern LLM systems, spanning reinforcement learning (RL) post-training, on-policy distillation (OPD), and sampling-heavy evaluation pipelines. Unlike online serving, which is typically optimized for request-level latency and throughput, a small number of long-tail generations can dominate the end-to-end makespan of an entire rollout step. In practice, rollout requests are often routed uniformly across replicas, which can place extremely long generations inside high-concurrency decoding batches. To address this, we present TailSieve, a partial-rollout-guided framework that jointly controls tail routing and replica allocation for LLM rollouts. In an idealized setting with known completion lengths, we show that makespan-optimal routing in the long-tail regime combines tail isolation with load balancing, and that a simple top-k policy closely approximates this offline optimum. Leveraging the observation that long-tail prompts tend to remain long-tailed across policy updates, TailSieve uses partial rollouts as a training-free signal for identifying candidate tail groups. A hierarchical controller then jointly adapts the number of isolated groups and the replica split between the tail and bulk pools using collected response-work history and a measured concurrency-throughput model. TailSieve achieves up to 1.67x routing-only speedup over uniform group routing. The resulting low-concurrency tail pool further enables route-specialized speculative decoding with MTP or DFlash, achieving up to 2.59x speedup over uniform routing. Selected prompts are regenerated under the current policy, preserving on-policy generation and avoiding additional routing-induced length bias in steady state.

大模型生成调度优化长尾处理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。