RLVR landscapes for iterated multiplications can be benign: Insights from spin-glass theory
Organizations: Racah Institute of Physics The Hebrew University of Jerusalem
Abstract
Despite the importance of reinforcement learning with verifiable rewards (RLVR), the extent to which it can learn new reasoning capabilities remains debated. Here we study the optimization landscape of RLVR on algorithmic tasks, such as iterated group and quasigroup multiplication. To this end, we map entropy-regularized RLVR over myopic tabular policies onto an energy-based (spin-glass) model over deterministic policies. This mapping upper-bounds what RLVR can achieve, and lets us rigorously characterize the landscape in this tabular setting. We show, both theoretically and experimentally, that for a wide class of models and tasks with uncorrelated inputs, this landscape is benign, containing no local minima that could trap RLVR training. Rather, the practical difficulty of these tasks appears to stem, at least in part, from issues such as diffusive barriers and gradient-estimation error in traversing the landscape. These are genuine obstacles that can prevent a solution from being found, but they are distinct from the landscape itself being rugged. We show that these obstacles can often be mitigated through the choice of entropy regulator. Consistent with this theory, we find that a transformer trained from scratch, using only last-token rewards, successfully learns an algorithmic chain of thought for iterated non-Abelian group multiplications.
Figures & tables
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Approach to equilibrium, parity , , replicas | ||||
|---|---|---|---|---|
| exact equilibrium reward | ||||
| sweeps | greedy reward | at the optimum | gap to equilibrium | |
| class | window, tying | chain | average RLVR reward | RLVR exactly |
|---|---|---|---|---|
| parity, , | untied | |||
| , | untied | |||
| quasigroup, | untied | |||
| quasigroup, | untied | |||
| quasigroup, | untied | |||
| quasigroup, | untied |
| class | reachable cells | steps to agreement |
|---|---|---|
| untied | ||
| untied | ||
| untied | ||
| untied | ||
| untied | ||
| tied |
| class | stretch | steps | barrier | largest barrier (in quanta= ) |
|---|---|---|---|---|
| untied | first | |||
| middle | ||||
| last | ||||
| untied | first | |||
| middle | ||||
| last |
| best of the scan | J @ | ||||||||
| class | seeds | sweeps | range of | at | at | ||||
| parity , | none | — | — | ||||||
| , | none | — | — | ||||||
| quasigroup , | |||||||||
| quasigroup , | |||||||||
| quasigroup , | |||||||||
| Correlated inputs, : every policy visited, every local maximum classified | ||||||
| cells | policies | local maxima | traps | trap frac. | escape barriers | |
| , | ||||||
| , | ||||||
| , , | ||||||