Vocabulary Dropout for Curriculum Diversity in LLM Co-Evolution
Organizations: Arizona State University
Abstract
Co-evolutionary self-play, where one language model generates problems and another solves them, promises curriculum learning without human supervision. The promise breaks down early in practice. The proposer converges to a narrow distribution of problems that satisfy the reward function, and the collapsed curriculum teaches the solver little, stalling the loop. We introduce vocabulary dropout, a lightweight intervention that randomly masks the proposer's output logits during both policy training and curriculum generation. The mask is hard and non-stationary, so the proposer cannot lock into fixed token sequences. Training Qwen3-4B and Qwen3-8B on mathematical reasoning via R-Zero, vocabulary dropout sustains proposer diversity throughout training across lexical, semantic, and functional measures, and improves the solver by an average of +4.4 points at 8B with the largest gains on competition-level benchmarks. Explicit action-space constraints, filling the structural role that game rules fill in classical self-play, can keep co-evolution in language productive. Vocabulary dropout is one simple way to impose them.
Figures & tables
| Pass@1 (%) | |||||||
| Setting | MATH500 | GSM8K | AMC | Olympiad | AIME’24 | AIME’25 | Avg. |
| Qwen3-4B | |||||||
| Base (no training) | 54.1 2.2 | 63.5 5.2 | 29.2 1.8 | 20.6 1.3 | 12.2 0.9 | 1.1 0.9 | 30.1 |
| KL-Cov | 60.8 1.9 | 82.6 2.0 | 42.0 3.8 | 23.0 0.8 | 9.3 2.6 | 4.0 1.1 | 37.0 |
| Clip-Cov | 65.8 1.1 | 85.4 1.0 | 39.5 3.4 | 24.5 0.2 | 12.7 1.7 | 4.0 1.7 | 38.7 |
| Baseline ( ) | 64.2 0.8 | 85.2 0.7 | 41.7 1.4 | 23.2 0.3 | 8.9 0.9 | 6.7 1.6 | 38.3 |
| Trained on | MATH500 | GSM8K | AMC | Olympiad | AIME’24 | AIME’25 | Minerva | Avg. |
|---|---|---|---|---|---|---|---|---|
| Base (no training) | 55.0 | 75.1 | 25.0 | 20.0 | 6.7 | 3.3 | 25.4 | 30.1 |
| Baseline | 71.0 | 88.6 | 40.0 | 28.2 | 3.3 | 6.7 | 29.8 | 38.2 |
| KL-Cov | 69.8 | 88.6 | 45.0 | 27.6 | 10.0 | 3.3 | 31.6 | 39.4 |
| Clip-Cov | 66.6 | 86.7 | 40.0 | 24.2 | 10.0 | 3.3 | 32.0 | 37.5 |
| VD | 71.0 | 89.8 | 50.0 | 27.8 | 3.3 | 10.0 | 37.1 | 41.3 |
| Scale | Method | Filter pass | Pass rate | Well-posed | Correct | Vendi (WP) | Vendi (C) |
|---|---|---|---|---|---|---|---|
| 4B | Baseline | 946 | 11.8% | 850 | 275 | — | — |
| VD75 | 1,222 | 15.3% | 925 | 269 | |||
| 8B | Baseline | 510 | 6.4% | 465 | 163 | — | — |
| VD75 | 1,408 | 17.6% | 1,112 | 325 |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Hyperparameter | Value |
|---|---|---|
| KL-Cov | (top-covariance token fraction) | |
| (KL coefficient on selected tokens) | 1.0 | |
| Clip-Cov | (masked-token fraction) | |
| (covariance bounds) | 1.0, 5.0 | |
| Both | Entropy bonus coefficient | 0 |
| Scope | Proposer only |
| Model | Setting | It. 1 | It. 2 | It. 3 | It. 4 | It. 5 |
|---|---|---|---|---|---|---|
| Qwen3-8B | Baseline | 81.4 | 45.5 | 42.2 | 39.1 | 38.0 |
| Train-only | 79.0 | 43.2 | 42.0 | 40.4 | 39.0 | |
| Gen-only | 62.0 | 38.3 | 33.7 | 33.4 | 31.9 | |
| VD75 | 61.8 | 34.9 | 33.1 | 36.7 | 32.4 | |
| Qwen3-4B | Baseline | 75.6 | 38.1 | 38.0 | 42.2 | 37.7 |
| Train-only | 76.0 | 37.2 | 44.2 | 38.0 | 38.2 |
| Pass@1 (%) | |||||||
|---|---|---|---|---|---|---|---|
| Setting | MATH500 | GSM8K | AMC | Olympiad | AIME’24 | AIME’25 | Avg. |
| Qwen3-4B | |||||||
| Base (no training) | 54.1 2.2 | 63.5 5.2 | 29.2 1.8 | 20.6 1.3 | 12.2 0.9 | 1.1 0.9 | 30.1 |
| Baseline ( ) | 56.0 1.1 | 72.7 3.4 | 33.3 3.0 | 20.3 0.9 | 10.0 1.6 | 4.4 0.9 | 32.8 |
| VD85 ( ) | 58.4 0.8 | 60.5 4.1 | 37.5 4.1 | 19.3 1.1 | 11.1 0.9 | 4.4 2.4 | 31.9 |
| Qwen3-8B | |||||||
| Semantic | Logical | ||||
|---|---|---|---|---|---|
| Model | Setting | Value | Value | ||
| Qwen3-4B | Baseline | — | — | ||
| VD | {\color[rgb]{0,0.4688,0.3125}\textbf{0.692}}\pm{<}0.001 | {\color[rgb]{0,0.4688,0.3125}\textbf{+0.405}}\pm 0.014 | |||
| Qwen3-8B | Baseline | — | — | ||
| VD | {\color[rgb]{0,0.4688,0.3125}\textbf{0.677}}\pm{<}0.001 | {\color[rgb]{0,0.4688,0.3125}\textbf{+0.376}}\pm 0.016 | |||
| Cumulative Vendi Score | |||||||
|---|---|---|---|---|---|---|---|
| Model | Setting | It. 1 | 1–2 | 1–3 | 1–4 | 1–5 | Growth |
| Qwen3-4B | Baseline | 44.6 | 46.1 | 48.9 | 49.5 | 52.1 | +7.5 |
| Train-only | 46.1 | 51.0 | 52.0 | 53.7 | 55.7 | +9.6 | |
| Gen-only | 59.7 | 62.9 | 63.9 | 63.9 | 64.3 | +4.6 | |
| Both | 61.5 | 63.9 | 68.6 | 69.7 | 70.2 | +8.7 | |
| Qwen3-8B | Baseline | 40.0 | 41.5 | 42.1 | 43.2 | 42.2 | +2.2 |
| Pass@1 (%) | ||||||||
|---|---|---|---|---|---|---|---|---|
| Proposer | Setting | MATH500 | GSM8K | AMC | Olympiad | AIME’24 | AIME’25 | Avg. |
| 4B | Baseline | 64.2 0.8 | 85.2 0.7 | 41.7 1.4 | 23.2 0.3 | 8.9 0.9 | 6.7 1.6 | 38.3 |
| 4B | + VD85 | 65.7 0.9 | 81.8 0.4 | 50.0 2.4 | 24.7 0.9 | 8.9 1.8 | 4.4 0.9 | 39.3 |
| 8B | Baseline | 63.5 2.7 | 82.7 2.5 | 44.2 2.4 | 24.4 0.1 | 11.1 1.6 | 6.7 5.4 | 38.8 |
| 8B | + VD85 (gen-only) | 66.1 1.1 | 83.4 2.7 | 35.8 4.3 | 23.4 0.7 | 13.3 2.7 | 2.2 1.6 | 37.4 |
| Model | MATH500 | AMC |
|---|---|---|
| Base (no training) | 47.7 | 25.3 |
| Baseline solver | 47.3 | 26.0 |
| VD75 solver | 47.4 | 25.2 |
| Qwen3-4B | Qwen3-8B | |||
|---|---|---|---|---|
| Setting | Unique tok. | # Q | Unique tok. | # Q |
| Baseline | 2,717 | 6,771 | 1,881 | 7,044 |
| VD75 | 3,697 | 3,377 | 2,868 | 4,864 |
| +36% | +52% | |||
| Setting | MATH500 | GSM8K | AMC | Olympiad | AIME’24 | AIME’25 | Minerva |
|---|---|---|---|---|---|---|---|
| Baseline ( ) | 70.6 | 86.3 | 47.5 | 23.6 | 6.7 | 3.3 | 29.8 |
| VD ( ) | 63.0 | 87.9 | 45.0 | 24.1 | 10.0 | 0.0 | 34.9 |
| Benchmark | # Examples | Domain |
| GSM8K ( Cobbe et al., 2021 ) | 1,319 | Grade-school math |
| MATH500 ( Hendrycks et al., 2021 ) | 500 | Competition math |
| AMC ( Mathematical Association of America, 2023 ) | 40 | Competition math |
| OlympiadBench ( He et al., 2024 ) | 910 | Olympiad math |
| AIME 2024 ( Mathematical Association of America, 2024 ) | 30 | Olympiad math |
| AIME 2025 ( Mathematical Association of America, 2024 ) | 30 | Olympiad math |
| Parameter | Proposer | Solver |
|---|---|---|
| Base model | Qwen3-4B / Qwen3-8B | |
| Optimizer | AdamW | |
| Learning rate | ||
| Weight decay | ||
| LR warmup | 0 | |
| Max grad norm | 1.0 | |
| Parameter | Value |
|---|---|
| Embedding model | text-embedding-3-small |
| Embedding dimension | 1536 |
| Normalization | L2 (cosine kernel) |
| Max samples (eigendecomp.) | 5,000 |
| Subsampling seed | 42 |
| Eigenvalue threshold |
| Parameter | Value |
|---|---|
| Observer model | (same as base) |
| LoRA rank ( ) | 16 |
| LoRA alpha | 32 |
| LoRA dropout | 0.05 |
| Target modules | q/k/v/o_proj, gate/up/down_proj |
| Max sequence length | 512 |
Curriculum RL
Curriculum reinforcement learning organizes tasks, examples, or training conditions to help agents learn complex behaviors efficiently.
See all 121 papers