mlsanityData to 7 Oct 2026

Latent Phase-Shift Rollback: Inference-Time Error Correction via Residual Stream Monitoring and KV-Cache Steering

arXiv 2604.18567 ↗First posted 20 Apr 2026cs.LG cs.AI cs.CLcited by 0

On the map: LLM reasoning → Test-time compute and efficient thinking

also in 1 more
Abstract (arXiv)

Large language models frequently commit unrecoverable reasoning errors mid-generation: once a wrong step is taken, subsequent tokens compound the mistake rather than correct it. We introduce Latent Phase-Shift Rollback (LPSR): at each generation step, we monitor the residual stream at a critical layer lcrit, detect abrupt directional reversals (phase shifts) via a cosine-similarity + entropy dual gate, and respond by rolling back the KV-cache and injecting a pre-computed steering vector. No fine-tuning, gradient computation, or additional forward passes are required. LPSR achieves 44.0% on MATH-500 with an 8B model versus 28.8% for standard AR (+15.2 pp; McNemar chi2 = 66.96, p < 10-15). Critically, prompted self-correction, the most natural inference-time baseline, scores only 19.8%, below standard AR; LPSR exceeds it by +24.2 pp (chi2 = 89.4, p ≈ 0). LPSR also outperforms Best-of-16 (+7.8 pp) at 5.4× lower token cost, and surpasses a standard 70B model (35.2%) with 8.75× fewer parameters at ~3× the token budget. A 32-layer sweep reveals a novel detection-correction dissociation: error-detection AUC peaks at layer~14 (0.718) but task accuracy peaks at layer~16 (44.0% vs. 29.2%), demonstrating that optimal monitoring depth differs for detection and correction.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 17 of its 27 references that are in the map. First posted on arXiv 20 Apr 2026.

Builds on

17 of 27 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025The Illusion of Thinking: Understanding the Strengths and Limitations of Reasoning Models via the Lens of Problem Complexity7 Jun
  2. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters6 Aug
  3. 2023Large Language Models Cannot Self-Correct Reasoning Yet3 Oct
  4. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Models17 May
  5. 2023Self-Refine: Iterative Refinement with Self-Feedback30 Mar

Other references · 12

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  3. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  4. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  5. 2023Let's Verify Step by Stepcited by 3,598
  6. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
  7. 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,239
  8. 2023Graph of Thoughts: Solving Elaborate Problems with Large Language Modelscited by 873
  9. 2022Solving math word problems with process- and outcome-based feedbackcited by 871
  10. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  11. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
  12. 2024KV Cache Compression, But What Must We Give in Return? A Comprehensive Benchmark of Long Context Capable Approachescited by 43

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026BackPlay: Head-Only Look-Back Self-Correction for Diffusion Language Models
  2. 2026Entropy-Gated Latent Recursion
  3. 2026The Shape of Overthinking: Backtracking Bursts in Long Reasoning Traces
  4. 2026"I May Not Have Articulated Myself Clearly": Diagnosing Dynamic Instability in LLM Reasoning at Inference Time
  5. 2026Archer: Adaptive Reuse of Cached Hidden States for Efficient Rollback in Diffusion Language Models
  6. 2026Dissecting Failure Dynamics in Large Language Model Reasoning
  7. 2026SPREG: Structured Plan Repair with Entropy-Guided Test-Time Intervention for Large Language Model Reasoning
  8. 2026Reasoning Errors Have a Region and a Direction in the Residual-Stream Trajectory of LLMs