回到最新动态
研究论文 / THE SIGNAL

NgramRL:用回放感知的 N-gram 推测解码加速强化学习后训练

NgramRL: Accelerating RL Post-Training for Structured Generation via Rollout-Aware N-gram Speculative Decoding

重要性 5.5 AI 辅助整理

NgramRL 用近期 rollout 历史做 n-gram 草稿,加速 RL 后训练生成。

资讯摘要

论文提出 NgramRL,针对强化学习后训练中自回归 rollout 生成耗时占比高的问题,用同一提示近期 rollout 历史构建逐提示 n-gram 草稿器,无需辅助模型、草稿器训练或常驻 GPU 状态,并由目标模型验证保持分布。作者还引入自适应 fan-out 预取器预测验证结果。论文报告在两个模型家族、三类推理与结构化生成任务上,热缓存下每次目标模型前向提交 3.60–7.77 个 token。

要点速读

  1. NgramRL 从有界近期 rollout 历史构建逐提示 n-gram 草稿器,无需辅助模型、草稿器训练或常驻 GPU 草稿状态,目标模型验证保持 rollout 分布。
  2. 论文报告热缓存下每次目标模型前向提交 3.60–7.77 个 token,比 SuffixDecoding 最多高 1.99 倍,比 prompt-lookup decoding 高 7.0 倍。
  3. 在 Spider 上,预取器在候选预算 16 时覆盖 Qwen3-8B 的 96.6% 结果,实测 rollout 吞吐比同步 NgramRL 提升 1.38 倍,接近 1.40 倍重叠上限。

中文摘要由 AI 根据公开来源生成,可能存在理解偏差;具体能力、数字、开放时间等信息请以原文为准。
采集于 2026/10/27 08:00:28(北京时间)。 arXiv 日期为本次公告时间,论文初次投稿日期请查看原文版本记录。

阅读 arXiv · cs.AI 原文arxiv.org · 在新窗口打开