Right Direction, Wrong Step: Geometric Analysis of Finite-Step Failure in Looped Transformers

Zhihao Guo, Zonghan Wu, Haizhou Du, Huan Huo, Yilei Shao +2 more
9/15/2026
cs.LGcs.AI

No confident code implementation found for this paper.

Abstract

Looped Transformers offer a parameter-efficient route to test-time scaling by reusing shared layers for iterative latent reasoning. However, additional iterations can reduce support for a reference answer, leaving unclear whether an update's direction is locally unhelpful or its full displacement moves too far. We study this distinction by analysing reference utility, which measures this support, along the model's own update direction, varying the fraction of the proposed displacement supplied to the readout. This reveals finite-step failures in which a locally improving direction produces a harmful full update. A pathwise curvature decomposition characterises how initial progress is lost, while a local quadratic model predicts full-step gains and useful step scales. Bounds based on accumulated curvature variation characterise the approximation error of these predictions. Experiments across two model families reveal this separation on mathematical and commonsense tasks. A fixed quarter step produces positive gains in reference utility for 72.2--83.2% of selected failures across four settings. These findings identify a mismatch between update direction and step scale as a mechanism of lost progress, explaining how some harmful updates retain useful computation.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{guo2026right,
  title  = {Right Direction, Wrong Step: Geometric Analysis of Finite-Step Failure in Looped Transformers},
  author = {Zhihao Guo and Zonghan Wu and Haizhou Du and Huan Huo and Yilei Shao and Athanasios V. Vasilakos and Qingsong Wen},
  year   = {2026},
  eprint = {2609.16665},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2609.16665v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2609.16665.svg)](https://codeofpaper.com/paper/2609.16665)

Discussion