Attention Is All You Need

《Attention Is All You Need》是一篇於2017年6月12日由八位谷歌員工提交至arXiv的學術論文。該論文提出了一種名為Transformer的全新神經網路架構,其核心是完全基於自注意力機制,取代了傳統的循環神經網路和長短期記憶網路,並實現了高效的並行計算。這一架構成為後續ChatGPT、Dall-E等AI產品的核心技術基礎。截至2025年6月,該論文的被引次數已超過18萬次。論文標題“Attention Is All You Need”由利昂·瓊斯提出,其靈感來源於披頭士樂隊的歌曲“All You Need Is Love”。

論文的創作始於雅各布·烏斯克爾特關於自注意力機制的想法,後經多位研究者合作推進。諾姆·薩澤爾的加入顯著提升了系統性能。團隊在2017年5月19日的截止日期前提交論文至神經信息處理系統大會。該論文最終被會議接收,但最初僅被安排在晚間海報展示環節,並未獲得口頭報告機會。此後,論文的八位作者已全部離開了谷歌。隨著作者離職,谷歌也刪除了論文中所有作者的信箱。

技術架構,角色介紹,

技術架構

Transformer是一種完全基於注意力機制的全新神經網路架構。該架構旨在解決傳統循環神經網路因時序依賴導致的並行計算瓶頸,實現了高效的並行計算。
Transformer模型採用編碼器-解碼器架構,編碼器和解碼器均由多個相同的層堆疊而成,在原論文中均為6層。每個編碼器層包含一個多頭自注意力子層和一個位置式前饋神經網路子層,每個子層周圍都使用了殘差連線和層歸一化。解碼器層在結構上類似,但包含三個子層,在自注意力和前饋網路層之外,增加了一個編碼器-解碼器注意力層。
自注意力機制使模型能夠直接計算序列中任意位置間的依賴關係,從而有效建模長距離依賴。其核心計算公式為縮放點積注意力:Attention(Q, K, V) = softmax(QK^T/√d_k)V。該機制摒棄了RNN的時序依賴,使得所有位置可以同時參與計算,實現了高效的並行計算。
多頭注意力機制將輸入投影到多個子空間並行計算注意力,以增強模型的表示能力。其計算公式為:MultiHead(Q, K, V) = Concat(head1, ..., head_h)W^O。在Transformer中,編碼器使用自注意力,解碼器使用掩碼自注意力,並在編碼器與解碼器之間使用編碼器-解碼器注意力。
由於模型本身不具備處理序列順序的能力,因此引入了位置編碼。原始Transformer採用正弦函式生成固定位置編碼:PE(pos, 2i) = sin(pos/10000^(2i/d_model)), PE(pos, 2i+1) = cos(pos/10000^(2i/d_model))。每個編碼器和解碼器層中都包含一個位置式前饋神經網路。編碼器和解碼器的每個子層周圍都使用了殘差連線和層歸一化操作以穩定訓練。
該架構的優勢在於其並行計算能力、對長距離依賴的有效建模以及良好的可擴展性。

角色演員介紹

論文《Attention Is All You Need》共有八位作者,均來自谷歌大腦團隊。論文在腳註中聲明“各人的貢獻均等”,“作者排序按隨機排列”。
雅各布·烏斯克爾特提出用自注意力機制替代循環神經網路(RNN)的構想,並推動早期嘗試。
阿什什·瓦斯瓦尼與伊利亞·波洛蘇欣共同設計並實現了首批Transformer模型,深度參與了架構的各方面工作。
諾姆·薩澤爾提出了縮放點積注意力、多頭注意力和無參數位置表示等關鍵概念,並深度參與了幾乎所有細節的實現。
尼基·帕爾瑪在原始代碼庫和Tensor2Tensor框架中設計、實現、調優和評估了無數模型變體。
利昂·瓊斯負責初始代碼庫、高效的推理實現和模型可視化,並嘗試了新的模型變體。
烏卡什·凱撒與艾丹·N·戈麥斯花費大量時間設計和實現Tensor2Tensor庫的各個部分,替換了早期代碼庫,極大地改善了結果並加速了研究進度。
論文發表後,據2022年的報導,八位作者大多已離開谷歌,投身於新的創業或研究項目,例如共同創立Adept、Character.AI、Inceptive、Cohere等公司,或加入OpenAI等機構,僅利昂·瓊斯一人仍留在谷歌工作。隨著作者全部離職,谷歌更新了論文版本,刪除了原論文中所有作者的谷歌信箱。

相關詞條

熱門詞條

聯絡我們