Steering Recurrent Reasoners at Inference Time with Readout Feedback

Shunsuke Kamiya, Masanori Koyama, Seongcheol Jeong, Fumiya Uchiyama, Kenji Kubo +3 more
8/25/2026
cs.LG

No confident code implementation found for this paper.

Abstract

Recurrent models, which repeatedly update latent states with shared computation blocks, have emerged as powerful architectures for solving complex reasoning tasks. Existing inference-time methods scale computation by running more steps or sampling more trajectories, but ignore information revealed within each trajectory. Here we show that recurrent models can be improved at inference time by using their own readout probabilities to steer latent dynamics without retraining. We introduce Readout Feedback (RoFB), a test-time intervention that converts intermediate predictions into token-wise pairwise coupling forces injected into the latent dynamics. Across three recurrent models (AKOrN, ItrSA++, TRM) on Sudoku and Maze, RoFB yields clear gains in four of six model-task pairs, achieving performance unattainable by merely running more steps or selecting from multiple trajectories, at comparable or lower computational cost. These results suggest that closed-loop steering of latent dynamics can serve as a complementary inference-time control mechanism for recurrent reasoning models.

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{kamiya2026steering,
  title  = {Steering Recurrent Reasoners at Inference Time with Readout Feedback},
  author = {Shunsuke Kamiya and Masanori Koyama and Seongcheol Jeong and Fumiya Uchiyama and Kenji Kubo and Kohei Hayashi and Masahiro Suzuki and Yutaka Matsuo},
  year   = {2026},
  eprint = {2608.24136},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2608.24136v1}
}

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/2608.24136.svg)](https://codeofpaper.com/paper/2608.24136)

Discussion