arXiv:2501.06382cs.CLcs.AI2025-01NeurIPS被引 3

揭示自注意力模型中自发话题切换的机制与人类思维的差异

Dynamics of Spontaneous Topic Changes in Next Token Prediction with Self-Attention

  • 用令牌优先图定义话题,理论分析模型话题维持规律
  • 话题突变需低优先级令牌数量超过高优先级者,且上下文越长越难突变
  • 首次在大模型中验证该现象,适合研究认知对比与模型可解释性

人类认知中存在由情绪、上下文或联想线索引发的突发性话题切换,即神经科学中的自发思维。而基于自注意力的模型依赖输入结构化模式预测下一个词,缺乏自发性。本文聚焦这一差异,刻画自注意力架构中的自发话题切换动态。首先,在简化单层自注意力模型下,通过定义话题为令牌优先图(TPGs),理论上证明:(1) 模型保持输入话题相关令牌的优先顺序;(2) 仅当低优先级令牌数量超过所有高优先级令牌时,才可能发生话题突变;(3) 与人类不同,上下文长度越长或输入话题越模糊,话题突变概率越低。其次,实证验证这些动态在现代先进大模型中依然存在,凸显人类认知与人工智能在自发话题切换上的根本差异。据我们所知,这是首个以贴近人类思维视角研究此类问题的工作。

原文摘要 · Abstract (English)

Human cognition is punctuated by abrupt, spontaneous shifts between topics-driven by emotional, contextual, or associative cues-a phenomenon known as spontaneous thought in neuroscience. In contrast, self-attention based models depend on structured patterns over their inputs to predict each next token, lacking spontaneity. Motivated by this distinction, we characterize spontaneous topic changes in self-attention architectures, revealing both their similarities and their divergences from spontaneous human thought. First, we establish theoretical results under a simplified, single-layer self-attention model with suitable conditions by defining the topic as a set of Token Priority Graphs (TPGs). Specifically, we demonstrate that (1) the model maintains the priority order of tokens related to the input topic, (2) a spontaneous topic change can occur only if lower-priority tokens outnumber all higher-priority tokens of the input topic, and (3) unlike human cognition, the longer context length or the more ambiguous input topic reduces the likelihood of spontaneous change. Second, we empirically validate that these dynamics persist in modern, state-of-the-art LLMs, underscoring a fundamental disparity between human cognition and AI behaviour in the context of spontaneous topic changes. To the best of our knowledge, no prior work has explored these questions with a focus as closely aligned to human thought.

自注意力话题切换认知对比大模型机制

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