AI硬體與軟體堆疊

在推薦系統的模型等所使用的大型嵌入表(embedding table)之查表,難以用AI硬體處理,其原因下列何者正確?

  1. A因為查表需要非常多的乘法
  2. B因為嵌入表很小,必定能放進晶片上
  3. C因為幾乎沒有計算,而是讀取巨大表中零散的位置,所以無法發揮矩陣運算單元,速度由記憶體的容量、頻寬與延遲決定
查看答案與解析

正確答案:C. 因為幾乎沒有計算,而是讀取巨大表中零散的位置,所以無法發揮矩陣運算單元,速度由記憶體的容量、頻寬與延遲決定

在推薦系統中,為使用者或商品等的每個ID持有向量的表可達數百GB,每筆輸入都要讀取表中零散的列再加總。幾乎沒有計算,讀取又不規則,合併存取與快取都難以發揮作用,與擅長矩陣相乘的加速器所擅長的形態大不相同。必須將表分散放在多顆GPU或CPU的記憶體上,設法確保記憶體的容量與頻寬。

  • A. 查表以讀取為主,乘法很少。
  • B. 可達數百GB,無法放進晶片。
  • C. 正確。關鍵在於記憶體的機制,而不是運算單元。
練習這個領域 前往領域頁面

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

  1. 將AI的軟體堆疊從貼近使用者的上層,依序排列到貼近硬體的下層,下列何者正確?
  2. 關於神經網路的「訓練」與「推論」之差異,下列何者正確?
  3. 下列何者「不是」訓練所需記憶體遠大於同一模型推論的原因?
  4. 以FP32(4位元組)持有權重、梯度與Adam的兩個狀態,來訓練參數量70億(7B)的模型。排除啟動值後,所需記憶體最接近下列何者?
  5. 若以FP16(2位元組)持有參數量70億(7B)的模型權重,權重的記憶體大約是多少?