Beyond the Sampled Token: Preserving Candidate Support in RLVR
About
We revisit exploration collapse in reinforcement learning with verifiable rewards (RLVR), from the perspective of the \emph{candidate distribution} for next-token prediction. We formally show that as probability concentrates on the top-$1$ candidate, the expected number of distinct responses collapses to one regardless of the sampling budget $K$. This theoretical implication is further verified by our empirical tracking of top-$N$ candidate probabilities during training, where the top-$1$ candidate progressively dominates while plausible alternatives are suppressed. These findings suggest a key desideratum for effective exploration: \emph{preserving non-negligible probability mass on the top-$N$ candidates}. To this end, we propose Candidate-aware Support Preservation (CaSP), with two complementary designs. Specifically, CaSP redistributes positive gradients among top-$N$ candidates for correct responses, and applies a stronger penalty to the top-$1$ candidate for incorrect responses. Unlike many exploration-oriented methods that improve pass@$K$ at the cost of pass@1, CaSP improves pass@$K$ across the full $K$ spectrum. These gains generalize to 6 math, 2 logical-reasoning, and 2 coding benchmarks, and scales to 32B-parameter models and sampling budgets up to $K=1024$, positioning it as a principled, candidate-level approach for RLVR exploration.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Code Generation | HumanEval | Pass@156.2 | 171 | |
| Mathematical Reasoning | AIME25 (test) | Pass@127.5 | 45 | |
| Mathematical Reasoning | AIME 2024 | Pass@132.8 | 22 | |
| Mathematical Reasoning | AIME 2025 | Pass@117.8 | 22 | |
| Mathematical Reasoning | AMC 2023 | Pass@168.5 | 22 | |
| Mathematical Reasoning | MATH 500 | Pass@183.8 | 22 | |
| Mathematical Reasoning | Minerva Math | Pass@145.2 | 22 | |
| Mathematical Reasoning | OlympiadBench | Pass@147.1 | 22 | |
| Mathematical Reasoning | OmniMath (test) | -- | 20 | |
| Code Generation | LiveCodeBench | Rate @32 Score46.5 | 17 |