On-Policy Self-Training: Data-Free Self-Improvement in Large Language Models
Organizations: Google · Ulm University
Abstract
On-policy self-distillation enables post-training foundation models without relying on stronger external teachers. However, existing approaches rely on ground-truth question-answer pairs, human-curated problem sets, or other supervised signals. In this work, we reformulate on-policy self-distillation as a self-training framework, driven by self-generated question-answer pairs and privileged reasoning demonstrations. We push supervision to the limit: guided solely by coarse, high-level topic descriptions, the model synthesizes its own curriculum of queries and privileged reasoning traces without access to any human-curated prompts or answers. We find that self-distillation on fully synthetic privileged information enables a model to improve across multiple domains while retaining its capabilities on domains it is not self-trained on. We demonstrate the generality of this framework across text and multimodal domains, achieving consistent self-improvement on standard language benchmarks and Visual Question Answering. Our findings demonstrate that foundation models possess rich latent capabilities that can be amplified without external task supervision or curated problem distributions.
Figures & tables
| Method | Signal | Gen. Answer | MMLU | GPQA | GSM8K ∗ | HellaSwag | Average | |
|---|---|---|---|---|---|---|---|---|
| Gemma 4 E2B | ||||||||
| Base | 37.5 | 28.8 | – | 48.7 | 38.3 | |||
| + SFT | Dense | Target | 38.3 +0.8 | 30.8 +2.0 | – | 50.8 +2.1 | 40.0 +1.7 | |
| + GRPO | Sparse | Reward | 37.3 -0.2 | 30.3 +1.5 | – | 50.3 +1.6 | 39.3 +1.0 | |
| w/o Think. | + OPST (Ours) | Dense | Context | 47.7 +10.2 | 32.8 +4.0 | – | 56.2 +7.5 | 45.6 +7.3 |
| Base | 69.9 | 27.8 | 72.7 | 63.1 | 58.4 | |||
| Human-Curated | ||||||||
| Method | Question | Answer | MMLU | GPQA | GSM8K ∗ | HellaSwag | Average | |
| Gemma 4 E2B | ||||||||
| Base | 37.5 | 28.8 | – | 48.7 | 38.3 | |||
| + Supervised SFT | 47.3 +9.8 | 31.3 +2.5 | – | 52.8 +4.1 | 43.8 +5.5 | |||
| + Supervised OPSD | 49.6 +12.1 | 33.3 +4.5 | – | 56.8 +8.1 | 46.6 +8.3 | |||
| + u-OPSD ‡ | 45.9 +8.4 | 33.3 +4.5 | – | 54.7 +6.0 | 44.6 +6.3 | |||
| In-domain | Out-of-domain | |||||||||
| Training Topics | MMLU | HellaSwag | GSM8K ∗ | Avg. | GPQA | ARC-C | MATH-500 ∗ | Winogrande | Avg. | |
| Gemma 4 E2B | ||||||||||
| Base | 37.5 | 48.7 | 72.7 | 53.0 | 28.8 | 39.1 | 56.5 | 50.0 | 43.6 | |
| Reference | + Single domain | 47.7 +10.2 | 56.2 +7.5 | 80.9 +8.2 | 61.6 +8.6 | |||||
| + MMLU + HellaSwag | 46.9 +9.4 | 55.6 +6.9 | 74.3 +1.6 | 58.9 +5.9 | 30.3 +1.5 | 44.9 +5.8 | 48.0 -8.5 | 51.8 +1.8 | 43.8 +0.2 | |
| + HellaSwag + GSM8K | 42.6 +5.1 | 54.6 +5.9 | 77.0 +4.3 | 58.1 +5.1 | 31.3 +2.5 | 43.0 +3.9 | 55.8 -0.7 | 51.8 +1.8 | 45.5 +1.9 | |
| Privileged Info | VQAv2 | POPE | |||||
| Method | Caption | Spatial | Acc. ( ) | F1 ( ) | Acc. ( ) | Rec. ( ) | |
| Gemma 4 E2B | |||||||
| Base | 49.1 | 82.2 | 83.2 | 72.9 | |||
| + Visual OPST | 55.7 +6.6 | 84.7 +2.5 | 85.1 +1.9 | 77.4 +4.5 | |||
| + Visual OPST | 54.0 +4.9 | 83.0 +0.8 | 83.9 +0.7 | 74.0 +1.1 | |||
| Ours | + Visual OPST | 56.0 +6.9 | 84.6 +2.4 | 85.0 +1.8 | 77.4 +4.5 | ||
| Sample selection | MMLU |
|---|---|
| Base | 37.5 |
| + OPST w/o verification | 42.7 |
| + Verification | 45.4 |
| + Consensus gate | 47.7 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| rate | MMLU |
|---|---|
| 1.00 (no EMA) | 38.1 |
| 0.20 | 44.2 |
| 0.05 | 47.7 |
| 0.01 | 45.2 |
| Setting | MMLU | GPQA | GSM8K | HellaSwag | Average |
|---|---|---|---|---|---|
| w/o Think. | 47.7 0.8 | 32.8 1.0 | – | 56.2 0.8 | 45.6 0.3 |
| w/ Think. | 71.1 0.1 | 34.3 1.0 | 80.9 1.2 | 65.6 0.5 | 63.0 0.4 |
| Method | MMLU | GPQA | GSM8K | HellaSwag | Average |
|---|---|---|---|---|---|
| Qwen3-4B-Base | |||||
| Base | 72.0 | 32.3 | 79.7 | 82.8 | 66.7 |
| + OPST (Ours) | 75.1 +3.1 | 37.4 +5.1 | 93.0 +13.3 | 85.4 +2.5 | 72.7 +6.0 |
Language Model Self-Improvement
Language model self-improvement studies how models use their own outputs, feedback, and experience to improve with less external supervision.
See all 172 papers