AI硬體與軟體堆疊

將上一題的7B模型從FP16(約14 GB)量化為4位元整數(INT4),權重的記憶體大約是多少?

  1. A約7 GB
  2. B約1.75 GB
  3. C約3.5 GB
查看答案與解析

正確答案:C. 約3.5 GB

每個參數從16位元變成4位元,因此記憶體變為四分之一,約3.5 GB(實際上還會多一些量化縮放比例等資訊)。由於記憶體變小,即使在個人電腦或智慧型手機上也能執行大型模型,而且從記憶體讀取的量減少,推論也會變快。如何減少精度的下降成為重要課題。

  • A. 這是INT8(8位元)的情形。
  • B. 這是2位元的情形。
  • C. 正確。從16位元到4位元,為四分之一。
練習這個領域 前往領域頁面

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

  1. 計算M×K矩陣與K×N矩陣的乘積時,浮點運算的次數(乘法與加法分開計算)大約是多少?
  2. 計算N×N的FP16(2位元組)矩陣相乘,兩個輸入各讀取1次,結果寫入1次。運算強度(每位元組的運算數)大約是多少?
  3. 以尖峰效能100 TFLOPS、記憶體頻寬2 TB/s的加速器執行運算強度為10 FLOP/位元組的處理。以Roofline模型估算,可達成的效能為何?
  4. 輸入只有1筆(批次大小為1)時,全連接層的計算(權重矩陣與向量的乘積)容易受記憶體限制,其原因下列何者正確?
  5. 在推論中加大批次大小(一次處理的輸入數量),整體效率容易提升,其原因為下列何者?