When Does Exploration Prevent Collapse? Three Regimes and the Path of the Reward in Self-Consistency Training
Organizations: Human-Centered Artificial Intelligence Research Institute, Ewha Womans University · Department of Computer Science and Engineering, Ewha Womans University
Abstract
Test-time reinforcement learning (TTRL) adapts a model without labels by rewarding answers that agree with the majority, but it collapses the policy onto its majority answer and can remove the correct answer from the samples. Exploration can prevent this collapse, yet its strength is chosen by trial and error, and how the training algorithm changes the required strength is unclear. We model training on a single problem as a continuous flow of the logits of a softmax policy and analyze which policies are stable as the exploration coefficient varies. Only collapsed policies are stable below a lower boundary and only policies spread over many answers above an upper boundary, and we derive these boundaries in closed form for an agreement reward with smoothing toward the uniform policy, compute them for the majority reward of TTRL on policies with one leading answer and bound them with interval arithmetic for actions on ordered bins. We prove that under exact gradients, the learning rate and, in a mean-field model, the normalization of GRPO leave these boundaries unchanged when exploration passes through them together with the reward but shift them when exploration bypasses them. Exact sums on these policies then show that a bonus for rare answers needs a coefficient above for groups of samples before the normalization and about two to three times that after it. In free-form TTRL on MATH-500, a bonus placed before the normalization prevented collapse and kept more correct answers among the samples, although it lowered pass@1.
Figures & tables
| before, | ||||
| after, broad, ten answers | ||||
| after, broad, free-form answers | ||||
| after, near collapse |
| method | acts as exploration |
|---|---|
| temperature | no, it rescales time (Proposition B.2 ) |
| clipping, Adam | no, they move no equilibrium of exploration in the loss but can change outcomes in the window (Lemma 4.2 , Appendix D ) |
| normalization | no, it rescales exploration it does not divide (Proposition 4.3 ) |
| KL penalty | partly, it pulls toward the reference policy and bypasses the division of GRPO (Section 5.2 ) |
| smoothing to | yes, boundaries and (Proposition 4.4 ) |
| entropy bonus | yes, a Potts window under NPG, the committed branch stable only numerically (Theorem B.3 ) |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| claim | section | status | scope |
|---|---|---|---|
| learning rate and preconditioner keep equilibria and folds | § 4.1 | proved | exact gradients, networks of full rank |
| dividing only selection rescales exploration by | § 4.1 , § 5.2 | proved | removal of the window numerical, mean-field model |
| window for unordered answers | § 4.2 | proved | both geometries, classical under NPG |
| collapse onto the initial plurality | § 4.2 | proved | smoothing toward |
| prevention costs least early | § 4.2 | proved | one leading answer and equal others |
| boundaries of the majority reward | § 4.2 | exact computation | one leading answer and equal others |
| ring | lower fold (peak) | eigenvalue | upper fold (peak) | eigenvalue | ||
|---|---|---|---|---|---|---|
| , | , | none | none | |||
| ( ) | ( ) | |||||
| ( ) | ( ) | |||||
| ( ) | ( ) |
| arm C, | ||||
|---|---|---|---|---|
| arm C, | ||||
| arm C, , | ||||
| arm R, end of branch ( there) | ( ) | ( ) | ( ) | ( ) |
| arm R, end of branch, |
| setting, seeds and boundaries | below the lower | inside the window | above the upper |
|---|---|---|---|
| testbed, , seeds 0, 1 (2 to 4), | collapsed at | prevented from , , released from | broad at , |
| testbed, both normalized, seed 0 and partly 1 (2 to 4), estimated | collapsed at , | prevented from to , released at by AdamW in seeds 0, 1 and 3 | broad at , |
| letter test, agreement, arm U, seeds 0 to 2 (3, 4), , arm B, seed 0, | committed at | depended on start, seed and learning rate, ramps of seeds 0 to 3 released at or just below the fold | broad at , |
| letter test, agreement, arm R, seeds 0 (3, 4), | committed at to | at , broad start committed in 2 of 3 seeds and collapsed start released, both broad at | broad at to |
| letter test, majority ramps, arm U, seeds 0, 1 (2, 3), | committed at the ends | committed at to , released at to | none committed |
| letter test, majority ramps, arm R, seeds 0, 1 (2, 3), | committed at the ends | loop closed in seed 0, open in seeds 1 to 3 | one brief excursion in seed 0 |
| arm | seed | release | collapse | loop, release minus collapse |
|---|---|---|---|---|
| U | 0 | |||
| U | 1 | |||
| R | 0 | |||
| R | 1 |
| arm C, threshold | ||||
|---|---|---|---|---|
| arm C, lower and upper coefficient | , | , | , | , |
| arm R, threshold, end of branch | ||||
| arm R, lower and upper coefficient | , | , | , | , |
| ratio of the brackets of arms R and C | to | to | to | to |
| arm R, threshold near collapse |
| difference | seed 1 | seed 2 | seed 3 | seed 4 | seed 5 | seed 6 | mean |
|---|---|---|---|---|---|---|---|
| arm C minus control | |||||||
| arm R minus control | |||||||
| arm C minus before training |
| setting | lower boundary | upper boundary | source |
|---|---|---|---|
| agreement reward, smoothing toward , answers | Proposition 4.4 | ||
| same, leave-one-out reward of groups of | Section 4.2 | ||
| majority reward, , , smoothing toward | Appendix C.5 | ||
| agreement reward, entropy bonus under NPG | , for | Theorem B.3 | |
| bonus before the division, arm C | , for | Section 5.2 | |
| bonus after the division, arm R | end of the interior branch, to | about | Table 1 |
Test-Time RL
Test-Time Reinforcement Learning adapts policies at deployment using interaction or self-generated feedback, without a separate task-specific training phase.
See all 49 papers