日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

週刊ニュースレター購読
解釈可能性arXiv:2602.23164

MetaOthello:トランスフォーマーにおける複数世界モデルの制御された研究

MetaOthello: A Controlled Study of Multiple World Models in Transformers

シェア:XThreadsFacebookLINEはてブBluesky

オセロのルールやトークン化を変えた複数変種を混ぜて学習させ、単一のトランスフォーマーが複数の世界モデルを共有表現空間でどう組織するかを調べた。

著者: Aviral Chawla, Galen Hall, Juniper Lovato

分類: cs.LG

原文アブストラクト

Foundation models must handle multiple generative processes, yet mechanistic interpretability largely studies capabilities in isolation; it remains unclear how a single transformer organizes multiple, potentially conflicting "world models". Previous experiments on Othello playing neural-networks test world-model learning but focus on a single game with a single set of rules. We introduce MetaOthello, a controlled suite of Othello variants with shared syntax but different rules or tokenizations, and train small GPTs on mixed-variant data to study how multiple world models are organized in a shared representation space. We find that transformers trained on mixed-game data do not partition their capacity into isolated sub-models; instead, they converge on a mostly shared board-state representation that transfers causally across variants. Linear probes trained on one variant can intervene on another's internal state with effectiveness approaching that of matched probes. For isomorphic games with token remapping, representations are equivalent up to a single orthogonal rotation that generalizes across layers. When rules partially overlap, early layers maintain game-agnostic representations while a middle layer identifies game identity, and later layers specialize. MetaOthello offers a path toward understanding not just whether transformers learn world models, but how they organize many at once.

関連論文