Tier 1: 世界のルールを不可逆に変えた原典アーキテクチャHF 141 votes
TransformerSelf-Attention深層学習基盤

Transformer:アテンション機構による言語モデルの再定義

再帰や畳み込みを排し、自己注意機構のみで並列計算と長距離文脈処理を可能にした現代AIの統一基盤

Attention Is All You Need

論文の書誌情報と関連リンク

読み方の2つの軸

現在:しくみ × 全体像

語彙 使うことば・数式・例え方

直感数式なし
しくみ基礎的な数式
原論文原論文の表現

深さ 研究のどこまで読むか

核心肝だけ
全体像背景から評価
読み解く前提から限界

この2本は、本文下のスライダーからいつでも変更できます。

概要

Transformerは、文章を一語ずつ順番に処理する仕組みを使わず、文章内の言葉同士の関係をまとめて調べる翻訳モデルです。著者らは、英語からドイツ語への翻訳で28.4点、英語からフランス語への翻訳で41.8点を報告しました。英仏モデルの学習には8台のP100を3.5日使いました。後に広く使われたことと、原論文が直接確かめた結果は分けて読む必要があります。

従来の代表的な翻訳モデルは、文章を先頭から一語ずつ処理していました。前の計算が終わるまで次へ進みにくく、学習を一斉に計算しづらい構成です。Transformerは、文章内の各言葉がほかの言葉をどれだけ参照するかをまとめて計算します。言葉の順番は別の位置情報として加えます。著者らは英独・英仏翻訳で従来法と比べ、英語の文構造を見分ける課題も試しました。後世の言語・画像モデルへの影響は大きいものの、それ自体は原論文の翻訳実験の証拠ではありません。

この論文が問うたのは、翻訳モデルに一語ずつ状態を受け渡す仕組みが本当に必要かという点です。Transformerは、入力側と出力側を、言葉同士の関係を調べる処理、各位置を個別に変換する処理、元の情報を足し戻す近道から組み立てます。順番は波の形を使った位置の目印として加えます。著者らは二つの翻訳データで精度と学習計算量を比較し、英語の文構造を見分ける課題でも評価しました。一方、文章が長くなると調べる組み合わせが二乗で増えること、位置情報を別に与える必要があること、実験の中心が翻訳であることが結論の範囲を決めます。

Transformerは、recurrent/convolutional sequence modelを使わず、self-attentionを中心にencoder–decoderを構成する手法です。系列内の全位置を学習時に並列処理し、任意の二位置間のパス長を一定にします。著者らはWMT 2014 En–Deで28.4 BLEU、En–Frで41.8 BLEUを報告し、後者を8台のP100で3.5日学習しました。

Vaswaniらは、逐次状態更新を必要とするRNN/LSTMと、受容野を層ごとに広げるCNNに代えて、self-attentionを主要な系列変換とするTransformerを提案しました。Scaled Dot-Product Attention、Multi-Head Attention、位置表現、位置ごとのfeed-forward層、残差接続とLayer Normalizationを組み合わせます。WMT 2014 En–De/En–FrではBLEUと推定学習計算量を先行法と比較し、構文解析への適用も報告しました。後続モデルでの普及は歴史的影響であり、この実験が直接示す範囲とは区別します。

Attention Is All You Needは、sequence transductionをself-attention中心のencoder–decoderとして定式化した研究です。論文は層あたりの計算量、逐次演算数、任意の二位置間の最大パス長をRNN/CNN/self-attentionで比較します。モデルはscaled dot-product attentionを複数の射影空間で計算し、位置表現とposition-wise feed-forward層を加えます。WMT 2014の翻訳、推定FLOPs、学習時間、構文解析、構成変更の比較が主な証拠です。長系列での二次計算量と、順序を位置表現へ委ねる仮定が適用範囲を制約します。

Vaswani et al. (2017) は、recurrenceとconvolutionを用いず、self-attentionを主要演算とするencoder–decoder Transformerを提案した。著者らはWMT 2014 En–Deで28.4 BLEU、En–Frで41.8 BLEUを報告し、En–Fr Bigの訓練は8基のP100で3.5日だった。原論文の直接証拠は翻訳と構文解析であり、後続の基盤モデルへの影響は別の歴史的主張である。

本論文は、sequence transductionにおける逐次計算と長距離依存のパス長を問題にし、self-attentionのみでencoder–decoderの主要な表現変換を構成するTransformerを提案する。Scaled Dot-Product Attention、Multi-Head Attention、sinusoidal positional encoding、position-wise FFN、residual connection/Post-LNを統合した。WMT 2014 En–De/En–FrでBLEU、学習時間、推定FLOPsを比較し、Penn Treebank constituency parsingも評価した。BERTやGPT等への波及は原論文の実験結果ではなく後世の影響として区別する。

