Recent alignment work on Large Language Models (LLMs) suggests preference optimization can improve reasoning by shifting probability mass toward better solutions. We test this claim in a resource-constrained setting by applying GRPO with LoRA to SLMs (up to 3B) for math reasoning on GSM8K and MATH datasets with difficulty-stratified analyses. As problem difficulty increases, accuracy plateaus, revealing a capacity boundary: GRPO primarily reshapes output preferences without reliably improving hardest-tier solving. Consistent with this, training GRPO only on lower-difficulty problems matches full-dataset accuracy across difficulty tiers while using only ~45% training steps, indicating diminishing returns from harder samples in this regime. We also find a cross-dataset generalization effect: GSM8K-trained GRPO achieves higher accuracy on the numeric subset of MATH than MATH-trained GRPO, exceeding it by ~5% at 1.5B and by ~3% at 3B. We show that the best achievable gains depend strongly on the base model's prior reasoning competence and the dataset's difficulty profile.
Figures & tables
Figure 1: Comparison of accuracy on the MATH dataset for Qwen2.5 0.5B, 1.5B, and 3B models Qwen et al. (2025) . Models trained on the lower-difficulty subsets (L1–L3) achieve comparable or superior performance to those trained on the full dataset (L1–L5).
Figure 2: Difficulty-stratified MATH accuracy for base and GRPO-tuned models. GRPO trained on full data (L1–L5) performs similarly to GRPO trained on L1–L3, suggesting limited benefit from including higher-difficulty problems (especially for smaller models).
Figure 3: Cross-dataset generalization under GRPO. Accuracy on the Numerical subset of MATH for 0.5B/1.5B/3B models trained with GRPO on (i) MATH L1–L3, (ii) MATH L1–L5, and (iii) GSM8K.
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
Dataset
Level 1
Level 2
Level 3
Level 4
Level 5
Total
Numerical
214
361
407
466
444
1,892
Full
242
476
585
661
716
2,680
Appendix
Table 1: Distribution of Hendrycks MATH test samples across difficulty levels.
Model Size
Per-Device Batch
Grad Accum
Effective Batch
0.5B
20
1
20
1.5B
16
1
16
3B
12
1
12
Appendix
Table 2: GRPO batch scaling for Hendrycks MATH. All runs use K=4 generations.
Model Size
Per-Device Batch
Grad Accum
Effective Batch
0.5B
24
2
48
1.5B
20
2
40
3B
16
2
32
Appendix
Table 3: GRPO batch scaling for GSM8K ( K=4 ).
Figure 4: Answer Extraction Failure Rate vs. Difficulty (MATH Full). Percentage of generations without a parsable <answer>...</answer> span. Failures increase with difficulty; the 1.5B model spikes at Level 5 (18.1%), while 3B remains more stable (13.3%).
Figure 5: Average Completion Tokens vs. Difficulty (MATH Full). Generation length increases with difficulty across model sizes (approaching ∼ 400 tokens at L5 for 1.5B/3B), indicating sustained reasoning effort even when answer extraction fails.
0.5B
1.5B
3B
Subject
Level
L1-L5
L1-L3
L1-L5
L1-L3
L1-L5
L1-L3
Algebra
L1
51.1
57.8
83.0
78.5
86.7
87.4
L2
38.8
41.8
61.7
60.2
72.6
70.6
L3
26.8
32.6
50.6
56.7
67.8
66.7
L4
18.0
20.1
44.9
45.9
51.6
63.6
L5
10.1
12.7
24.8
30.6
36.5
37.5
Appendix
Table 4: Impact of Filtering Hard Data. Accuracy (%) when trained on Full Data (MATH L1–L5) vs. Easy Data (MATH L1–L3).
0.5B
1.5B
3B
Subject
Level
GSM
L1-L5
L1-L3
GSM
L1-L5
L1-L3
GSM
L1-L5
L1-L3
Algebra
L1
61.6
53.6
56.0
91.2
81.6
80.8
93.6
88.8
84.0
L2
49.1
40.9
44.0
73.6
63.5
62.3
83.6
76.1
72.3
L3
37.8
29.5
33.2
67.9
51.3
57.0
75.6
68.9
66.8
L4
21.8
19.4
18.5
52.6
38.9
41.2
62.6
54.0
64.5
L5
12.1
6.1
11.6
27.3
21.7
22.2
34.3
29.3
31.3
Appendix
Table 5: Cross-Domain Transfer Analysis. Accuracy (%) comparing models trained on GSM8K, MATH (L1-L5), and MATH (L1-L3), evaluated by subject and difficulty.
Figure 6: Reward Mean Trajectories. Left/Center: On the MATH datasets (L1-L5 and L1-L3), the 3B model (Green) demonstrates steady monotonic improvement. In contrast, the 0.5B model (Orange) exhibits a flat or noisy trajectory, indicating an inability to consistently optimize the complex reasoning objective. Right: On the GSM8K dataset, all models—including the 0.5B converge reliably, confirming that the instability is specific to the difficulty of the task relative to model capacity.
Figure 7: Reward Standard Deviation: Right (GSM8K): Variance decreases significantly as training progresses (from ∼1.8 to ∼0.6 ), indicating the models are converging on high-confidence solutions. Left (MATH): Variance remains high and volatile for the 0.5B and 1.5B models, suggesting they are relying on stochastic exploration (”guessing”) rather than settling on stable reasoning paths.
Figure 8: Full Fine-Tuning (FFT) accuracy comparison for 0.5B model on the MATH dataset. Models trained on the easier subset (MATH L1–L3) match the performance of models trained on the full difficulty spectrum (MATH L1–L5), mirroring the LoRA-based capacity boundary.
Figure 9: Accuracy comparison of the 1.5B model across three training distributions: the MATH L1–L3 dataset, the MATH L1–L5 subset, and a size-matched random sample of the full dataset MATH L1–L5 (Random 45%).
Figure 10: Training dynamics for the size-matched Random 45% control experiment. (a) Mean Reward trajectory. (b) Reward Standard Deviation. The volatile reward and persistently high variance further illustrate the destabilizing effect of hard samples under a reduced step count.
Group Relative Policy Optimization (GRPO) has emerged as a memory-efficient reinforcement fine-tuning (RFT) technique for reasoning-intensive tasks. How- ever, GRPO training dynamics on small language models (SLMs) remain poorly understood, limiting its reliable adoption and reproducibility in open and resource- constrained environments. In this work, we present a systematic study of GRPO fine-tuning for SLMs ranging from 1.5B to 7B parameters under a practical single- node 8xA100 compute budget. Our study spans multiple model families and reasoning domains, including mathematics, coding, and multiple-choice question answering (MCQ) in science. Across these settings, we analyze how group size affects policy convergence, training stability, and downstream benchmark per- formance. We further characterize tensor-level update dynamics during GRPO training and investigate whether the choice of LoRA target modules and layers can improve the performance of GRPO-tuned models. While our initial GRPO-tuned models outperform their base counterparts on approximately 80% of mathematical benchmark evaluations, they demonstrate limited capability on MCQ and code reasoning tasks. Guided by our mechanistic evaluations, we refined our LoRA and reward-shaping configurations to improve performance in latter domains. These findings provide practical guidance for GRPO training for SLMs.
RL with verifiable rewards can substantially improve LLM reasoning, yet standard GRPO-style training often treats easy, hard, and learnable questions alike through uniform sampling and weighting, leading to inefficient compute allocation. We study GRPO by tracking token log-probabilities, group-normalized advantages, and the induced token-level update weights. This reveals three recurring dynamics as training proceeds: (1) confidence inflation, (2) advantage contraction, and (3) hierarchical convergence. These findings suggest that the utility of each update depends strongly on both question difficulty and the model's current competence. Motivated by this, we propose Confidence and Difficulty-adaptive Policy Optimization (CoDaPO), which assigns each question a bounded value from rollout confidence and empirical difficulty. CoDaPO then uses this value to reweight policy updates and resample high-value learnable questions within mini-batches, thereby increasing discovery within the learnable band under a fixed compute budget. Across twelve benchmarks, CoDaPO consistently improves accuracy over existing RL methods. Our code is publicly available at https://github.com/tmlr-group/CoDaPO.
Zhanke Zhou, Xiangyu Lu, Chentao Cao +4
1TMLR Group, Department of Computer Science, Hong Kong Baptist University · 2Stanford University · 3Sydney AI Centre, The University of Sydney
Reinforcement learning with verifiable rewards (RLVR) improves reasoning in large language models. Yet, typical RLVR approaches fail on difficult problems: when a model cannot generate any correct solutions, it receives zero learning signal. Providing privileged guidance during training, such as solution prefixes, can help overcome this learning cliff by steering the model towards {correct solutions with non-zero reward}. {We call these rollouts off-context: they are generated from a training prompt that contains privileged guidance, while the target objective is defined by the original prompt without that guidance.} {We introduce} Off-Context GRPO (OC-GRPO), a minimally modified variant of GRPO that uses guided rollouts but applies an importance-corrected objective to steer the update back toward the original unguided objective, avoiding the mismatch that destabilizes uncorrected guided training. Empirically, our algorithm achieves a 3.9% absolute improvement (13.8% relative gain) over vanilla GRPO on average across standard mathematical reasoning benchmarks with negligible additional cost.