AI硬體與軟體堆疊

在大型語言模型的推論中處理長文章或大批次時,容易成為問題的是下列何者?

  1. A權重的數量隨文章長度成正比增加
  2. BKV快取的記憶體隨文章長度與批次大小成正比增加,有時會比權重還大
  3. Ctoken數量增加時,模型的回答必定變短
查看答案與解析

正確答案:B. KV快取的記憶體隨文章長度與批次大小成正比增加,有時會比權重還大

KV快取的大小,由層數×Key與Value的維度×token數×批次大小×每個元素的位元組數決定。在數萬token的長文章或大量同時請求的情況下,這可能達到數十GB,甚至超過權重的記憶體。因此會採用以較低的精度持有KV快取、由多個注意力頭共用Key與Value、以細小的區塊管理記憶體等做法。

  • A. 權重的數量是固定的。
  • B. 正確。會採用KV快取的壓縮與量化等做法。
  • C. 與回答的長度無關。
練習這個領域 前往領域頁面

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

  1. 以記憶體頻寬2 TB/s的GPU,在批次大小為1的情況下,對權重為FP16約14 GB的模型進行解碼。產生1個token所需時間的下限最接近下列何者?
  2. Transformer的自注意力之計算量,會如何隨輸入序列長度n增加?
  3. FlashAttention等高速注意力實作的主要概念,下列何者正確?
  4. AI計算中使用的「運算融合(運算子融合)」,其主要目的為下列何者?
  5. 將矩陣相乘的結果加上偏差、再通過ReLU的處理,從3個各別的kernel融合為1個。主要減少的是下列何者?