日本フィジカルAI新聞

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

週刊ニュースレター購読
解釈性arXiv:2609.15975

Transformerにおける表現進化の方向分解による分離

Disentangling Representation Evolution in Transformers through Directional Decomposition

シェア:XThreadsFacebookLINEはてブBluesky

Transformerの表現更新を平行・垂直成分に分解し、編集ロバスト性や圧縮誤差、事前学習への介入効果を解析した研究。

詳しい要約

1. どんなもの?

- Transformer の表現が学習された加法的変換によってどのように進化するかを、方向の保存と変更という機能的幾何として研究する。 - 学習された更新を parallel 成分と perpendicular 成分に分解する。 - 事前学習済みモデルで residual identity path を超える大きな parallel 成分を見出し、attention/MLP 更新と value 集約の2空間に適用する。

2. 先行研究と比べてどこがすごい?

- 従来の残差経路中心の見方に対し、更新を parallel/perpendicular に分解して表現進化を成分レベルで捉える点が新しい。 - 編集の頑健性が空間依存的に非対称であることを示し、exclude-self value-space の parallel 操作が residual-space や perpendicular より顕著に頑健である。 - 圧縮誤差の診断で perpendicular 誤差が parallel 誤差より圧縮手法を明確に分離できることを示す。

3. 技術・手法の肝は?

- 学習された更新を現在の方向に平行な成分と垂直な成分に分解する。 - 2つの空間で適用: hidden state に対する attention/MLP 更新、現在トークンの value に対する attention value 集約。 - 編集では exclude-self value-space の parallel 操作により、self message を保ちつつ non-self aggregate のみをスケールする。

4. どうやって有効だと検証した?

- 事前学習済みモデルで residual identity path を超える parallel 成分を観測。 - ターゲット編集により空間依存的な非対称性を確認。 - 圧縮 induced update error を成分分解し、perpendicular 誤差が手法分離に有効と示す。 - from-scratch pretraining 中の full-aggregate parallel suppression が validation-loss 軌跡を下げ、downstream 平均を改善し、value-space 変種が最強。

5. 議論はある?

- 表現幾何が編集頑健性、圧縮診断、学習時介入を結びつけることを示唆。 - ただし具体的な限界や反例、理論的説明の深さは要旨からは不明。

6. 次に読むべき論文は?

- 要旨で参照/比較されている研究は明示されていない。 - 関連手法として Transformer の residual stream 解析、attention value 集約、モデル圧縮、pretraining 介入の定番研究を挙げる。

※ AIが要旨から生成した要約です。正確性は原文をご確認ください。

著者: Shwai He, Haichao Zhang, Shen Yan

分類: cs.CL, cs.LG

原文アブストラクト

Transformer representations evolve through learned additive transformations that either preserve their current direction or redirect it. We study this evolution as a functional geometry, decomposing learned updates into parallel and perpendicular components. Across pretrained models, we find substantial parallel components beyond the residual identity path. We then apply the decomposition in two spaces: to attention and MLP updates relative to the hidden state, and to attention value aggregation relative to the current token's value. Targeted edits reveal a strongly space-dependent asymmetry: exclude-self value-space parallel manipulation is markedly more robust than residual-space and perpendicular counterparts, preserving the direct self message while scaling only the non-self aggregate. The same decomposition gives a component-resolved description of compression-induced update error: perpendicular error separates compression methods more clearly than parallel error. Extensive experiments further demonstrate that full-aggregate parallel suppression during from-scratch pretraining lowers validation-loss trajectories and improves downstream averages, with the value-space variant strongest. Together, these results connect representation geometry to editing robustness, compression diagnosis, and training-time intervention. Code is available in the \href{https://github.com/Shwai-He/Transformer-Geometry}{project repository}.