paperbot · PL 论文追踪

全部论文 待读 收藏

P2 Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter

驯服长尾:基于自适应草稿模型的高效推理强化学习训练
ASPLOS Proceedings of the 31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 22026匹配度 2/10 · 论文聚焦于通过自适应推测解码加速LLM推理强化学习训练,属于机器学习系统与LLM训练效率优化,与用户关注的程序分析、形式化验证、浮点/贝叶斯程序分析以及LLM与静态分析交叉方向均无直接关联。引用 16 inbox
Qinghao Hu, Shang Yang, Junxian Guo, Xiaozhe Yao, Yujun Lin, Yuxian Gu, Han Cai, Chuang Gan, Ana Klimovic, Song Han
crossref: 2026-09-12dblp: 2026-09-12arxiv: 2026-09-12

AI 速览

一句话:提出 TLT 系统,通过自适应推测解码无损加速推理强化学习训练,缓解长尾生成瓶颈。

大语言模型通过强化学习训练推理能力时,响应生成呈现持续的长尾分布:少数超长响应主导执行时间,浪费资源并推高成本。为此,论文提出 TLT 系统,将自适应推测解码引入 RL 训练以无损加速。其挑战在于动态工作负载、不断演化的目标模型以及 draft 模型训练开销。TLT 通过两个协同组件解决:自适应 Drafter 在长尾生成期间利用空闲 GPU 持续训练轻量 draft 模型,使其与目标模型保持对齐且不增加额外成本;自适应 Rollout 引擎维护内存高效的预捕获 CUDAGraph 池,并为每个输入批次自适应选择合适的推测解码策略。评估表明,TLT 相比最先进系统实现超过 1.7 倍的端到端 RL 训练加速,保持模型精度,并免费产出适合高效部署的高质量 draft 模型。

问题
推理模型的强化学习训练中,响应生成呈长尾分布,少数超长响应主导执行时间,造成资源浪费和成本膨胀。
方法
TLT 集成自适应推测解码:轻量 draft 模型在长尾生成期间利用空闲 GPU 持续训练以对齐目标模型,自适应 Rollout 引擎维护 CUDAGraph 池并为每批输入选择合适策略。
评估
与最先进系统相比,端到端 RL 训练加速超过 1.7 倍,同时保持模型精度,并额外获得高质量 draft 模型。
原文摘要(Abstract)

The emergence of Large Language Models (LLMs) with strong reasoning capabilities marks a significant milestone, unlocking new frontiers in complex problem-solving. However, training these reasoning models, typically using Reinforcement Learning (RL), encounters critical efficiency bottlenecks: response generation during RL training exhibits a persistent long-tail distribution, where a few very long responses dominate execution time, wasting resources and inflating costs. To address this, we propose TLT, a system that accelerates reasoning RL training losslessly by integrating adaptive speculative decoding. Applying speculative decoding in RL is challenging due to the dynamic workloads, evolving target model, and draft model training overhead. TLT overcomes these obstacles with two synergistic components: (1) Adaptive Drafter, a lightweight draft model trained continuously on idle GPUs during long-tail generation to maintain alignment with the target model at no extra cost; and (2) Adaptive Rollout Engine, which maintains a memory-efficient pool of pre-captured CUDAGraphs and adaptively select suitable SD strategies for each input batch. Evaluations demonstrate that TLT achieves over 1.7x end-to-end RL training speedup over state-of-the-art systems, preserves the model accuracy, and yields a high-quality draft model as a free byproduct suitable for efficient deployment. Code is released at https://github.com/mit-han-lab/fastrl.

链接与引用

DOI 原文 · arXiv · PDF(开放获取) · DBLP

BibTeX
@inproceedings{abs-2511-16665,
  title = {Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter},
  author = {Qinghao Hu and Shang Yang and Junxian Guo and Xiaozhe Yao and Yujun Lin and Yuxian Gu and Han Cai and Chuang Gan and Ana Klimovic and Song Han},
  booktitle = {ASPLOS},
  year = {2026},
  doi = {10.1145/3779212.3790231}
}