Latent Chain-of-Thought? Decoding the Depth-Recurrent Transformer

Wenquan Lu, Yuechuan Yang, Kyle Lee, Yanshu Li, Enqi Liu
7/2/2025
cs.CLcs.AIcs.LG

Official code: wenquanlu/huginn-latent-cot (21 stars, last commit 2025-10-04).

Abstract

Chain-of-thought (CoT) reasoning has enabled transformer-based language models to excel at complex mathematics and multi-step planning. However, in standard decoder-only architectures, these reasoning steps are externalized in natural language, improving interpretability at the cost of efficiency. To capture reasoning that is not easily represented in words, many works have explored recurrent architectures that aim to internalize reasoning in latent space, potentially supporting latent CoT. In this paper, we investigate whether such reasoning structures emerge in Huginn-3.5B, a depth-recurrent Transformer that reuses layers at inference time without increasing parameter count. We examine the model's internal behavior on arithmetic tasks using a suite of probing techniques including the Logit Lens and Coda Lens. Our findings reveal limited evidence of interpretable latent CoT by tracking rank trajectories of final and intermediate result tokens. Furthermore, we uncover significant probing inconsistencies across recurrent blocks, where the interpretability of hidden states depends heavily on both the layer index and the decoding method. Finally, we empirically show that increasing recurrence depth yields only marginal gains and falls well short of models that explicitly externalize reasoning steps. The code is available at https://github.com/wenquanlu/huginn-latent-cot.

View on arXivView PDF

Code Implementations(1)

[COLM 2025: 1st Workshop on the Application of LLM Explainability to Reasoning and Planning] Latent Chain-of-Thought? Decoding the Depth-Recurrent Transformer

213PythonApr 13, 20251 years ago

Similar Papers

Cite this paper

@article{lu2025latent,
  title  = {Latent Chain-of-Thought? Decoding the Depth-Recurrent Transformer},
  author = {Wenquan Lu and Yuechuan Yang and Kyle Lee and Yanshu Li and Enqi Liu},
  year   = {2025},
  eprint = {2507.02199},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2507.02199v2}
}

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

Discussion