AI硬體與軟體堆疊

關於訓練中使用的梯度檢查點(重新計算啟動值),下列說明何者正確?

  1. A只保存順向傳播中間結果的一部分,反向傳播需要時再重新計算其餘部分,以增加計算為代價來減少記憶體
  2. B在訓練途中將模型的權重存到磁碟
  3. C不計算梯度來進行訓練
查看答案與解析

正確答案:A. 只保存順向傳播中間結果的一部分,反向傳播需要時再重新計算其餘部分,以增加計算為代價來減少記憶體

反向傳播需要各層順向傳播的中間結果,但全部保存的話,記憶體會隨層數與批次大小成正比增加。在梯度檢查點中,只保存每隔數層的結果,反向傳播到該區間時,再從保存的位置重新做一次順向傳播,重建所需的值。雖然順向傳播的計算約多出一次,但可大幅減少啟動值的記憶體。

  • A. 正確。是記憶體不足時常用的方法。
  • B. 與中途保存(儲存檢查點)是不同的東西。
  • C. 梯度是要計算的。
練習這個領域 前往領域頁面

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

  1. AI加速器所使用的脈動陣列(systolic array)能有效率地計算矩陣相乘,其主要原因為下列何者?
  2. AI加速器有時會在晶片上搭載大容量的SRAM(數十~數百MB),其主要原因為下列何者?
  3. 關於AI加速器資料流方式之一的「權重固定(weight stationary)」,下列說明何者正確?
  4. 在智慧型手機等裝置內(邊緣)而非雲端伺服器進行AI推論的優點,下列何者「不」適用?
  5. 智慧型手機SoC上搭載的NPU,在AI推論上比CPU電力效率高,其主要原因為下列何者?