Unbiased Dynamic Pruning for Efficient Group-Based Policy Optimization
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Group Relative Policy Optimization (GRPO) effectively scales LLM reasoning but incurs prohibitive computational costs due to its extensive group-based sampling requirement. While recent selective data utilization methods can mitigate this overhead, they could induce estimation bias by altering the underlying sampling distribution, compromising theoretical rigor and convergence behavior. To address this limitation, we propose Dynamic Pruning Policy Optimization (DPPO), a framework that enables dynamic pruning while preserving unbiased gradient estimation through importance sampling-based correction. By incorporating mathematically derived rescaling factors, DPPO significantly accelerates GRPO training without altering the optimization objective of the full-batch baseline. Furthermore, to mitigate the data sparsity induced by pruning, we introduce Dense Prompt Packing, a window-based greedy strategy that maximizes valid token density and hardware utilization. Extensive experiments demonstrate that DPPO consistently accelerates training across diverse models and benchmarks. For instance, on Qwen3-4B trained on MATH, DPPO achieves 2.37× training speedup and outperforms GRPO by 3.36% in average accuracy across six mathematical reasoning benchmarks.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. First posted on arXiv 4 Mar 2026.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Demystifying Group Relative Policy Optimization: Its Policy Gradient is a U-Statistic
- 2026Calibration-Aware Policy Optimization for Reasoning LLMs
- 2025Slow-Fast Policy Optimization: Reposition-Before-Update for LLM Reasoning
- 2025Adaptive Group Policy Optimization: Towards Stable Training and Token-Efficient Reasoning
- 2026Understanding Evolution Strategies for LLM Reasoning: Broader Reasoning Coverage than GRPO
- 2026PrAg-PO: Prompt Augmented Policy Optimization for Robust and Diverse Mathematical Reasoning
- 2025Mitigating Think-Answer Mismatch in LLM Reasoning Through Noise-Aware Advantage Reweighting
- 2026Smaller Models are Natural Explorers for Policy-Level Diversity in GRPO