GrowMTP: Can RL Grow Its Own Draft Head?
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
- 2025Qwen3 Technical Reportcited by 8,820
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
- 2025OpenAI GPT-5 System Cardcited by 989
- 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 843
- 2025Group Sequence Policy Optimizationcited by 835
27 more
- 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 578
- 2025GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Modelscited by 467
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 370
- 2023SpecInfer: Accelerating Generative Large Language Model Serving with Tree-based Speculative Inference and Verificationcited by 365
- 2024Better & Faster Large Language Models via Multi-token Predictioncited by 360
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 331
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decodingcited by 269
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 250
- 2023DistillSpec: Improving Speculative Decoding via Knowledge Distillationcited by 165
- 2023TACO: Topics in Algorithmic COde generation datasetcited by 160
- 2025MiMo: Unlocking the Reasoning Potential of Language Model -- From Pretraining to Posttrainingcited by 121
- 2024Hydra: Sequentially-Dependent Draft Heads for Medusa Decodingcited by 117
- 2023Online Speculative Decodingcited by 113
- 2024Sequoia: Scalable, Robust, and Hardware-aware Speculative Decodingcited by 103
- 2025The Art of Scaling Reinforcement Learning Compute for LLMscited by 101
- 2024Learning Harmonized Representations for Speculative Samplingcited by 72
- 2026Qwen3-Coder-Next Technical Reportcited by 71
- 2024Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Modelscited by 70
- 2024NeMo-Aligner: Scalable Toolkit for Efficient Model Alignmentcited by 41
- 2025History Rhymes: Accelerating LLM Reinforcement Learning with RhymeRLcited by 38
- 2026Nemotron 3 Super: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 38
- 2025ReSpec: Towards Optimizing Speculative Decoding in Reinforcement Learning Systemscited by 18
- 2026Draft-OPD: On-Policy Distillation for Speculative Draft Modelscited by 11
- 2025Bridging Draft Policy Misalignment: Group Tree Optimization for Speculative Decodingcited by 10
- 2026Breaking Entropy Bounds: Accelerating RL Training via MTP with Rejection Samplingcited by 9
- 2026Scaling Data Difficulty: Improving Coding Models via Reinforcement Learning on Fresh and Challenging Problemscited by 2
- 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
- 2025SPEC-RL: Accelerating On-Policy Reinforcement Learning with Speculative Rollouts
- 2026Spend Your Rollouts Where It Counts: Rollout Allocation for Group-Based RL Post-Training
- 2025RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Training
- 2025Beat the long tail: Distribution-Aware Speculative Decoding for RL Training
- 2026SpecRoll: Fast-Slow Verifier-Feedback Adaptation for Speculative Reinforcement Learning Rollouts
- 2025RLHFSpec: Breaking the Efficiency Bottleneck in RLHF Training via Adaptive Drafting
- 2026SortedRL: Accelerating RL Training for LLMs through Online Length-Aware Scheduling
- 2026Towards Better RL Training Data Utilization via Second-Order Rollout