QaRL: Rollout-Aligned Quantization-Aware RL for Fast and Stable Training under Training--Inference Mismatch
Not on the map. Why?
Abstract (arXiv)
Large language model (LLM) reinforcement learning (RL) pipelines are often bottlenecked by rollout generation, making end-to-end training slow. Recent work mitigates this by running rollouts with quantization to accelerate decoding, which is the most expensive stage of the RL loop. However, these setups destabilize optimization by amplifying the training-inference gap: rollouts are operated at low precision, while learning updates are computed at full precision. To address this challenge, we propose QaRL (Rollout Alignment Quantization-Aware RL), which aligns training-side forward with the quantized rollout to minimize mismatch. We further identify a failure mode in quantized rollouts: long-form responses tend to produce repetitive, garbled tokens (error tokens). To mitigate these problems, we introduce TBPO (Trust-Band Policy Optimization), a sequence-level objective with dual clipping for negative samples, aimed at keeping updates within the trust region. On Qwen3-30B-A3B MoE for math problems, QaRL outperforms quantized-rollout training by +5.5 while improving stability and preserving low-bit throughput benefits.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 21 of its 31 references that are in the map. First posted on arXiv 9 Apr 2026.
Builds on
21 of 31 references.
References · 21
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2023Let's Verify Step by Stepcited by 3,789
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2015Trust Region Policy Optimizationcited by 2,580
- 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
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,848
- 2025s1: Simple test-time scalingcited by 1,341
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
9 more
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2025Learning to Reason under Off-Policy Guidancecited by 281
- 2019Mastering Complex Control in MOBA Games with Deep Reinforcement Learningcited by 161
- 2025Defeating the Training-Inference Mismatch via FP16cited by 66
- 2021Batch size-invariance for policy optimizationcited by 35
- 2024COAT: Compressing Optimizer states and Activation for Memory-Efficient FP8 Trainingcited by 33
- 2025When Importance Sampling Misallocates Credit: Asymmetric Ratios for Outcome-Supervised RLcited by 21
- 2025Deterministic Inference across Tensor Parallel Sizes That Eliminates Training-Inference Mismatchcited by 14
- 2025A Comprehensive FP8 Training Recipe for Reasoning-Enhanced Language Modelscited by 3
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026IndexMem: Learned KV-Cache Eviction with Latent Memory for Long-Context LLM Inferencecited by 7
- 2026QUADS: Stabilizing NVFP4 Reinforcement Learning for MoE via QUantization-error Alignment across Dual Sidescited by 3
- 2026Where Rollouts Begin: Low-Load, High-Leverage First-Token Diversification for RLVRcited by 2
- 2026QATFactory: A Versatile, Deployment-Aligned Framework for Quantization-aware Training and Distillation of LLMscited by 1
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directionscited by 0
- 2026TRIAGE: Direction-Aware Mismatch Stabilization of Native NVFP4 Reinforcement Learningcited by 0
- 2026S3R: Fast and Faithful RL Rollouts via Self-Sparse Speculative Decodingcited by 0
- 2026S3R: Fast and Faithful RL Rollouts via Self-Sparse Speculative Decoding9 Oct 2026
- 2026TRIAGE: Direction-Aware Mismatch Stabilization of Native NVFP4 Reinforcement Learning5 Oct 2026
- 2026QATFactory: A Versatile, Deployment-Aligned Framework for Quantization-aware Training and Distillation of LLMs30 Sep 2026
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directions21 Sep 2026
- 2026QUADS: Stabilizing NVFP4 Reinforcement Learning for MoE via QUantization-error Alignment across Dual Sides17 Jul 2026
- 2026Where Rollouts Begin: Low-Load, High-Leverage First-Token Diversification for RLVR27 May 2026
- 2026IndexMem: Learned KV-Cache Eviction with Latent Memory for Long-Context LLM Inference25 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026TRACE: Rollout-Guided Quantization-Aware Training for FP4 Reinforcement Learning of MoE Language Models
- 2026QuRL: Efficient Reinforcement Learning with Quantized Rollout
- 2025QeRL: Beyond Efficiency -- Quantization-enhanced Reinforcement Learning for LLMs
- 2026FP8-RL: A Practical and Stable Low-Precision Stack for LLM Reinforcement Learning
- 2026Jet-RL: Enabling On-Policy FP8 Reinforcement Learning with Unified Training and Rollout Precision Flow
- 2025RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Training
- 2026AIS: Adaptive Importance Sampling for Quantized RL
- 2026HiFloat4 Format for End-To-End Reinforcement Learning Post-Training of Large Language Models