Unbiased Dynamic Pruning for Efficient Group-Based Policy Optimization
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 training speedup and outperforms GRPO by 3.36% in average accuracy across six mathematical reasoning benchmarks.
Figures & tables
| Setting | Method | GSM8K Dataset | Math Dataset | ||||||
| GPU-h | Speedup | Acc (%) | GPU-h | Speedup | Acc (%) | ||||
| Qwen3-4B | GRPO | 0% | 0% | 64.68 | 1.00 | 93.85 | 137.04 | 1.00 | 77.00 |
| DPPO | 50% | 50% | 39.28 | 1.65 | 93.49^{{\color[rgb]{1,0,0}-0.36}} | 88.32 | 1.55 | 78.10^{{\color[rgb]{0,0,1}+1.10}} | |
| DPPO | 0% | 50% | 49.47 | 1.31 | 94.01^{{\color[rgb]{0,0,1}+0.16}} | 110.80 | 1.24 | 80.02^{{\color[rgb]{0,0,1}+3.02}} | |
| DPPO | 50% | 0% | 45.74 | 1.41 | 94.24^{{\color[rgb]{0,0,1}+0.37}} | 102.48 | 1.34 | 77.65^{{\color[rgb]{0,0,1}+0.65}} | |
| [2pt/1pt] | DPPO | 30% | 30% | 45.94 | 1.41 | 93.86^{{\color[rgb]{0,0,1}+0.01}} | 101.12 | 1.36 | 77.75^{{\color[rgb]{0,0,1}+0.75}} |
| Model | Method | Speedup | MATH500 | AIME25 | AIME24 | AMC | Minerva. | Olympiad. | Avg. |
| Qwen3-4B | GRPO | 1.00 | 76.20 | 8.33 | 7.50 | 45.78 | 29.41 | 34.72 | 33.44 |
| CPPO | 1.35 | 75.15 | 6.67 | 5.83 | 41.87 | 28.49 | 31.49 | 31.58 | |
| GRESO | 2.31 | 76.40 | 10.83 | 10.83 | 47.29 | 30.15 | 37.28 | 35.10 | |
| DPPO | 2.37 | 77.90 | 11.67 | 17.50 | 46.99 | 29.96 | 38.02 | 36.80 | |
| Qwen3-8B | GRPO | 1.00 | 75.80 | 8.33 | 7.50 | 46.08 | 29.32 | 34.27 | 33.55 |
| CPPO | 1.32 | 75.40 | 6.67 | 5.83 | 41.57 | 29.23 | 31.53 | 31.71 |
| Setting | Method | GPU-h | Speedup | Acc (%) |
| Qwen3-4B | DAPO | 48.53 | 1.00 | 94.09 |
| + DPPO | 24.57 | 1.98 | 94.10 | |
| GSPO | 61.40 | 1.00 | 93.78 | |
| + DPPO | 25.07 | 2.45 | 94.17 | |
| Qwen3-8B | DAPO | 55.24 | 1.00 | 95.14 |
| + DPPO | 28.72 | 1.92 | 94.85 |
| Method | Rollout ( ) | GPU-h | Speedup | Acc (%) |
| GRPO | 8 | 89.76 | 1.00 | 94.84 |
| DPPO | 8 | 54.80 | 1.64 | 94.47 |
| GRPO | 16 | 173.36 | 1.00 | 94.69 |
| DPPO | 16 | 107.76 | 1.61 | 94.47 |
| Module | Pruning Rules | Acc (%) | ||
| Baseline (No Pruning) | – | – | – | 93.85 |
| Completion-level Ablation | ||||
| Ours (Prune Low) | 0 | 0.5 | 94.01 | |
| Reverse (Prune High) | 0 | 0.5 | 93.52 | |
| Prompt-level Ablation | ||||
| Ours (Prune Low) | 0.5 | 0 | 94.24 | |
| Method | Training Time (GPU hours) | Speedup | Acc (%) |
| Baseline | 78.93 | 1.00 | 94.85 |
| + Comp. Pruning | 55.10 | 1.34 | 94.77 |
| + Prompt Pruning | 48.79 | 1.52 | 94.94 |
| + Dense Packing | 47.00 | 1.68 | 95.15 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Method | GPU-h | Speedup | Acc (%) | ||
| Qwen3-4B | DAPO | 0% | 0% | 48.53 | 1.00 | 94.09 |
| + DPPO | 50% | 50% | 32.43 | 1.50 | 94.02^{{\color[rgb]{1,0,0}-0.07}} | |
| + DPPO | 0% | 50% | 42.90 | 1.13 | 94.24^{{\color[rgb]{0,0,1}+0.15}} | |
| + DPPO | 50% | 0% | 42.45 | 1.14 | 94.24^{{\color[rgb]{0,0,1}+0.15}} | |
| [2pt/1pt] | + DPPO | 30% | 30% | 37.27 | 1.30 | 94.70^{{\color[rgb]{0,0,1}+0.61}} |
| + DPPO | 70% | 70% | 31.74 | 1.53 | 94.70^{{\color[rgb]{0,0,1}+0.61}} |
| Setting | Method | GPU-h | Speedup | Acc (%) | ||
| Qwen3-4B | GSPO | 0% | 0% | 61.40 | 1.00 | 93.78 |
| + DPPO | 50% | 50% | 35.85 | 1.71 | 93.49^{{\color[rgb]{1,0,0}-0.29}} | |
| + DPPO | 0% | 50% | 47.09 | 1.30 | 94.17^{{\color[rgb]{0,0,1}+0.39}} | |
| + DPPO | 50% | 0% | 44.64 | 1.38 | 94.17^{{\color[rgb]{0,0,1}+0.39}} | |
| [2pt/1pt] | + DPPO | 30% | 30% | 42.85 | 1.43 | 94.17^{{\color[rgb]{0,0,1}+0.39}} |
| + DPPO | 70% | 70% | 27.84 | 2.21 | 93.94^{{\color[rgb]{0,0,1}+0.16}} |
| Setting | Method | GPU-h | Speedup | Acc (%) | ||
| Qwen2.5-7B-Instruct | GRPO | 0% | 0% | 58.39 | 1.00 | 71.84 |
| + DPPO | 50% | 50% | 37.31 | 1.56 | 71.97^{{\color[rgb]{0,0,1}+0.13}} | |
| + DPPO | 0% | 50% | 41.31 | 1.41 | 72.47^{{\color[rgb]{0,0,1}+0.63}} | |
| + DPPO | 50% | 0% | 42.84 | 1.36 | 71.76^{{\color[rgb]{1,0,0}-0.08}} | |
| [2pt/1pt] | + DPPO | 30% | 30% | 42.60 | 1.37 | 72.07^{{\color[rgb]{0,0,1}+0.23}} |
| + DPPO | 70% | 70% | 32.35 | 1.80 | 71.56^{{\color[rgb]{1,0,0}-0.28}} |
| Setting | Method | GPU-h | Speedup | Acc (%) | ||
| Qwen3-32B | GRPO | 0% | 0% | 204.08 | 1.00 | 80.43 |
| + DPPO | 50% | 50% | 127.92 | 1.60 | 82.55^{{\color[rgb]{0,0,1}+2.12}} | |
| + DPPO | 0% | 50% | 166.08 | 1.23 | 81.95^{{\color[rgb]{0,0,1}+1.52}} | |
| + DPPO | 50% | 0% | 159.84 | 1.28 | 80.80^{{\color[rgb]{0,0,1}+0.37}} | |
| [2pt/1pt] | + DPPO | 30% | 30% | 157.28 | 1.30 | 81.25^{{\color[rgb]{0,0,1}+0.82}} |
| + DPPO | 70% | 70% | 101.20 | 2.02 | 84.62^{{\color[rgb]{0,0,1}+4.19}} |
| Question: |
| Let be positive real numbers such that Find the maximum value of Let’s think step by step and output the final answer within box. |
| Ground Truth: 5 |
| GRPO Response ✗ : |
| <think> Okay, so I need to find the maximum value of the sum , given that the sum of the squares of all the ’s is . Hmm, let me think. First, the expression we need to maximize is the sum of all products of with each of the other variables. So that’s like multiplied by . Let me write that down: Let me denote . Then the sum becomes . So maybe I can use some inequality here. Since we have a constraint on the sum of squares, maybe Cauchy–Schwarz or AM–QM or something? Wait, Cauchy–Schwarz might be useful here. Let me recall that for vectors and , . But here, maybe I can think of and as part of some vectors? Alternatively, maybe I can use the Cauchy–Schwarz inequality on and the vector . Let me try that. The sum is the dot product of and . Then by Cauchy–Schwarz: So then the sum is . Now, to maximize this expression with respect to , let me set . To find its maximum, take derivative: Set derivative to zero: . Therefore, the maximum value is So the maximum sum is . Therefore, the answer is . Wait, but let me check if this is correct. Because when we apply Cauchy–Schwarz, we have that the maximum of is , given that . So yes, that seems right. Therefore, the maximum value is . Final Answer </think> |
| GRESO Response ✗ : |
| <think> Okay, so I need to find the maximum value of the sum given that the sum of the squares of all the variables is . Let me think about how to approach this. First, the problem is about maximizing a sum of products of variables, given a constraint on their squares. This seems like a problem where the Cauchy–Schwarz inequality or Lagrange multipliers might be useful. Since all variables are positive, maybe Cauchy–Schwarz is the way to go. Let me write down the expression I need to maximize. The sum is . Let me denote . Then the expression becomes . So I need to maximize , given that . Hmm. Maybe I can use the Cauchy–Schwarz inequality on and . Let me recall that for any vectors and , . If I take and , then their dot product is . The norm of is , and the norm of is . Therefore, . But then, the right-hand side is . So we consider the maximum of . Let . Taking derivative: . Setting to zero gives . Then maximum value is . Therefore, the maximum value is . But wait, is this tight? Because we have equality in Cauchy–Schwarz when vectors are proportional. So for equality, we need and to be proportional, which would mean that . Let me check. If all to are equal, denote them as . Then the sum of squares is . The expression to maximize is . From the earlier result, maximum of is ? Wait, no. Earlier we found that has maximum , but here . Therefore, . Therefore, the maximum of is . Therefore, the total sum is , so . Therefore, maximum value is . Therefore, the answer is . Final Answer </think> |
| CPPO Response ✗ : |
| <think> Okay, so I need to find the maximum value of the sum , given that the sum of the squares of all the ’s is . Let me think about how to approach this. First, the problem is about maximizing a sum of products involving with each of the other variables. The constraint is on the sum of squares. This seems like an optimization problem with a constraint, so maybe I can use Lagrange multipliers? Or perhaps there’s a way to apply the Cauchy–Schwarz inequality or some other inequality? Let me write down what I need to maximize: . So that’s multiplied by the sum of through . Let me denote that sum as . Then . So the problem becomes maximizing , given that . Hmm. Maybe I can express in terms of the other variables? Or maybe use Cauchy–Schwarz on ? Alternatively, think of as . Let me consider variables . Let me denote them as , so that . Then the constraint becomes . So we need to maximize with the constraint . This seems like a standard problem. Maybe use Cauchy–Schwarz on the terms and the sum of ? Wait, let me think. Let me consider vectors. Suppose I have two vectors: one is and another… Hmm, maybe not. Alternatively, think of the sum as the dot product of and . But I don’t know if that helps directly. Alternatively, use Cauchy–Schwarz on the terms and the sum of . Let me recall that for any vectors and , . If I take and , then their dot product is . The norm of is . The norm of is . Therefore, by Cauchy–Schwarz: . But the constraint is . Let me denote that sum of squares of as . Then . Therefore: . So now, the problem reduces to maximizing over . Let me set . To find its maximum, take derivative: . Set derivative equal to zero: . Therefore, maximum value is . Therefore, the answer is . Final Answer </think> |
| DPPO Response (Ours) ✓ : |
| <think> Okay, so I need to find the maximum value of the sum , given that the sum of the squares of to is . All the are positive real numbers. Hmm, let me think about how to approach this. First, the expression we need to maximize is the sum of all products of with each of the other variables. So that’s multiplied by . Let me denote . Then the expression becomes . Now, we know from the given condition that . So maybe I can relate and using some inequality. Since all variables are positive, perhaps Cauchy-Schwarz inequality could be useful here. By Cauchy-Schwarz, . There are terms in the sum for to , so that’s terms. Therefore, . Hence, . Therefore, . So our expression . Now, we need to maximize this expression with respect to . Let me set . To find its maximum, take derivative: . Set derivative equal to zero: . Since is positive. So maximum value is . Therefore, the maximum value is . So the answer is . Final Answer </think> |