Reasoning Fails Where Step Flow Breaks

arXiv cs.AI / 4/10/2026

💬 OpinionIdeas & Deep AnalysisModels & Research

Key Points

  • Step-Saliency is proposed to analyze long-chain “step-to-step” information flow in large reasoning models by pooling attention–gradient scores along the question→thinking→summary trajectory.
  • The paper identifies two recurring failure modes in LRM reasoning: Shallow Lock-in (over-focusing on the current step and underusing earlier context) and Deep Decay (saliency on the thinking segment fades while the summary increasingly attends to itself/last steps).
  • Based on these findings, StepFlow is introduced as a saliency-inspired test-time intervention that corrects shallow-layer saliency via Odds-Equal Bridge and mitigates deep-layer “decay” using Step Momentum Injection without retraining.
  • Experiments across multiple LRM architectures show StepFlow improves accuracy on math, science, and coding tasks, suggesting that repairing information flow can partially restore missing reasoning performance.
  • The work emphasizes that existing interpretability/analysis tools struggle with long structured reasoning traces and provides a more targeted mechanism to diagnose where reasoning breaks down.

Abstract

Large reasoning models (LRMs) that generate long chains of thought now perform well on multi-step math, science, and coding tasks. However, their behavior is still unstable and hard to interpret, and existing analysis tools struggle with such long, structured reasoning traces. We introduce Step-Saliency, which pools attention--gradient scores into step-to-step maps along the question--thinking--summary trajectory. Across several models, Step-Saliency reveals two recurring information-flow failures: Shallow Lock-in, where shallow layers over-focus on the current step and barely use earlier context, and Deep Decay, where deep layers gradually lose saliency on the thinking segment and the summary increasingly attends to itself and the last few steps. Motivated by these patterns, we propose StepFlow, a saliency-inspired test-time intervention that adjusts shallow saliency patterns measured by Step-Saliency via Odds-Equal Bridge and adds a small step-level residual in deep layers via Step Momentum Injection. StepFlow improves accuracy on math, science, and coding tasks across multiple LRMs without retraining, indicating that repairing information flow can recover part of their missing reasoning performance.