EfficientRollout: System-Aware Self-Speculative Decoding for RL Rollouts
面向强化学习Rollout的系统感知自推测解码
🧭 LLM训练加速 · 自推测解码
FuriosaAI · University of California, Berkeley
系统感知自推测解码框架,自适应切换推测解码与草稿长度,将RL rollout延迟降低19.6%且不损失模型质量。
📌 工作参考价值(4 点)
- 在RL微调LLM(如GRPO/PPO)时,rollout生成是瓶颈,可直接采用自推测解码(self-speculative decoding):从目标模型量化出轻量draft模型,无需额外训练,保持与进化策略分布对齐,降低工程维护成本。
- 系统感知的SD切换策略(toggle policy)根据批次大小和计算状态动态开启/关闭推测解码:只在内存密集、并行验证能利用闲置算力时启用,避免计算密集阶段引入额外开销。此动态调度思想可迁移到生产环境的在线推理加速中。
- 接受率感知的草稿长度自适应(acceptance-aware draft-length adaptation):根据近期接受率动态调节每次推测的token数量,匹配draft质量变化,避免无效生成。类似机制可用于自适应调整推理时的speculative长度,降低延迟抖动。
- 整体框架强调系统级的协同设计,而不只是算法优化。做推荐/Agent类LLM应用时,若涉及线上更新或频繁微调,可考虑将系统负载感知纳入推测解码的调度策略,提升资源利用率。