言語モデルの操作文に残るKVキャッシュの記録を調べる
Written as a Record, Read as an Address: What a Forward Pass Leaves in an Operation's KV Cache
この論文をやさしく読む
ひとことで言うと
言語モデルが交換などの操作文を読んだとき、その内部キャッシュに、どこを参照するかだけでなく、読んだ値も残るかを調べた研究です。読み手だけを訓練すると、もともとのキャッシュから値を取り出せました。
何に役立つ?
モデル内部に情報が存在することと、元のモデルがその情報を実際に利用することを区別するための実験手法になります。言語モデルの状態追跡や内部表現の理解に役立ちます。
この研究の面白いところ
値が同じ世界同士でもキャッシュの移植で参照先が変わり、経路付けの働きを因果的に調べています。値への直接アクセスも、同じオペランド名トークンの限られた層に局在していました。
どこまで分かった?
主な数値は合成的な箱の課題の復元率です。読み手の訓練にはカバー範囲の制約があり、オープンブック条件の精度低下も報告されています。元のモデルが常に値を直接利用しているという結果ではありません。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
言語モデルが「箱Fと箱Bの中身を交換する」といった操作を読むと、順伝播により、そのトークンのキーと値がKVキャッシュに書き込まれる。実体追跡に関する先行研究は、モデルが何を利用するかを明らかにしている。対応関係は明示的な潜在状態として保存されるのではなく、問い合わせ時に解決される。本研究では、モデルが操作を記述する区間に何を書き込み、それにどうアクセスするかを問う。 順伝播を、固定した書き手と読み手に分ける。書き手のキャッシュは勾配計算なしで再計算し、読み手には指示と操作のトークンだけを見せ、状態の記述をすべて隠して、読み手だけを学習させる。したがって、読み手が復元できた情報は、変更を加えていないキャッシュにすでに存在していたことになる。合成的な箱の課題では、問い合わせた対応関係を基礎の読み手が復元できる割合は0.06以下だったのに対し、学習後は0.75~1.00となった。また、復元可能性は操作の読み書きの対象範囲に対応した。 情報へのアクセスには2つの形が見つかった。Llama-3.1-8BとMistral-7Bでは、操作区間を移植すると、2つの世界で保持する値が同じでも、可視のどの状態を読むかが因果的に変化し、参照先を選ぶ経路付けの記録があることが分かった。読み手だけの学習は、この経路付けを保ちながら、操作が読んだ値であるペイロードへの直接アクセスを加える。この値は、オペランド名を表す単一トークンから、中間深度の狭い層範囲、すなわちLlama-3.1-8Bの全32層中12~15層、Mistral-7Bの14~17層で取り出せる。これは経路付けの記録がある場所と同じである。 同じ手順は別の操作やToMi、GSM8Kにも拡張できるが、学習でカバーした範囲に制約され、必要な情報を参照できるオープンブック条件での正答率を犠牲にする。したがって、操作トークンには、局所化され、因果的に取り出せる記録が残り、経路付けとペイロードへの直接アクセスの両方を支える。ただし、それを書き込むモデル自身は、主に記録の値ではなく、そこに含まれる参照先を読んでいる。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-21(UTC)
- 最新改訂
- 2026-09-21 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-21 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
When a language model reads an operation such as "Swap the contents of Box F and Box B", its forward pass writes keys and values for those tokens into the KV cache. Prior work on entity tracking establishes what models use: bindings are resolved at query time rather than stored as explicit latent state. We ask what they write at the operation span and how it is accessed. We split a forward pass into a frozen writer and a reader: the writer's cache is recomputed without gradients, while the reader sees only the instruction and operation tokens, with all state descriptions hidden, and is trained in isolation. Anything the reader recovers was therefore already present in the unmodified cache. On a synthetic boxes task, a base reader recovers $\leq 0.06$ of queried bindings against $0.75$--$1.00$ after training, and recoverability tracks the operation's read/write footprint. We find two modes of access. Across Llama-3.1-8B and Mistral-7B, operation-span transplants causally redirect which visible state is read even when the two worlds hold identical values, revealing a routing record. Isolation training preserves routing and adds direct access to the payload, the value the operation read, from the single operand-name token in a narrow mid-depth band (layers 12--15 of 32 in Llama-3.1-8B, 14--17 in Mistral-7B) --- the same site that holds the routing record. The same recipe extends to further operations, ToMi and GSM8K, but is bounded by training coverage and costs open-book accuracy. Operation tokens thus leave localized, causally recoverable records that support both routing and direct payload access, though the model that writes them reads mainly the address they carry and not the value.
arXiv ID: 2609.24635 / 要約の誤りについて