Vaswani et al. (2017) は、sequence transductionをrecurrent state updateから全対全attentionへ置き換え、層あたり計算量、sequential operations、maximum path lengthの比較を設計根拠としてTransformerを提示した。Base/Big構成をWMT 2014 En–De/En–Frで評価し、BLEU、wall-clock、estimated FLOPs、構文解析、Table 3の構成変更を報告する。結論はこのデータ、tokenization、比較条件に依存する。長系列での二次計算量と外部位置表現は構造上の制約であり、後続の基盤モデルへの採用は原論文の直接評価とは分けて扱う。

著者をもっと詳しく知る(全4名)

論文と確認可能な公式情報に基づき、著者の所属と研究背景を掲載しています。

全著者と所属

  1. Ashish Vaswani
    Google Brain
  2. Noam Shazeer
    Google Brain
  3. Niki Parmar
    Google Research
  4. Jakob Uszkoreit
    Google Research

確認できた研究背景

Ashish Vaswani
所属
: Google Brain
学歴
: USC博士
主な関心
: 機械翻訳、深層学習アーキテクチャ
Noam Shazeer
所属
: Google Brain
学歴
: デューク大学
主な関心
: 大規模モデル並列化、Mixture of Experts
Niki Parmar
所属
: Google Research
学歴
: USC修士
主な関心
: 画像Transformer、効率的アテンション
Jakob Uszkoreit
所属
: Google Research
学歴
: ベルリン工科大学
主な関心
: 自然言語理解、Transformer原案提唱

なぜ歴史的イノベーションなのか

『AIの歴史で最も重要な論文』。昔のAIが言葉を1文字ずつ順に読んでいた遅い仕組みをやめ、『文章のすべての言葉同士の関係を同時に見渡す(アテンション)』という画期的な発想で、今あるChatGPTなどの基礎を作りました。

Googleの研究チームが発表。従来のRNNは順番にしか計算できず長い文章が苦手でしたが、Transformerは全単語を並列に処理する『自己注意機構』を考案。学習スピードが劇的に速くなり、AIが驚異的な賢さを獲得する原点となりました。

NeurIPS 2017発表。再帰(RNN)の完全撤廃により、計算時間O(1)の直接的な長距離依存モデリングと、GPUを限界まで回す超並列学習を実現。後のBERT、GPTシリーズ、拡散モデル、Vision Transformerに至る現代AIのすべての礎です。

Self-AttentionとPosition-wise Feed-Forward Networkのみで構成されたエンコーダ・デコーダ構造。RNNの逐次処理依存を解消し、学習効率と長距離文脈の表現力を飛躍的に向上させました。

Scaled Dot-Product AttentionとMulti-Head Attentionの定式化。Query/Key/Valueの射影空間における注意計算と残差接続・LayerNormの統合により、WMT翻訳で最高精度を更新しつつ、学習時間を数分の一に短縮しました。

Transformerの完全解説。入力シーケンスの自己注意行列 A = softmax(QK^T / sqrt(d_k))V により、パス長O(1)で大域的コンテキストを集約。Positional Encodingによる順序保持と8並列Multi-Head設計が、現代のスケーリング則の数学的土台となっています。

Attention機構のみに基づく新規配列変換モデル『Transformer』の原典。英独・英仏翻訳タスクにおいてアンサンブルを含む従来最先端を凌駕するBLEUスコアを達成。

Vaswani et al. (2017)。再帰構造の排除による並列訓練の極大化と、Multi-Head Attentionによる異種部分空間からの情報抽出を実証。8台のP100 GPUでわずか3.5日の学習により単一モデル新記録を樹立。

現代NLP・深層学習のパラダイムを決定づけた論文。自己注意の計算複雑性O(n^2・d)と再帰O(n・d^2)のトレードオフを論理的に検証し、長距離依存性の勾配伝播パスを大幅に短縮した設計理論を詳細に解説。

コミュニティの評価・歴史的インパクト(2件)
  • RNNを完全に捨て去りアテンションだけで翻訳SOTAを叩き出した衝撃が、以後の全深層学習研究の方向性を一変させた。

    原文を見る ↗
  • Scaled Dot-Product AttentionとMulti-Head Attentionの数学的・ハードウェア的親和性が天才的と評される。

    原文を見る ↗
この読み方に出てくる言葉(7語)
自己注意

同じ系列内の各トークンが、ほかのトークンとの関係を重みとして計算する仕組み。

Query・Key・Value

参照したい条件、照合される特徴、取り出す情報を分けて表す三つの表現。

マルチヘッド注意

異なる射影空間で複数の注意計算を並行し、その出力を結合する構造。

位置エンコーディング

自己注意だけでは失われる語順を、各位置の追加表現として与える方法。

