Jackpot: Optimal Budgeted Rejection Sampling for Extreme Actor-Policy Mismatch Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Reinforcement learning (RL) for large language models (LLMs) remains expensive, particularly because the rollout is expensive. Decoupling rollout generation from policy optimization (e.g., leveraging a more efficient model to rollout) could enable substantial efficiency gains, yet doing so introduces a severe distribution mismatch that destabilizes learning. We propose Jackpot, a framework that leverages Optimal Budget Rejection Sampling (OBRS) to directly reduce the discrepancy between the rollout model and the evolving policy. Jackpot integrates a principled OBRS procedure, a unified training objective that jointly updates the policy and rollout models, and an efficient system implementation enabled by top-k probability estimation and batch-level bias correction. Our theoretical analysis shows that OBRS consistently moves the rollout distribution closer to the target distribution under a controllable acceptance budget. Empirically, sys substantially improves training stability compared to importance-sampling baselines, achieving performance comparable to on-policy RL when training Qwen3-8B-Base for up to 300 update steps of batchsize 64. Taken together, our results show that OBRS-based alignment brings us a step closer to practical and effective decoupling of rollout generation from policy optimization for RL for LLMs.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 25 of its 34 references that are in the map. First posted on arXiv 5 Feb 2026.
Builds on
25 of 34 references.
References · 25
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,193
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2023AgentBench: Evaluating LLMs as Agentscited by 1,211
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
- 2018IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architecturescited by 866
13 more
- 2025Group Sequence Policy Optimizationcited by 788
- 2020Deep Reinforcement Learning for Autonomous Driving: A Surveycited by 698
- 2023Llemma: An Open Language Model For Mathematicscited by 385
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 234
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 184
- 2025Act Only When It Pays: Efficient Reinforcement Learning for LLM Reasoning via Selective Rolloutscited by 108
- 2025Every Step Evolves: Scaling Reinforcement Learning for Trillion-Scale Thinking Modelcited by 50
- 2025Prosperity before Collapse: How Far Can Off-Policy RL Reach with Stale Data on LLMs?cited by 49
- 2025LlamaRL: A Distributed Asynchronous Reinforcement Learning Framework for Efficient Large-scale LLM Trainingcited by 37
- 2025Seer: Online Context Learning for Fast Synchronous LLM Reinforcement Learningcited by 21
- 2025Kinetics: Rethinking Test-Time Scaling Lawscited by 11
- 2023Optimal Budgeted Rejection Sampling for Generative Modelscited by 5
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026EfficientRollout: System-Aware Self-Speculative Decoding for RL Rolloutscited by 5
- 2026Where Rollouts Begin: Low-Load, High-Leverage First-Token Diversification for RLVRcited by 2
- 2026Smaller Models are Natural Explorers for Policy-Level Diversity in GRPOcited by 1
- 2026Multimodal Diffusion to Mutually Enhance Polarized Light and Low Resolution EBSD Datacited by 0
- 2026EfficientRollout: System-Aware Self-Speculative Decoding for RL Rollouts17 Jun 2026
- 2026Smaller Models are Natural Explorers for Policy-Level Diversity in GRPO29 May 2026
- 2026Where Rollouts Begin: Low-Load, High-Leverage First-Token Diversification for RLVR27 May 2026
- 2026Multimodal Diffusion to Mutually Enhance Polarized Light and Low Resolution EBSD Data24 Apr 2026