arXiv:2608.07077cs.AIcs.LG2026-08

大模型会建世界模型却记不住,导致解复杂汉诺塔失败

Transformers Struggle to Use Their Emergent World Models: Revisiting the Tower of Hanoi, and the Illusion of Thinking

  • 小模型通过训练学会构建几何化的汉诺塔状态空间表示
  • 大模型虽能完美编码状态图,但3环以上任务仍大量失败
  • 失败主因是推理中世界模型信息逐渐丢失,可注入修复

汉诺塔是检验推理能力的经典规划谜题。尽管当前大型推理模型(LRMs)能解决标准形式,但在初始与目标状态不限于单柱的平铺变体上仍表现不佳。本文深入研究了小型自研Transformer与大型第三方模型(Qwen3.6-27B和DeepSeek-R1-Distill-Qwen-32B)的求解机制。首先,我们从零训练小型Transformer,结合多种可解释性技术发现其发展出一个可线性解码、几何忠实的‘世界模型’——即汉诺塔状态空间的谢尔宾斯基三角形结构,且该表征在解题中起因果作用。接着,对两个前沿推理模型进行分析,发现它们在提示末尾几乎完美地编码了该世界模型,但在超过3个环时仍普遍失败。我们追踪推理过程中的表征演化,确认失败根源在于世界模型的衰减。通过在推理中注入提示阶段的表征,性能显著提升,证明问题不在于缺失模型,而在于维护失效。因此,性能下降实为‘建模后遗忘’,且可部分恢复。

原文摘要 · Abstract (English)

The Tower of Hanoi is a simple planning puzzle that in prior work has proven challenging for large reasoning models (LRMs). Current models solve the standard formulation of the puzzle, but still struggle with the flat-to-flat variant (where initial and goal states are not restricted to have all rings on a single peg). This paper presents an in-depth study of how both small, in-house Transformers and large, third-party LRMs solve this task. To understand the failures mechanistically, we first train small Transformers from scratch on precomputed solution traces. Using a variety of interpretability techniques, we show that these Transformers develop an emergent world model: a linearly decodable, geometrically faithful representation of the puzzle's state space (the Sierpinski triangle), that is causally involved in solving the puzzles. Second, we return to the large LLMs and apply our techniques to two frontier reasoning models, Qwen3.6-27B and DeepSeek-R1-Distill-Qwen-32B, that attempt to solve the task through extended chain-of-thought. Surprisingly, we find that both models encode the Sierpinski world model near-perfectly at the end of the prompt, and yet fail at the majority of tasks when there are more than 3 rings. We locate the source of this failure in the decaying representation of the world model. We probe for the representation at different stages during planning, and establish causality by showing that performance can be improved by injecting the prompt-time representation at inference. The failure of the models is thus one of maintenance of the required representations, not their absence, and performance is at least partially recoverable. These results thus reframe the reported collapse in performance from prior work: current Large Reasoning Models build a world model, and then lose it.

大模型世界模型汉诺塔可解释性

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