arXiv:2512.25026cs.CLcs.AI2025-12被引 3

让语言模型像人一样分层思考,提升推理与泛化能力

Modeling Language as a Sequence of Thoughts

  • 用双重抽象建模:词元与句子级‘思想状态’并行生成
  • 通过工作记忆跨句注意力,使早期句子表示可被后续优化
  • 在关系推理和数据效率上显著优于GPT-2,适合长文本理解任务

Transformer语言模型虽能生成自然文本,但依赖表层共现统计,难以形成全局一致的实体与事件隐式表征,导致关系泛化差(反转谬误)、上下文错误及数据效率低。认知科学表明,人类理解将语言转化为持久的、事件化的紧凑表征,而原文记忆短暂。受此启发,我们提出思想整体(Thought Gestalt, TG)模型,一种双层抽象的循环Transformer:同时建模词元与句子级‘思想’状态。TG逐句生成,并通过交叉注意力访问先前句子的表征工作记忆。词元与句子表征共享同一Transformer堆栈,以统一的下一个词预测损失训练。通过保留句子表征写入工作记忆的计算图,未来词预测损失的梯度可反向通过交叉注意力优化早期句子向量参数。在缩放实验中,TG相较匹配的GPT-2运行,在数据与参数效率上持续更优,缩放拟合显示GPT-2需约5-8%更多数据和33-42%更多参数才能达到TG的测试损失。TG还在父亲-儿子反转谬误探针中减少关系方向泛化错误。

原文摘要 · Abstract (English)

Transformer language models can generate strikingly natural text by modeling language as a sequence of tokens, but by relying primarily on surface-level co-occurrence statistics they fail to form globally consistent latent representations of entities and events, which contributes to poor relational generalization (the reversal curse), contextualization errors, and data inefficiency. Cognitive science, by contrast, shows that human comprehension converts linguistic input into compact, event-like representations that persist in memory while verbatim form is short-lived. Motivated by these findings, we introduce the Thought Gestalt (TG) model, a recurrent transformer that models language at two levels of abstraction: tokens and sentence-level "thought" states. TG generates one sentence at a time while cross-attending to a working memory of prior sentence representations. Token and sentence representations are generated using a shared stack of transformer blocks and trained with a single objective, next-token prediction loss. By retaining the computation graph of sentence representations written to working memory, gradients from future token losses flow backward through cross-attention to optimize the parameters that generate earlier sentence vectors. In scaling experiments, TG consistently improves data and parameter efficiency compared to matched GPT-2 runs and other baselines, with scaling fits indicating GPT-2 requires ~5-8% more data and ~33-42% more parameters to match TG's test loss. TG also reduces errors in relational-direction generalization on a father-son reversal curse probe.

语言建模思想状态关系推理高效学习

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