Entropy-Gated Latent Recursion

TL;DR

Introduces Entropy-Gated Latent Recursion (EGLR), leveraging layer span L for deterministic inference diversity, boosting reasoning accuracy.

cs.LG 🔴 Advanced 2026-06-15 56 views
Soham Bhattacharjee Dushyant Singh Chauhan Salem Lahlou Martin Takac Nils Lukas
NLP reasoning enhancement model interpretability diversity inference scaling

Key Findings

Methodology

This paper proposes a deterministic inference axis based on layer span L, combined with an entropy gating mechanism that recursively re-applies the top-L layers at high-uncertainty tokens. This process forms an L×T Cartesian sampling space, where T is the temperature. Without training, the method iterates up to Kmax times until the token distribution converges. The approach expands the inference trajectory space, capturing diverse reasoning paths. Experiments across 8 instruction-tuned models and 6 math reasoning benchmarks demonstrate that L and temperature axes are complementary, yielding significant performance gains.

Key Results

  • On Qwen2.5-3B-Instruct’s MATH-500 dataset, the joint L×T oracle achieves 91.6%, surpassing the temperature-only oracle (83.4%) by 8.2 percentage points and the layer-only oracle (81.2%) by 10.4 points, confirming the axes' complementarity.
  • Across multiple models and tasks, EGLR-SC (self-consistency) outperforms greedy decoding in 44/48 cases and exceeds FLOP-matched beam search in most instances.
  • Different L values produce distinct, deterministic reasoning trajectories, with no single L dominating, highlighting the structural diversity introduced by the layer span axis.

Significance

This work introduces a novel, training-free method to enhance inference diversity by exploiting deterministic structural variations, addressing the limitations of stochastic noise-based methods. It significantly improves reasoning performance on complex tasks, offering a scalable, interpretable, and efficient approach. The method’s ability to generate richer candidate sets benefits downstream applications like reranking, ensemble methods, and reinforcement learning, advancing the state of AI reasoning capabilities.

Technical Contribution

The paper pioneers the use of layer span L as a deterministic inference axis, combined with an entropy gating mechanism that recursively re-invokes the top-L layers at uncertain tokens. This creates a multi-dimensional inference space, expanding beyond traditional stochastic sampling. The approach requires no additional training or parameters, enabling easy integration into existing models. The formulation guarantees convergence and provides theoretical insights into the diversity of reasoning paths, opening new avenues for inference-time scaling.

Novelty

This is the first systematic use of layer span L as a deterministic, structural axis for inference diversity, distinct from stochastic sampling. It leverages entropy gating to selectively recurse, forming a multi-dimensional inference space that captures reasoning trajectories inaccessible to temperature-based methods alone. This innovation bridges the gap between deterministic and stochastic approaches, offering a new paradigm for reasoning enhancement.

Limitations

  • The optimal choice of L varies across tasks and models, requiring hyperparameter tuning, which adds complexity.
  • Recursive iterations are capped at Kmax, possibly limiting exploration of all potential reasoning paths in complex scenarios.
  • In extremely high-uncertainty situations, the recursion may not fully capture all relevant trajectories, necessitating external knowledge integration in future work.

Future Work

Future research will focus on adaptive, dynamic adjustment of layer span L based on context complexity, integrating external knowledge bases, and feedback mechanisms to further improve reasoning robustness. Extending the approach to multi-modal tasks and real-time applications could broaden its impact, making AI systems more reliable and interpretable in diverse domains.

AI Executive Summary

In the rapidly evolving field of natural language processing, enhancing model reasoning remains a central challenge. Traditional methods like self-consistency and beam search rely heavily on stochastic sampling, which introduces noise but limits structural diversity. This paper presents a groundbreaking approach called Entropy-Gated Latent Recursion (EGLR), which introduces a deterministic inference axis based on the layer span L. By recursively reapplying the top-L layers at high-uncertainty tokens, EGLR generates multiple distinct reasoning trajectories without additional training or parameters.

The core innovation lies in combining this deterministic layer span axis with the stochastic temperature T, forming an L×T Cartesian sampling space. This multi-dimensional space significantly expands the inference candidate pool, capturing reasoning paths that stochastic noise alone cannot produce. Experiments across eight instruction-tuned models and six math reasoning benchmarks demonstrate that this approach yields substantial performance improvements. For instance, on Qwen2.5-3B-Instruct’s MATH-500 dataset, the joint oracle achieves 91.6%, outperforming single-axis methods by over 8 percentage points.

