本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

コンテキスト対応トランスフォーマー:新たなリカレントニューラルネットワークアーキテクチャ

記事の要約

本論文は、D層のトランスフォーマーブロックに入る前に各トークンを事前にコンテキスト化するコンテキスト対応トランスフォーマーを提案する。修正ネットワークが前位置のブロック出力と現在のトークン埋め込みを組み合わせ、逐次推論時にはリカレントニューラルネットワークとして機能する。訓練では修正プロセスをK回展開し、各ステップで全位置を並列処理する。実験では、D=5モデルが12層トランスフォーマーを凌駕し、A100で1.7倍高速な生成を実現。単一層モデル(D=1)はK=10で6層トランスフォーマーを上回り、推論速度は2.6倍。このアーキテクチャは広い表現と長いコンテキストで最も効果を発揮し、ポインタチェイシングタスクではD=1が10段階すべてを解決する。

ソースarXiv Computational Linguistics著者: Mahesh Godavarti
コンテキスト対応トランスフォーマー:新たなリカレントニューラルネットワークアーキテクチャ
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

本論文「The Context-Ready Transformer」は、Mahesh Godavarti氏によって2026年6月25日にarXivに投稿され、NeurIPSに採択された研究である。提案するアーキテクチャは、D層のトランスフォーマーブロックの前に各トークンを事前にコンテキスト化する修正ネットワークを組み込んだ、新しいリカレントニューラルネットワーク(RNN)である。

従来のトランスフォーマーでは、トークンは生の埋め込みとしてブロックに入力され、文脈情報は自己注意機構によって内部で処理される。これに対し、コンテキスト対応トランスフォーマーでは、左から右への生成時に、修正ネットワークが前位置のブロック出力(過去のコンテキストのキャッシュされた要約)と現在のトークン埋め込みを組み合わせる。これにより、トークンはすでにコンテキスト化された状態でブロックに入る。逐次推論時には、この修正チェーンがアーキテクチャ全体をリカレントニューラルネットワークへと変える。

訓練時には、修正プロセスをシーケンス全体にわたってK回展開し、各ステップで全位置を並列に処理する。これにより、RNNの逐次性を回避しつつ、効率的な学習が可能となる。また、事前学習済みのトランスフォーマーにゼロ初期化された修正フィードフォワードネットワーク(FFN)を追加し、ファインチューニングすることで、コンテキスト対応モデルに変換できる。

評価は、幅、深さ、ブロックサイズ、および2つのデータセットにわたって行われ、標準トランスフォーマー、その変種、アブレーションとの比較がなされた。特筆すべき結果として、D=5モデルは12層トランスフォーマーを性能で上回り、A100 GPU上での生成速度は1.7倍高速である。さらに、K=10の場合、単一層モデル(D=1)が6層トランスフォーマーを上回り、推論速度は2.6倍に達する。逐次推論は並列K=10と0.01 PPL以内で一致する。

このアーキテクチャは、広い表現と長いコンテキストから最も大きな利益を得る。ポインタチェイシングタスクでは、BPTTで訓練されたD=1モデルが10段階の合成レベルすべてを解決する一方、標準トランスフォーマーは階段状の深さ依存性を示す。これは、コンテキスト対応トランスフォーマーが本質的にリカレントな構造を持つため、深さに依存せずに合成を扱えることを示唆している。

本研究は、トランスフォーマーの並列性を維持しつつ、リカレントな逐次推論を可能にする新しい方向性を提示しており、今後の効率的な言語モデル設計に影響を与える可能性がある。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • トークンをトランスフォーマーブロックに入る前に事前コンテキスト化し、逐次推論時にリカレントニューラルネットワークを形成
  • 訓練時は修正プロセスをK回展開し、各ステップで全位置を並列処理。事前学習済みトランスフォーマーからの変換も可能
  • D=5モデルは12層トランスフォーマーを性能で上回り、生成速度は1.7倍。D=1モデルはK=10で6層トランスフォーマーを上回り、推論速度2.6倍
  • 広い表現と長いコンテキストで最も効果を発揮。ポインタチェイシングタスクでD=1が全10段階を解決し、標準トランスフォーマーは階段状の深さ依存を示す

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。