让大模型推理加速器实时学习,实现秒级部署与持续优化。
When RL Meets Adaptive Speculative Training: A Unified Training-Serving System
- 将推理加速器训练与服务闭环融合,通过在线反馈持续优化。
- 上线首日即提速1.5倍,动态适应用户流量变化再提1.25倍。
- 支持无中断热更新,适合需快速响应的生产级大模型服务场景。
推测解码可显著加速大模型服务,但现有部署普遍将推测器训练与服务分离,导致高延迟、反馈滞后和领域漂移问题。我们提出Aurora,一个统一的训练-服务系统,通过实时推理日志持续学习推测器。将在线学习建模为异步强化学习:被接受的标记提供正反馈,被拒绝的预测则隐含负反馈,提升样本效率。系统集成SGLang推理服务与异步训练模块,支持无中断热更新。关键优势在于支持‘当天部署’:推测器可立即投入使用并快速适配真实流量,同时即时获取性能反馈。实验表明,Aurora在近期前沿模型(如MiniMax M2.1 229B、Qwen3-Coder-Next 80B)上实现1.5倍的初始加速;在应对用户流量分布变化时,相较静态预训练推测器额外获得1.25倍加速,适用于广泛使用的Qwen3与Llama3模型。
原文摘要 · Abstract (English)
Speculative decoding can significantly accelerate LLM serving, yet most deployments today disentangle speculator training from serving, treating speculator training as a standalone offline modeling problem. We show that this decoupled formulation introduces substantial deployment and adaptation lag: (1) high time-to-serve, since a speculator must be trained offline for a considerable period before deployment; (2) delayed utility feedback, since the true end-to-end decoding speedup is only known after training and cannot be inferred reliably from acceptance rate alone due to model-architecture and system-level overheads; and (3) domain-drift degradation, as the target model is repurposed to new domains and the speculator becomes stale and less effective. To address these issues, we present Aurora, a unified training-serving system that closes the loop by continuously learning a speculator directly from live inference traces. Aurora reframes online speculator learning as an asynchronous reinforcement-learning problem: accepted tokens provide positive feedback, while rejected speculator proposals provide implicit negative feedback that we exploit to improve sample efficiency. Our design integrates an SGLang-based inference server with an asynchronous training server, enabling hot-swapped speculator updates without service interruption. Crucially, Aurora supports day-0 deployment: a speculator can be served immediately and rapidly adapted to live traffic, improving system performance while providing immediate utility feedback. Across experiments, Aurora achieves a 1.5x day-0 speedup on recently released frontier models (e.g., MiniMax M2.1 229B and Qwen3-Coder-Next 80B). Aurora also adapts effectively to distribution shifts in user traffic, delivering an additional 1.25x speedup over a well-trained but static speculator on widely used models (e.g., Qwen3 and Llama3).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。