Sparrow: Sparse Rollout for Stable and Efficient Long-context RL of Large Language Models

28d ago · Global · primary source: export.arxiv.org

A new method called Sparrow uses dynamic sparse attention to accelerate the training of large language models under reinforcement learning with verifiable rewards, or RLVR, according to research posted to arXiv on June 7, 2026 [1][2]. The technique achieves rollout speedups of 2.2x, 2.4x, and 2.0x for three model sizes while maintaining training stability [1][2]. RLVR is a powerful training paradigm but induces extremely long chain-of-thought reasoning, making the process computationally expensive [1][2]. The dominant per-step cost comes from generating these long-context rollouts, and sparse attention has been proposed as a way to accelerate the dense rollout generation [2]. However, the authors note a fundamental tension: overly aggressive sparsity causes training collapse, while overly lenient sparsity provides insufficient speedup [1][2]. The Sparrow method addresses this by studying what the researchers call the sparse-to-dense actor-policy mismatch [2]. They observed that collapse is not driven by uniform degradation across all tokens; most sparse tokens align perfectly with their dense counterparts even under aggressive sparsity [2]. This led to the hypothesis that training remains stable if the lower tail of the per-token mismatch stays above a critical threshold throughout the trajectory [2]. The team introduced a dynamic sparsity schedule that keeps this tail statistic constant during generation [1][2]. They validated the approach across the Qwen3 thinking-family models, finding that maintaining the tail mismatch near a consistent threshold generally enabled stable training [2]. A cost model was then used to find the sparsity schedule that maximizes speedup under that mismatch threshold, yielding the reported gains for the 1.7-billion, 4-billion, and 8-billion parameter versions [2]. The researchers also showed empirically that the thresholds generalize to a larger 14-billion-parameter model and to a separate reinforcement-learning domain, coding [2]. The analysis further motivated a technique called DistillSparse, a lightweight LoRA-based distillation on sparse rollouts that allows more aggressive sparsity to reach the same mismatch threshold, yielding even higher speedup [2]. The paper was submitted to arXiv, an open-access repository of electronic preprints that, as of late 2024, receives about 24,000 articles per month and is not peer-reviewed [6].

regulationresearch-paperapplicationcommentary

Background sources we checked (7)
  • arxiv.org ↗ Despite being powerful, reinforcement learning with verifiable rewards (RLVR) induces extremely long COT, making it computationally expensive. Since RLVR per-step cost is dominated by long-context rollout generation, sparse attention offers a promising way to accelerate dense rol…
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
  • info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository [...] # arXivLabs: Showcase [...] arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. [...] While the arXiv team is focused on our core miss…
  • blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
  • en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
  • en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
  • en.wikipedia.org ↗ A large language model (LLM) is a neural network trained on a vast amount of text for natural language processing tasks, especially language generation. LLMs can typically generate, summarize, translate, and analyze text in many contexts, and are a foundational technology behind …

Sources

Spot something wrong? Report an issue