The method’s strength is its ability to produce diverse, structurally distinct reasoning paths efficiently, with minimal overhead. It also enhances downstream tasks such as reranking, ensemble, and reinforcement learning, by providing richer candidate sets. The results confirm that the layer span L and temperature axes are genuinely complementary, unlocking new possibilities for inference-time scaling.

Looking ahead, future work aims to develop adaptive L selection, integrate external knowledge, and extend to multi-modal reasoning. This approach opens a promising new direction for scalable, interpretable, and robust AI reasoning, moving beyond the limitations of stochastic noise-based methods and setting a new standard for inference enhancement.

Deep Analysis

Background

The evolution of large-scale pre-trained language models has significantly advanced NLP capabilities. Early approaches like beam search and stochastic sampling improved answer diversity but were limited by their reliance on randomness, which often led to redundant trajectories. Recent work introduced recursive reasoning modules (Giannou et al., 2023; Hao et al., 2024), requiring training, to explore latent reasoning paths. Li et al. (2026) proposed entropy-based gating to select uncertain tokens for refinement, but still depended on external modules. The challenge remains to generate diverse, meaningful reasoning paths efficiently without retraining, especially for complex tasks like math reasoning. This paper addresses this gap by proposing a deterministic, structure-based inference axis, leveraging the model’s internal layer span, to produce rich reasoning trajectories without additional training or parameters.

Core Problem

Current inference-time scaling methods primarily depend on stochastic token sampling, which limits the diversity of reasoning paths to noise-driven variations. This stochasticity often results in near-duplicate trajectories, reducing the effectiveness of ensemble and reranking strategies. Furthermore, stochastic methods cannot access certain reasoning pathways that are structurally distinct but deterministic in nature. The core problem is how to systematically generate multiple, diverse, and meaningful reasoning trajectories without retraining or incurring significant computational overhead. Addressing this problem is crucial for improving the robustness, interpretability, and accuracy of large language models in complex reasoning tasks.

Innovation

The paper introduces three key innovations: 1) The layer span L as a deterministic, structural axis for inference, enabling multiple reasoning paths by re-invoking the top-L layers at high-uncertainty tokens. 2) An entropy gating mechanism that triggers recursive re-application only at uncertain tokens, avoiding unnecessary computation. 3) The formation of an L×T Cartesian sampling space by combining layer span L with temperature T, greatly expanding the candidate pool without retraining. These innovations collectively enable a structured, multi-path reasoning process that is both efficient and effective, surpassing traditional stochastic methods in diversity and performance.

Methodology

  • �� Use a frozen autoregressive transformer with N layers, compute residual states at each layer.
  • �� During decoding, calculate token entropy H(pt) to identify uncertain tokens.
  • �� When H(pt) exceeds threshold τH, trigger recursive re-application of the top-L layers, up to Kmax iterations.
  • �� For each high-uncertainty token, fuse the anchor state with the refined state via a weighted average controlled by α.
  • �� Generate multiple reasoning trajectories by varying L (layer span) deterministically.
  • �� Combine these with temperature T sampling to form an L×T inference space.
  • �� Use self-consistency (EGLR-SC) to aggregate multiple paths via majority voting, improving accuracy.
  • �� The entire process requires no training, only inference-time recursion guided by entropy thresholds.

Experiments

The experiments evaluate 8 instruction-tuned models across six benchmarks, including GSM8K and MATH-500. Hyperparameters like Kmax=3 and τH are auto-calibrated. The study compares greedy, single-axis (L or T), and joint L×T sampling, measuring accuracy, oracle coverage, and path diversity. Ablation studies analyze the impact of L and Kmax. Results show that the joint L×T approach consistently outperforms single-axis methods, with improvements up to +8.2 percentage points on MATH-500. The experiments also compare with FLOP-matched beam search, demonstrating superior performance and diversity, validating the method’s effectiveness across models and tasks.

Results

