大規模言語モデルにおける認知マップ
Cognitive Map in Large Language Models
言語モデルが関係推論課題と内部活性化を操作する実験を通じ、動物の認知マップに似た関係構造の抽象表現を形成し、実体を役割に結びつけて推論する仕組みを明らかにした。
著者: Albert Qin, Jonathan Cohen, Gautam Reddy
分類: cs.CL, cs.LG
原文アブストラクト
It has long been proposed that animals construct cognitive maps to support flexible behavior in novel environments. We tested whether large language models form similar representations using relational reasoning tasks and causal "mosaic experiments" that manipulate internal activations in open-weight language models. Models beyond the ten-billion-parameter scale successfully inferred unstated relationships between entities and generalized to synthetic entity names. Across model and task families, models generated abstract representations of relational structure and combined them additively with representations of entity identity, binding entities to roles within that structure. Models stored observed relationships as pairwise links that they composed retrospectively when queried, rather than prospectively positioning each entity in a global coordinate system. Binding was mediated by a sparse, distributed set of key--value pairs; by perturbing these pairs, we implanted "false memories" that altered model behavior in predictable ways. Overall, our findings reveal an internal organization analogous to a cognitive map that explains how language models reason with limited information.