ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
41
Citations
3
Influential Citations
arXiv.org
Venue
2025
Year
Reinforcement Learning (RL) has proven highly effective at enhancing the complex reasoning abilities of Large Language Models (LLMs), yet underlying mechanisms driving this success remain largely opaque. Our analysis reveals that puzzling phenomena like ``aha moments", ``length-scaling''and entropy dynamics are not disparate occurrences but hallmarks of an emergent reasoning hierarchy, akin to the separation of high-level strategic planning from low-level procedural execution in human cognition. We uncover a compelling two-phase dynamic: initially, a model is constrained by procedural correctness and must improve its low-level skills. The learning bottleneck then decisively shifts, with performance gains being driven by the exploration and mastery of high-level strategic planning. This insight exposes a core inefficiency in prevailing RL algorithms like GRPO, which apply optimization pressure agnostically and dilute the learning signal across all tokens. To address this, we propose Hierarchy-Aware Credit Assignment (HICRA), an algorithm that concentrates optimization efforts on high-impact planning tokens. Our extensive experiments validate that HICRA significantly outperforms strong baselines, and offer deep insights into how reasoning advances through the lens of strategic exploration.
This paper addresses a critical gap in understanding how reinforcement learning (RL) actually improves the reasoning abilities of large language models (LLMs). While RL has been empirically successful—producing phenomena like "aha moments" and length scaling—the underlying mechanisms have remained opaque. The authors provide a compelling framework: reasoning improvement follows a two-phase hierarchy, first focusing on low-level procedural correctness, then shifting to high-level strategic planning. This insight is significant because it explains disparate observations under a unified theory and exposes a fundamental inefficiency in current RL algorithms like GRPO, which treat all tokens equally.
By identifying that the learning bottleneck shifts from low-level to high-level reasoning, the paper offers a principled way to improve RL training. The proposed HICRA algorithm directly addresses this inefficiency by concentrating optimization pressure on planning tokens, which are the high-impact decisions that drive strategic reasoning. This is a practical advance that could lead to faster training, better performance, and more interpretable models.
The paper reports that HICRA significantly outperforms strong baselines, including GRPO, on reasoning tasks. While specific numerical metrics are not provided in the abstract, the claim is supported by extensive experiments. The key result is that HICRA achieves better final performance and more efficient training by concentrating optimization on planning tokens, as predicted by the two-phase hierarchy theory.
This work has broad implications for AI research and practice. First, it provides a mechanistic understanding of how RL improves reasoning, which can guide the design of more efficient training algorithms. Second, the HICRA algorithm offers a practical improvement that could be adopted by practitioners training reasoning models. Third, the hierarchical reasoning framework aligns with cognitive science theories of human reasoning, suggesting a path toward more human-like AI reasoning. Finally, by making the training process more interpretable, this work contributes to AI safety and alignment efforts.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba