Search papers, labs, and topics across Lattice.
This paper investigates unsupervised reinforcement learning (RL) for mathematical reasoning in LLMs, focusing on intrinsic rewards that promote concise and certain generation. The authors evaluate various intrinsic rewards across base models with differing reasoning capabilities, finding that the success of unsupervised RL depends heavily on the base model's inherent logical prior. Through a novel geometric analysis, they demonstrate that successful unsupervised RL training is characterized by the policy trajectory being enveloped by manifolds, providing a diagnostic tool for stability.
Unsupervised RL for math reasoning hinges on a model's pre-existing logical abilities, and its success can be predicted by whether the training trajectory stays within stable "manifolds" of good solutions.
Although outcome-based reinforcement learning (RL) significantly advances the mathematical reasoning capabilities of Large Language Models (LLMs), its reliance on computationally expensive ground-truth annotations imposes a severe scalability bottleneck. Unsupervised RL guided by intrinsic rewards offers a scalable alternative, yet it suffers from opaque training dynamics and catastrophic instability, such as policy collapse and reward hacking. In this paper, we first design and evaluate a suite of intrinsic rewards that explicitly enforce concise and certain generation. Second, to discover the boundaries of this approach, we test base models across a spectrum of intrinsic reasoning capabilities, revealing how a model's foundational logical prior dictates its success or failure. Finally, to demystify why certain configurations stabilize while others collapse, we introduce a novel geometric diagnostic lens, showing that successful cases are enveloped by manifolds. Ultimately, our work goes beyond merely demonstrating that enforcing concise and certain responses successfully boosts mathematical reasoning; we reveal when this unsupervised approach breaks down and geometrically diagnose why.
