Flat-RLVR: Sharpness-Aware Reinforcement Learning with Verifiable Rewards for Reasoning Capacity in LLMs
Organizations: Graduate School of Artificial Intelligence, Ulsan National Institute of Science and Technology, Ulsan, Republic of Korea · Department of Electrical Engineering, Ulsan National Institute of Science and Technology (UNIST), Ulsan, Republic of Korea.
Abstract
Reinforcement learning with verifiable rewards (RLVR), particularly through group-relative policy optimization (GRPO), has emerged as a central paradigm for the post-training of large language models (LLMs). While RLVR is highly effective at enhancing the reasoning capabilities of LLMs, it commonly induces distribution sharpening, whereby the policy distribution becomes excessively peaked and rare-yet-correct trajectories are progressively suppressed, thus diminishing sample diversity. To this end, we promote flat-reward optima in RLVR to smooth the policy distribution. We theoretically link flat reward RLVR to reduced distribution sharpening, and empirically show improved sample diversity while maintaining single-attempt accuracy. Our method, called Flat-RLVR, searches for flat reward regions via sharpness-aware minimization during LLM post-training, combining this with GRPO variants to achieve better trade-offs between sampling efficiency and reasoning capacity.
Figures & tables
| Base | RL-Tuned | GRPO | Flat -GRPO |
|---|---|---|---|
| Total | |||
| Model | Method | Avg. | MATH500 | AMC23 | AIME24 | AIME25 | AIME26 |
|---|---|---|---|---|---|---|---|
| Qwen2.5 | GRPO | 15.6 /64.8 | 34.6 / 72.7 | 30.2 /95.8 | 7.0 /62.2 | 2.9 /48.9 | 3.7 /44.4 |
| Flat -GRPO | 15.2 / 67.3 | 34.0 /72.5 | 28.9 / 96.7 | 6.5 / 64.4 | 3.3 / 51.1 | 3.8 / 52.2 | |
| DAPO | 16.3 /65.2 | 35.7 /72.1 | 32.0 / 98.3 | 6.9 /63.3 | 3.1 /51.1 | 3.9 /41.1 | |
| Flat -DAPO | 16.3 / 68.6 | 35.7 / 72.4 | 32.1 /97.5 | 6.9 66.7 | 3.1 / 57.8 | 3.9 / 48.9 | |
| CISPO | 15.7 /65.5 | 34.7 / 72.8 | 30.6 /98.3 | 6.9 /63.3 | 2.9 /50.0 | 3.9 /43.3 | |
| Flat -CISPO | 15.6 / 68.5 | 33.8 /72.6 | 30.7 / 99.2 | 6.5 / 67.8 | 3.1 / 55.6 | 4.2 / 47.8 |
| Benchmarks | GRPO | F-GRPO | Flat -GRPO |
|---|---|---|---|
| MATH500 | 34.6 / 72.7 | 33.2 / 72.1 | 34.0 / 72.5 |
| AMC23 | 30.2 / 95.8 | 28.4 / 98.3 | 28.9 / 96.7 |
| AIME24 | 7.0 / 62.2 | 6.4 / 66.7 | 6.5 / 64.4 |
| AIME25 | 2.9 / 48.9 | 2.6 / 47.8 | 3.3 / 51.1 |
| AIME26 | 3.7 / 44.4 | 3.5 / 41.1 | 3.8 / 52.2 |
| Avg. | 15.6 / 64.8 | 14.8 / 65.2 | 15.2 / 67.3 |
| Method | Avg. | MATH500 | AMC23 | AIME24 | AIME25 | AIME26 |
|---|---|---|---|---|---|---|
| BASE | 16.7 /60.8 | 44.5 /70.5 | 31.1 /98.3 | 4.5 /47.8 | 1.6 /50.0 | 2.0 /37.8 |
| GRPO | 18.5 /59.4 | 46.9 / 70.3 | 36.9 /95.8 | 5.2 /42.2 | 1.6 /46.7 | 2.3 /42.2 |
| Flat -GRPO | 18.2 / 61.4 | 46.7 /69.5 | 35.9 / 97.5 | 4.7 / 48.9 | 1.6 / 47.8 | 2.2 / 43.3 |
| Method | MBPP |
|---|---|
| BASE | 42.3 / 91.5 |
| GRPO | 56.7 / 85.6 |
| Flat -GRPO | 54.5 / 89.6 |
| Method | LPF |
|---|---|
| GRPO | 0.4434 |
| Flat -GRPO | 0.0598 |
| Method | Total (h) | allocated( ) |
|---|---|---|
| Flat -GRPO ( ) | ||
| Flat -GRPO ( ) | ||
| GRPO |
| per-step wall clock time (s) | |
|---|---|
| ordinary step | 90.17 |
| SAM step | 90.56 (+0.43 ) |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Avg. | MATH500 | AMC23 | AIME24 | AIME25 | AIME26 |
|---|---|---|---|---|---|---|
| GRPO | 15.2/63.1 | 34.6/73.0 | 28.8/92.5 | 6.4/63.3 | 2.8/43.3 | 3.3/43.3 |
| GRPO + SAM | 9.8 /64.6 | 22.5/73.0 | 18.8/100 | 4.2/63.3 | 1.6/46.7 | 2.1/40.0 |
| GRPO + SAM + Anne. | 13.8 /65.5 | 30.7/72.4 | 26.7/95.0 | 5.8/60.0 | 2.4/53.3 | 3.3/46.7 |
| GRPO + SAM + GC | 11.6 /64.7 | 25.8/72.8 | 22.5/97.5 | 4.4/56.7 | 2.3/50.0 | 3.1/46.7 |
| GRPO + SAM + F | 15.5 /65.8 | 35.1/71.6 | 28.8/97.5 | 7.2/66.7 | 2.6/46.7 | 3.9/46.7 |
| Flat -GRPO | 15.0 /67.5 | 33.9/72.4 | 27.7/95.0 | 7.1/66.7 | 2.8/50.0 | 3.4/53.3 |
| Backbone | Base | RL | MATH500 | AMC23 | AIME24 | AIME25 | AIME26 | Avg. |
|---|---|---|---|---|---|---|---|---|
| Qwen2.5-Math-1.5B (Base vs. GRPO) | ||||||||
| Qwen2.5 | ||||||||
| Qwen2.5-Math-1.5B (Base vs. Flat -GRPO) | ||||||||
| Method | Parameter | Value |
|---|---|---|
| SAM- | period | |
| Cosine Annealing | ||
| Focal Weight | ||
| Parameter | Qwen2.5-Math-1.5B | Llama-3.2-1B-Instruct | Qwen2.5-0.5B-Instruct | Qwen2.5-3B-Instruct | Qwen2.5-Coder-1.5B-Instruct |
|---|---|---|---|---|---|
| Optimizer | AdamW | ||||
| Weight decay | |||||
| Learning rate | |||||
| LR scheduler | Linear | ||||
| Num. generations ( ) | |||||
RL for Language Model Reasoning
Reinforcement learning for language model reasoning uses reward-driven training to improve multi-step problem solving, tool use, and reliable answers.
See all 1,103 papers