The Pareto Paradox of One-Shot RLVR: Marginal Analysis of Data Scale from 1 to 1,200 Examples
This post analyzes the paper "Reinforcement Learning for Reasoning in Large Language Models with One Training Example" (Wang et al., arXiv:2504.20571, NeurIPS 2025; code: https://github.com/ypwang61/One-Shot-RLVR).
Key points
- Data scale is a pseudo-variable in RLVR: 1-shot training reaches parity with 1.2K-shot training. On MATH500 (Qwen2.5-Math-1.5B): baseline 36.0% → 1-shot 73.6% → 2-shot 74.8% → 1,200-shot 73.6%. Average across 6 benchmarks: 17.6% → 35.7% (1-shot) vs 35.9% (1.2K-shot).
- The data-scaling curve resembles a step function rather than the logarithmic curve seen in supervised learning, suggesting RLVR "unlocks" existing capability rather than teaching new knowledge.
- Format correction vs. real gains: a format-correction deduction protocol shows ~40-50% of the gain is format learning; the non-format reasoning gain is +8.6% for 1-shot and ~8.4% for 1.2K-shot — essentially identical. Both converge to the same policy basin.
- Pure entropy loss alone lifts MATH500 by +27.4% (36.0% → 63.4%) with no answer rewards, and 1-shot + entropy reaches 73.6%. Exploration, not reward data, is the key catalyst.
- Post-saturation generalization: training accuracy saturates (~100%) within ~100 steps, yet MATH500 test accuracy keeps improving until ~1,400 steps. Afterwards, outputs on the training problem degenerate into multilingual gibberish (though answers remain correct), while test outputs stay coherent.
- Cross-domain generalization: training on a single geometry problem also improves algebra and number-theory accuracy — ruling out pure memorization, though transfer-of-reasoning vs. shared-format explanations remain undistinguished.
- The effect stems from policy gradient loss, not grokking, and holds across models (Qwen2.5-Math-1.5B/7B, Llama3.2-3B-Instruct, DeepSeek-R1-Distill-Qwen-1.5B) and algorithms (GRPO, PPO).
- Data selection: shift from coverage/diversity metrics to signal strength — policy gradient variance and alignment between problem difficulty and model capability.
- Stopping criteria: training-accuracy saturation is a poor stopping signal; test accuracy may keep improving for ~14x more steps.
- Recommended practice: prioritize exploration mechanisms (higher entropy coefficient, β ≥ 0.01 suggested; randomized perturbations; temperature annealing) over data scale; adopt a phased data strategy (1-10 problems → validate; 10-100 → cross-validate; scale only if necessary).
- Only mathematical reasoning is tested; code, science, and commonsense domains are unverified.
- Post-saturation generalization lacks a theoretical explanation; optimal training length cannot be predicted.
- Cannot fully separate "genuine reasoning" from "sophisticated format matching."
- Results are sensitive to which single problem is chosen; reproducibility is constrained.
- Effects on 32B+ models and industrial-scale setups are untested.
Method and experimental design
The paper systematically compares 1-shot / 2-shot / 1.2K-shot RLVR, using one or two math problems from a DeepScaleR subset, evaluated on MATH500 plus five other math reasoning benchmarks. A format-correction protocol compares scores under strict vs. lenient formatting requirements: if gains persist under lenient formats, the improvement is genuine reasoning rather than oxed{}-style formatting.
Interpretation
Most of the 1,200 training problems likely lie in the model's "already mastered" zone (no signal) or "unreachable" zone (no positive reward). Only a few problems sit in the zone of proximal development, providing effective policy gradients — a single well-chosen problem suffices.
System-level implications
Limitations
Paper details
| Item | Content | |:--|:--| | Title | Reinforcement Learning for Reasoning in Large Language Models with One Training Example | | Authors | Yiping Wang, Qing Yang, Zhiyuan Zeng, Liliang Ren, Lucas Liu, Baolin Peng, Hao Cheng, Xuehai He, Kuan Wang, Jianfeng Gao, Weizhu Chen, Shuohang Wang, Simon Shaolei Du, Yelong Shen | | Institutions | University of Washington, USC, Microsoft, UC Santa Cruz, Georgia Tech | | arXiv | 2504.20571 (April 29, 2025) | | Venue | NeurIPS 2025 | | Code | https://github.com/ypwang61/One-Shot-RLVR |