arXiv:2605.30233cs.CLcs.AI2026-05

语言模型用非逐步方式跟踪实体状态变化,机制与预期不同。

Do Language Models Track Entities Across State Changes?

论文配图:Do Language Models Track Entities Across State Changes?
图 1 · 摘自论文原文
  • 模型在最后一刻并行聚合信息,而非逐字跟踪状态变化
  • 删除操作依赖脆弱的全局抑制标记,导致多种失效行为
  • 结合行为与机制分析,可发现现有评估缺失的失败模式

实体跟踪(ET)是复杂推理的基础能力。尽管已有研究关注语言模型在无状态变化场景下的实体绑定,但对真实自然语言中复杂状态变化下的跟踪机制仍缺乏理解。本文研究了包含多个状态变更操作的复杂场景下语言模型的实体跟踪机制。结果表明,模型并非逐词或逐层增量跟踪世界状态,而是在查询明确后于最后一时刻并行聚合相关信息。进一步分析各类操作(PUT、REMOVE、MOVE)机制发现,模型通过一个脆弱的全局抑制标记实现删除操作,该机制可预测多种行为上的失败模式,且经验证成立。本文提出通过消除该标记来部分缓解问题。总体而言,语言模型以非顺序策略解决本质上顺序的任务。研究展示了行为分析与机制分析的相互促进:行为结果启发机制假设,机制洞察又能指导构建更完善的评估体系,预测现有评估中遗漏的失败模式。

原文摘要 · Abstract (English)

Entity tracking (ET), the ability to keep track of states, is a fundamental skill that underlies complex reasoning. An increasing amount of work investigates how transformer language models (LMs) solve entity binding $\textit{without}$ state changes. However, there is limited understanding of how non-toy LMs address ET problems of realistic difficulties expressed in natural language. To this end, we investigate the mechanisms underlying ET in more complex scenarios featuring multiple state-changing operations. We find that LMs do not incrementally track world states across tokens or query-relevant states across layers, but simply aggregate relevant information in parallel at the last token when the query becomes evident. We further investigate mechanisms of individual operations ($\texttt{PUT}$, $\texttt{REMOVE}$, $\texttt{MOVE}$) to characterize this non-incremental ET mechanism. Surprisingly, LMs implement the $\texttt{REMOVE}$ operation with a fragile global suppression tag; this global removal mechanism predicts various failure modes that we confirm behaviorally. We provide a mechanistic solution of nullifying this tag to partially address this issue. Overall, our findings reveal that LMs solve a fundamentally sequential task using a non-sequential strategy. More broadly, our work illustrates how behavioral and mechanistic analyses can fruitfully interact. Behavioral results inform mechanistic hypotheses, and insights from mechanistic analyses help build stronger behavioral evaluations by predicting failure modes missing from existing evaluations.

实体跟踪语言模型机制分析推理能力

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