Dynamical Systems Theory Behind a Hierarchical Reasoning Model
cs.AI, cs.LG, math.DS
Submitted: 2026-03-24
Updated: 2026-03-24
Journal ref: Neurocomputing, 2027, Volume 709, 135271, https://www.sciencedirect.com/science/article/pii/S092523122602669X?dgcid=author
DOI: 10.1016/j.neucom.2026.135271
Code: https://github.com/lucidrains/adam_atan2_pytorch
License: http://creativecommons.org/licenses/by/4.0/
The gist: Current large language models (LLMs) primarily rely on linear sequence generation and massive parameter counts, yet they severely struggle with complex algorithmic reasoning.
Terminology
Abstract
Current large language models (LLMs) primarily rely on linear sequence generation and massive parameter counts, yet they severely struggle with complex algorithmic reasoning. While recent reasoning architectures, such as the Hierarchical Reasoning Model (HRM) and Tiny Recursive Model (TRM), demonstrate that compact recursive networks can tackle these tasks, their training dynamics often lack rigorous mathematical guarantees, leading to instability and representational collapse. We propose the Contraction Mapping Model (CMM), a novel architecture that reformulates discrete recursive reasoning into continuous Neural Ordinary and Stochastic Differential Equations (NODEs/NSDEs). By explicitly enforcing the convergence of the latent phase point to a stable equilibrium state and mitigating feature collapse with a hyperspherical repulsion loss, the CMM provides a mathematically grounded and highly stable reasoning engine. On the Sudoku-Extreme benchmark, a 5M-parameter CMM achieves a state-of-the-art accuracy of 93.7 %, outperforming the 27M-parameter HRM (55.0 %) and 5M-parameter TRM (87.4 %). Remarkably, even when aggressively compressed to an ultra-tiny footprint of just 0.26M parameters, the CMM retains robust predictive power, achieving 85.4 % on Sudoku-Extreme and 82.2 % on the Maze benchmark. These results establish a new frontier for extreme parameter efficiency, proving that mathematically rigorous latent dynamics can effectively replace brute-force scaling in artificial reasoning.
Sources
- Hierarchical Reasoning Model
- A Logic for Expressing Log-Precision Transformers
- Transformers in Uniform TC$^0$
- Chain-of-Thought Prompting Elicits Reasoning in Large Language Models
- Hierarchical Reasoning Models: Perspectives and Misconceptions
- Less is More: Recursive Reasoning with Tiny Networks
- Emergent Hierarchical Reasoning in LLMs through Reinforcement Learning
- MLP-Mixer: An all-MLP Architecture for Vision
- Deep Equilibrium Optical Flow Estimation
- SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit models
- Stabilizing Equilibrium Models by Jacobian Regularization
- Neural Ordinary Differential Equations
- Grokking at the Edge of Numerical Stability
- Neural SDE: Stabilizing Neural ODE Networks with Stochastic Noise
- Neural SDEs as a Unified Approach to Continuous-Domain Sequence Modeling
- GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks
- Training Deep Nets with Sublinear Memory Cost
- Memory-Efficient Backpropagation Through Time
- Scaling Exponents Across Parameterizations and Optimizers
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems