mlsanityData to 14 Oct 2026

GrowMTP: Can RL Grow Its Own Draft Head?

arXiv 2609.16648 ↗First posted 15 Sep 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Reinforcement learning (RL) post-training drives the frontier capabilities of large language models, with its wall-clock dominated by autoregressive rollout generation. Speculative decoding is an established remedy for this bottleneck, but existing draft heads must be pretrained or warmed up before RL, introducing substantial training cost outside the RL run to be accelerated. We observe that RL training itself provides both conditions required for online draft-head training: its rollout distribution is far narrower than that of pretraining, and its verification step continuously produces supervision signals aligned with this distribution. Building on these observations, we propose GrowMTP, which uses this supervision to train a draft head from scratch entirely within the RL loop, with all head updates detached from the policy backbone. On Qwen3-4B (no draft head), MiMo-7B-SFT (weak head), and Qwen3.5-4B-Base (strong head), GrowMTP achieves rollout speedups of 2.13x, 1.93x, and 1.36x, and end-to-end speedups of 1.60x, 1.41x, and 1.20x, respectively. GrowMTP therefore serves existing RL training frameworks as a modular component, particularly offering a from-scratch acceleration path for models without pretrained draft heads.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 39 of its 45 references that are in the map. First posted on arXiv 15 Sep 2026.

Builds on

39 of 45 references.

References · 39

  1. 2025Qwen3 Technical Reportcited by 8,820
  2. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
  3. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
  4. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
  5. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
  6. 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
  7. 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
  8. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
  9. 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
  10. 2025OpenAI GPT-5 System Cardcited by 989
  11. 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 843
  12. 2025Group Sequence Policy Optimizationcited by 835
27 more
  1. 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 578
  2. 2025GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Modelscited by 467
  3. 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 370
  4. 2023SpecInfer: Accelerating Generative Large Language Model Serving with Tree-based Speculative Inference and Verificationcited by 365
  5. 2024Better & Faster Large Language Models via Multi-token Predictioncited by 360
  6. 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 331
  7. 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decodingcited by 269
  8. 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 250
  9. 2023DistillSpec: Improving Speculative Decoding via Knowledge Distillationcited by 165
  10. 2023TACO: Topics in Algorithmic COde generation datasetcited by 160
  11. 2025MiMo: Unlocking the Reasoning Potential of Language Model -- From Pretraining to Posttrainingcited by 121
  12. 2024Hydra: Sequentially-Dependent Draft Heads for Medusa Decodingcited by 117
  13. 2023Online Speculative Decodingcited by 113
  14. 2024Sequoia: Scalable, Robust, and Hardware-aware Speculative Decodingcited by 103
  15. 2025The Art of Scaling Reinforcement Learning Compute for LLMscited by 101
  16. 2024Learning Harmonized Representations for Speculative Samplingcited by 72
  17. 2026Qwen3-Coder-Next Technical Reportcited by 71
  18. 2024Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Modelscited by 70
  19. 2024NeMo-Aligner: Scalable Toolkit for Efficient Model Alignmentcited by 41
  20. 2025History Rhymes: Accelerating LLM Reinforcement Learning with RhymeRLcited by 38
  21. 2026Nemotron 3 Super: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 38
  22. 2025ReSpec: Towards Optimizing Speculative Decoding in Reinforcement Learning Systemscited by 18
  23. 2026Draft-OPD: On-Policy Distillation for Speculative Draft Modelscited by 11
  24. 2025Bridging Draft Policy Misalignment: Group Tree Optimization for Speculative Decodingcited by 10
  25. 2026Breaking Entropy Bounds: Accelerating RL Training via MTP with Rejection Samplingcited by 9
  26. 2026Scaling Data Difficulty: Improving Coding Models via Reinforcement Learning on Fresh and Challenging Problemscited by 2
  27. 2026Joint Training of Multi-Token Prediction in Reinforcement Learning via Optimal Coefficient Calibrationcited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025SPEC-RL: Accelerating On-Policy Reinforcement Learning with Speculative Rollouts
  2. 2026Spend Your Rollouts Where It Counts: Rollout Allocation for Group-Based RL Post-Training
  3. 2025RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Training
  4. 2025Beat the long tail: Distribution-Aware Speculative Decoding for RL Training
  5. 2026SpecRoll: Fast-Slow Verifier-Feedback Adaptation for Speculative Reinforcement Learning Rollouts
  6. 2025RLHFSpec: Breaking the Efficiency Bottleneck in RLHF Training via Adaptive Drafting
  7. 2026SortedRL: Accelerating RL Training for LLMs through Online Length-Aware Scheduling
  8. 2026Towards Better RL Training Data Utilization via Second-Order Rollout