AI硬體與軟體堆疊

Transformer的自注意力之計算量,會如何隨輸入序列長度n增加?

  1. A與n成正比增加
  2. B與n的平方成正比增加
  3. C與n無關,維持固定
查看答案與解析

正確答案:B. 與n的平方成正比增加

在自注意力中,要計算每個token的Query與所有token的Key之間的內積,建立n×n的權重表。因此計算量,以及按直覺實作時中間表所需的記憶體,都與序列長度的平方成正比。長度變為2倍時,注意力的部分會變為4倍,所以處理的文章越長,注意力的效率就越左右整體效能。

  • A. 全連接層等部分與n成正比,但注意力是平方。
  • B. 正確。因為要比較所有token的組合。
  • C. 會隨長度增加。
練習這個領域 前往領域頁面

AI硬體與軟體堆疊的其他題目

  1. FlashAttention等高速注意力實作的主要概念,下列何者正確?
  2. AI計算中使用的「運算融合(運算子融合)」,其主要目的為下列何者?
  3. 將矩陣相乘的結果加上偏差、再通過ReLU的處理,從3個各別的kernel融合為1個。主要減少的是下列何者?
  4. XLA、TVM、PyTorch的編譯功能這類AI編譯器所進行的主要處理,下列何者正確?
  5. AI框架或編譯器中使用的「計算圖」,下列說明何者正確?