Latent Phase-Shift Rollback: Inference-Time Error Correction via Residual Stream Monitoring and KV-Cache Steering
Abstract
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 (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 on MATH-500 with an 8B model versus for standard AR ( pp; McNemar , ). Critically, prompted self-correction, the most natural inference-time baseline, scores only , below standard AR; LPSR exceeds it by pp (, ). LPSR also outperforms Best-of-16 ( pp) at lower token cost, and surpasses a standard 70B model () with fewer parameters at the token budget. A 32-layer sweep reveals a novel \textbf{detection-correction dissociation}: error-detection AUC peaks at layer~14 () but task accuracy peaks at layer~16 ( vs.\ ), demonstrating that optimal monitoring depth differs for detection and correction.
Keywords
Cite
@article{arxiv.2604.18567,
title = {Latent Phase-Shift Rollback: Inference-Time Error Correction via Residual Stream Monitoring and KV-Cache Steering},
author = {Manan Gupta and Dhruv Kumar},
journal= {arXiv preprint arXiv:2604.18567},
year = {2026}
}
Comments
Under Review