arXiv:2507.02199cs.CLcs.AI2025-07被引 20

探索深度循环Transformer能否在隐空间内实现类思维链推理。

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

  • 用深度循环架构在不增加参数下复用层,尝试让模型在隐空间中进行推理。
  • 实验显示隐状态的可解释性弱,中间结果的秩轨迹不明显。
  • 递归深度提升对性能帮助有限,不如显式外部化推理的模型。

思维链(CoT)推理使基于Transformer的语言模型在复杂数学和多步规划任务中表现优异。然而,在标准解码器架构中,这些推理步骤以自然语言形式外化,虽提升可解释性但降低效率。为捕捉难以用语言表达的推理过程,许多研究尝试使用递归架构将推理内化于隐空间,可能支持隐空间思维链(latent CoT)。本文研究Huginn-3.5B这一深度循环Transformer在算术任务中的内部行为,采用日志透镜(Logit Lens)与音符透镜(Coda Lens)等探针技术。结果表明,通过追踪最终与中间结果令牌的秩轨迹,缺乏可解释的隐空间思维链证据。此外,发现递归模块间存在显著探针不一致性,隐藏状态的可解释性高度依赖层索引和解码方式。最后,实验证明增加递归深度仅带来微弱收益,远不及显式外化推理步骤的模型。代码已公开于https://github.com/wenquanlu/huginn-latent-cot。

原文摘要 · Abstract (English)

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.

深度循环隐空间推理可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。