Search papers, labs, and topics across Lattice.
Dynamic Pruning Policy Optimization (DPPO) is introduced to accelerate Group Relative Policy Optimization (GRPO) for LLM reasoning by dynamically pruning less important groups while maintaining unbiased gradient estimation via importance sampling. To combat data sparsity from pruning, Dense Prompt Packing is used to maximize valid token density. Experiments on Qwen3-4B trained on MATH show DPPO achieves 2.37x speedup and 3.36% higher accuracy than GRPO across mathematical reasoning benchmarks.
Achieve over 2x training speedup for LLM reasoning without sacrificing accuracy by dynamically pruning Group Relative Policy Optimization (GRPO) with a novel importance sampling correction.
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.
