AI硬體與軟體堆疊

大型語言模型的推論分為集中處理輸入文章的階段(預填充)與逐一產生一個token輸出的階段(解碼)。下列何者是分別主要決定兩者效能之因素的正確組合(批次較小的情形)?

  1. A預填充是記憶體頻寬,解碼是運算效能
  2. B預填充是運算效能,解碼是記憶體頻寬
  3. C兩者都是儲存裝置的速度
查看答案與解析

正確答案:B. 預填充是運算效能,解碼是記憶體頻寬

預填充會將輸入的數百至數千個token集中計算,因此權重可由許多token重複使用,成為矩陣與矩陣的乘積,受運算限制。解碼每產生一個token,就要讀取所有權重,卻只做一個token份的計算,因此運算強度低,受記憶體限制。所以解碼的速度大致由記憶體頻寬÷權重大小決定。

  • A. 正好相反。
  • B. 正確。解碼每個token都要讀取全部的權重。
  • C. 推論時權重是放在記憶體中。
練習這個領域 前往領域頁面

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

  1. 大型語言模型推論中使用的KV快取,其作用下列何者正確?
  2. 在大型語言模型的推論中處理長文章或大批次時,容易成為問題的是下列何者?
  3. 以記憶體頻寬2 TB/s的GPU,在批次大小為1的情況下,對權重為FP16約14 GB的模型進行解碼。產生1個token所需時間的下限最接近下列何者?
  4. Transformer的自注意力之計算量,會如何隨輸入序列長度n增加?
  5. FlashAttention等高速注意力實作的主要概念,下列何者正確?