解决多类型强化学习推演共用推理服务时的缓存竞争问题
Scheduling Mixed RL Rollouts Beyond Prefix Locality
- 基于自适应准入与驻留时间感知的动态调度策略
- 在多个模型上提升推演吞吐量超40%,保持高缓存命中率
- 适合大规模语言模型后训练中混合任务部署场景
大型语言模型的现代强化学习后训练流水线正越来越多地融合跨领域、多反馈范式的推演负载。前缀感知路由虽能通过缓存复用和负载均衡提升推理效率,但无法控制异构推演会话对KV缓存容量的竞争。当具备可验证奖励的强化学习(RLVR)、人类反馈强化学习(RLHF)与智能体式推演共享异步推理服务时,其不同的序列结构、交互模式及缓存驻留时间导致显著差异的服务需求。推演调度必须考虑这种异构性,同时不扭曲训练器指定的工作负载混合比例。本文提出MISA-T,一种用于混合推演服务的路由层准入策略,结合自适应会话准入、工作负载感知的KV容量分配与驻留时间感知的KV计账机制。在Step3.7与Qwen3.6-35B-A3B上的纯推演消融实验中,MISA-T相比调优后的缓存感知vLLM Router分别提升推演吞吐量53.3%和43.6%,同时保持高前缀缓存命中率。在匹配的50轮Step3.7实验中,吞吐量提升35.6%,平均迭代时间减少22.8%,且消耗的工作负载混合比例接近训练目标,任务得分相当。
原文摘要 · Abstract (English)
Modern reinforcement learning (RL) post-training pipelines for large language models (LLMs) increasingly combine rollout workloads across multiple domains and feedback paradigms. Prefix-aware routing improves inference efficiency through cache reuse and load balancing, but it does not control how heterogeneous rollout sessions compete for KV-cache capacity. When reinforcement learning with verifiable rewards (RLVR), reinforcement learning from human feedback (RLHF), and agentic rollouts share an asynchronous inference service, their distinct sequence structures, interaction patterns, and KV-residency times create substantially different serving demands. Rollout scheduling must account for this heterogeneity without distorting the workload mixture specified by the trainer. We present MISA-T, a routing-layer admission policy for mixed rollout serving. MISA-T combines adaptive session admission, workload-aware KV-capacity allocation, and residency-time-aware KV accounting. In rollout-only ablations on Step3.7 and Qwen3.6-35B-A3B, MISA-T improves rollout throughput over a sweep-tuned cache-aware vLLM Router by 53.3% and 43.6%, respectively, while maintaining high prefix-cache hit rates. In a matched 50-iteration Step3.7 experiment, it increases rollout throughput by 35.6% and reduces mean iteration time by 22.8%, while keeping the consumed workload mixture close to the trainer target and achieving comparable task scores.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。