Sparse-RL: Breaking the Memory Wall in LLM Reinforcement Learning via Stable Sparse Rollouts
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Reinforcement Learning (RL) has become essential for eliciting complex reasoning capabilities in Large Language Models (LLMs). However, the substantial memory overhead of storing Key-Value (KV) caches during long-horizon rollouts acts as a critical bottleneck, often prohibiting efficient training on limited hardware. While existing KV compression techniques offer a remedy for inference, directly applying them to RL training induces a severe policy mismatch, leading to catastrophic performance collapse. To address this, we introduce Sparse-RL empowers stable RL training under sparse rollouts. We show that instability arises from a fundamental policy mismatch among the dense old policy, the sparse sampler policy, and the learner policy. To mitigate this issue, Sparse-RL incorporates Sparsity-Aware Rejection Sampling and Importance-based Reweighting to correct the off-policy bias introduced by compression-induced information loss. Experimental results show that Sparse-RL reduces rollout overhead compared to dense baselines while preserving the performance. Furthermore, Sparse-RL inherently implements sparsity-aware training, significantly enhancing model robustness during sparse inference deployment. The corresponding training data and code are publicly available on the repository.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 22 of its 37 references that are in the map. First posted on arXiv 15 Jan 2026.
Builds on
22 of 37 references.
Closest ancestors · 4
Same direction, newest first
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning30 May
- 2025SimpleRL-Zoo: Investigating and Taming Zero Reinforcement Learning for Open Base Models in the Wild24 Mar
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 18
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 880
- 2024KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cachecited by 637
- 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 591
- 2023Model Tells You What to Discard: Adaptive KV Cache Compression for LLMscited by 474
- 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Timecited by 439
- 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inferencecited by 235
- 2023Evaluating the Performance of Large Language Models on GAOKAO Benchmarkcited by 186
6 more
- 2024ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inferencecited by 134
- 2025AgentRL: Scaling Agentic Reinforcement Learning with a Multi-Turn, Multi-Task Frameworkcited by 62
- 2025When to Continue Thinking: Adaptive Thinking Mode Switching for Efficient Reasoningcited by 26
- 2025Beat the long tail: Distribution-Aware Speculative Decoding for RL Trainingcited by 10
- 2025Accelerating Large-Scale Reasoning Model Inference with Sparse Self-Speculative Decodingcited by 10
- 2025RLHFSpec: Breaking the Efficiency Bottleneck in RLHF Training via Adaptive Draftingcited by 5
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026From Completion to Editing: Unlocking Context-Aware Code Infilling via Search-and-Replace Instruction Tuningcited by 6
- 2026ETS: Energy-Guided Test-Time Scaling for Training-Free RL Alignmentcited by 5
- 2026Sparrow: Sparse Rollout for Stable and Efficient Long-context RL of Large Language Modelscited by 5
- 2026Your "Pro" LLM Subscription May Actually Be "Free": Exposing Fingerprint Spoofing Risks in LLM Inference Servicescited by 3
- 2026How to Compress KV Cache in RL Post-Training? Shadow Mask Distillation for Memory-Efficient Alignmentcited by 2
- 2026Turning Off-Policy Tokens On-Policy: A Plug-in Approach for Improving LLM Alignmentcited by 2
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directionscited by 0
- 2026Beyond Masked Sparsity: SNACK Enables Truly Sparse Neural Networks on GPUcited 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
- 2026Beyond Masked Sparsity: SNACK Enables Truly Sparse Neural Networks on GPU2 Oct 2026
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directions21 Sep 2026
- 2026Turning Off-Policy Tokens On-Policy: A Plug-in Approach for Improving LLM Alignment6 Jul 2026
- 2026Your "Pro" LLM Subscription May Actually Be "Free": Exposing Fingerprint Spoofing Risks in LLM Inference Services15 Jun 2026
- 2026Sparrow: Sparse Rollout for Stable and Efficient Long-context RL of Large Language Models7 Jun 2026
- 2026How to Compress KV Cache in RL Post-Training? Shadow Mask Distillation for Memory-Efficient Alignment7 May 2026
- 2026ETS: Energy-Guided Test-Time Scaling for Training-Free RL Alignment29 Jan 2026
- 2026From Completion to Editing: Unlocking Context-Aware Code Infilling via Search-and-Replace Instruction Tuning19 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2026Rethinking RL for LLM Reasoning: It's Sparse Policy Selection, Not Capability Learning
- 2025RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Training
- 2025RL-PLUS: Countering Capability Boundary Collapse of LLMs in Reinforcement Learning with Hybrid-policy Optimization
- 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs
- 2025SPEC-RL: Accelerating On-Policy Reinforcement Learning with Speculative Rollouts
- 2026A Step Back: Prefix Importance Ratio Stabilizes Policy Optimization
- 2025Accelerating RL for LLM Reasoning with Optimal Advantage Regression
- 2026Addressing Performance Saturation for LLM RL via Precise Entropy Curve Control