The joint L×T oracle achieves 91.6% accuracy on MATH-500, surpassing the temperature-only oracle (83.4%) by 8.2 points and the layer-only oracle (81.2%) by 10.4 points. EGLR-SC improves accuracy in 44/48 model-task pairs, outperforming greedy and beam search at matched FLOPs. Different L values produce diverse, deterministic reasoning paths, with no single L dominating, confirming the structural diversity. The combined approach effectively captures reasoning trajectories inaccessible to stochastic sampling alone, demonstrating broad applicability and robustness.

Applications

This method can be integrated into systems requiring high reasoning accuracy, such as mathematical problem solving, scientific question answering, and legal reasoning. Its ability to generate multiple, diverse reasoning paths at low cost enhances ensemble, reranking, and reinforcement learning applications. The approach is suitable for deployment in real-time AI assistants, educational tools, and automated reasoning systems, especially where interpretability and robustness are critical. Future extensions could incorporate external knowledge bases and feedback loops for further improvement.

Limitations & Outlook

The choice of layer span L is task-dependent and requires tuning, adding complexity. Recursive iterations capped at Kmax may limit exploration of all potential paths, especially in highly complex scenarios. In cases of extremely high uncertainty, the recursive process might not fully capture all relevant trajectories. Computational overhead, though minimal, still exists compared to greedy decoding. Future work should focus on adaptive L selection, external knowledge integration, and scaling to multi-modal reasoning to address these limitations.

Plain Language Accessible to non-experts

想象你在做一道复杂的数学题,传统的方法是一步步按照公式推导,可能会遇到难题卡住。现在,你试试一种新办法:在遇到难题时,不只用一种思路,而是多试几种不同的解法,每次都用不同的角度去思考。最后,把这些不同的解法结合起来,找到最靠谱的答案。这就像模型在推理时,平时只用一种随机方法(像扔骰子一样)得到答案,但这次我们在关键点上多次递归,用不同的“思路”反复尝试,最后选出最合理的答案。这种方法不需要额外学习,只是在原有模型上多试几次,就能得到更丰富、更准确的推理路径,就像用不同的解法拼出一幅更完整的画。

ELI14 Explained like you're 14

想象你在玩一个拼图游戏,平时你只试一种拼法,看能不能拼成完整的图案。但如果你在关键的拼块上多试几次,用不同的拼法,最后把所有拼出来的部分拼在一起,就会拼出一个更漂亮、更完整的图。这就像模型在推理时,只用一种随机方法得到答案,而这次我们在关键点上多次递归,用不同的“拼法”尝试,最后把最好的答案拼出来。这个新办法不需要额外学习,只是在原有模型上多试几次,效果就会大大提升。就像拼图游戏一样,多试几次,总能拼出更完整、更漂亮的图案。

Abstract

Inference-time scaling has become the dominant lever for improving language-model reasoning, but existing methods derive rollout diversity from a single source: stochastic token-level sampling. We argue that this single-axis sampling space is fundamentally limiting, and identify a second, fully deterministic and complementary axis: the layer span $L$ at which a frozen model's top decoder layers are recursively re-applied at high-uncertainty tokens. Different choices of $L$ produce distinct rollouts that solve different subsets of problems, with no stochasticity. We instantiate this axis through Entropy-Gated Latent Recursion (EGLR), a training-free decoding procedure that re-applies the top-$L$ layers for at most $K_{\max}$ iterations until the next-token distribution converges. Combined with $T$ temperature samples, EGLR turns a single-axis stochastic rollout pool into an $L\times T$ Cartesian sampling space at almost the same per-rollout cost. We characterize this space across $8$ instruction-tuned models and $6$ math reasoning benchmarks, and show that the $L$-axis is genuinely complementary to temperature: on MATH-500 with Qwen2.5-3B-Instruct, the joint $L\times T$ oracle reaches $91.6\%$, $+8.2$ percentage points beyond the temperature-only oracle ($83.4\%$) and $+10.4$ points beyond the layer-only oracle ($81.2\%$), confirming that the two axes capture genuinely complementary problems. The expanded rollout pool provides richer per-prompt candidates for any downstream procedure that consumes rollouts, including self-consistency, best-of-$N$ with verifiers, and group-relative RL training (GRPO), opening a new direction for inference-time scaling that does not rely on stochastic noise.

cs.LG cs.AI