acceptodds
Under review as a conference paper at ICLR 2027

Right Makes Might: Aligning Verified Hidden States Empowers RL Reasoning

Abstract

Reinforcement Learning from Verifiable Rewards (RLVR) has become the dominant approach for improving mathematical reasoning in large language models, yet current methods reduce each correct rollout to a single reward bit, ignoring the geometric structure shared among their hidden states. Investigating this structure, we find that at the anchor token (the position immediately before the answer marker), correct rollouts converge naturally because they must produce the same answer (cosine similarity ≈ 0.84), yet each retains residual variance from its unique reasoning path. Encouraging full alignment at this point pushes the model to extract a unified “correct decision” representation, reducing sensitivity to which reasoning path was taken. Based on this observation, we propose Hidden-Align, an auxiliary loss function that aligns the last-layer hidden states of correct rollouts at the anchor token during RL training, without appreciable training overhead or additional inference cost. On eight mathematical reasoning benchmarks, Hidden-Align improves average pass@1 over the DAPO baseline by 3.8, 6.2, and 5.4 percentage points on Qwen3-1.7B, 4B, and 14B respectively, with consistent pass@k gains across all three scales. These benefits generalize across RL algorithms and model families, extending beyond mathematics to spatial reasoning. Systematic ablations support the design choices.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.