BLEU

機械翻訳の出力と参照訳の一致を測る、本論文が用いた自動評価指標。

ベースライン

提案法の効果を判断するための比較対象。

アブレーション

構成要素や設定を変え、どの部分が結果へ寄与したかを調べる実験。

どんな問いに向き合ったか

逐次状態更新を除き、self-attentionを主要な系列変換として、翻訳精度と学習効率を両立できるかが研究課題です。

従来の方法と課題

LSTMやGRUをベースとしたエンコーダ・デコーダモデルに、補助的にアテンションを付加する構成(Bahdanau Attention等)。

肝のアイデア

RNN/CNNを一切使用しない、純粋な自己注意機構のみによるEncoder-Decoderアーキテクチャ

内積の次元正規化を行うScaled Dot-Product Attentionの定式化

異なる射影部分空間から複数の相関パターンを抽出するMulti-Head Attention(標準8ヘッド構成)

系列順序を維持するための周波数ベースのSinusoidal Positional Encoding

自己回帰デコード時に未来トークンへのアテンション重みをマイナス無限大にマスクするCausal Masking

どういうしくみか

Transformerの中核は、以下の数式で定義される Scaled Dot-Product Attention です:

Attention(Q,K,V)=softmax(QKTdk)V\text{Attention}(Q, K, V) = \text{softmax}\left(\frac{QK^T}{\sqrt{d_k}}\right)V

ここで、Q,KRn×dkQ, K \in \mathbb{R}^{n \times d_k}VRn×dvV \in \mathbb{R}^{n \times d_v} です。内積値をキーの次元数 dkd_k の平方根で除算(スケーリング)する理由は、dkd_k が大きくなった際に内積値のマグニチュードが極大化し、Softmax関数の勾配が極小領域(サチュレーション領域)に押し込まれるのを防ぐためです。

さらに、単一のアテンション関数を適用するのではなく、Query、Key、Valueを異なる線形変換行列 WiQ,WiK,WiVW_i^Q, W_i^K, W_i^V によって hh 個の部分空間に射影し、並列にアテンションを計算する Multi-Head Attention を導入しました:

MultiHead(Q,K,V)=Concat(head1,,headh)WO\text{MultiHead}(Q, K, V) = \text{Concat}(\text{head}_1, \dots, \text{head}_h)W^O

各層は、Multi-Head Attentionサブレイヤーと、各位置に独立適用される2層MLP(Position-wise Feed-Forward Network)で構成され、それぞれの出力に残差接続(Residual Connection)とLayer Normalization(Post-LN: LayerNorm(x+Sublayer(x))\text{LayerNorm}(x + \text{Sublayer}(x)))が適用されます。

どう確かめたか

WMT 2014 En–De/En–FrでBLEUを評価し、single-modelおよびensemble baselineと比較した。訓練コストはP100 GPU上のwall-clockとestimated FLOPsで比較し、Penn Treebank constituency parsingを追加評価した。

何が分かったか

WMT 2014英独タスクにおいて28.4 BLEUを記録。先行研究の最高アンサンブルモデル(26.3 BLEU)を単一モデルで2.1ポイント更新

WMT 2014英仏タスクにおいて41.8 BLEUを達成。当時の文献におけるあらゆる単一モデルのSOTAを更新

英仏Bigモデルの訓練リソースは8台のNVIDIA P100 GPUで3.5日(約1.8×10^19 FLOPs)。先行手法(MoE: 1.5×10^20 FLOPs、GNMT: 2.5×10^20 FLOPs)と比較して訓練コストを約1/10に削減

英語成分構文解析(Penn Treebank WSJ 23)で41.3 F1を達成し、翻訳以外のドメインへの転移性を確認

どこまで使えるか

直接の根拠は主に機械翻訳と一つの構文解析課題です。後の大規模言語モデルへの採用は歴史的重要性を示しますが、2017年の実験が後世の全能力を検証したわけではありません。

限界と未解決の問い

Self-attentionは系列長 (N) に対して計算・attention matrixの記憶量が (O(N^2)) となる。演算自体は順序を持たないためpositional encodingが必要である。評価の中心はWMT 2014 En–De/En–Frであり、構文解析以外の領域や大規模事前学習への外的妥当性は原論文から直接は決まらない。デコーダの生成は自己回帰的で、学習時の並列性が生成時にそのまま得られるわけでもない。

この研究から考える

ここからは、論文の結果を踏まえた編集上の考察です。

翻訳精度と著者らが報告した学習計算量を合わせると、系列モデルの設計では逐次状態を持つことより、並列化可能性と位置間パス長を優先する判断が成り立ちます。この判断を他領域へ広げるには、系列長の二次コスト、位置表現、自己回帰デコードを含む条件ごとの評価が必要です。