cs.LGMar 4, 2026

Unbiased Dynamic Pruning for Efficient Group-Based Policy Optimization

Authors: Haodong Zhu, Yangyang Ren, Yanjing Li, Mingbao Lin, Linlin Yang, Xuhui Liu, Xiantong Zhen, Haiguang Liu, +1 more

Organizations: Beihang University · Zhongguancun Academy · Rakuten Singapore · Communication University of China · KAUST · United Imaging

Abstract

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×\times training speedup and outperforms GRPO by 3.36% in average accuracy across six mathematical reasoning benchmarks.

Figures & tables

Appendix figures & tables7 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Learn from the Gap: Differential-Aware Advantage Pruning with Adaptive Rollout Sampling for GRPO

    Sep 29, 2026Jiahua Yang, Zhiwei Yang, Xianpeng Zhang +7Group Relative Policy OptimizationRL for VLMs

  2. Smaller Models are Natural Explorers for Policy-Level Diversity in GRPO

    May 29, 2026Yiming Ren, Yiran Xu, Zicheng Lin +8Reinforcement LearningRL for Language Model Reasoning

  3. EP-GRPO: Entropy-Progress Aligned Group Relative Policy Optimization with Implicit Process Guidance

    May 6, 2026Song Yu, Li Li, Wenwen Zhao +1RL for Language Model ReasoningGroup Relative Policy Optimization