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

1パラメータが16ビットから4ビットになるので、メモリは4分の1の約3.5 GBです(実際には量子化のスケールなどの情報が少し加わります)。メモリが小さくなるので、パソコンやスマートフォンでも大きなモデルを動かせるようになり、またメモリから読み込む量が減るので推論も速くなります。精度の低下を小さくする工夫が重要になります。

  • A. これはINT8(8ビット)の場合です。
  • B. これは2ビットの場合です。
  • C. 正しい。16ビットから4ビットで4分の1です。
この分野を解く 分野ページへ

AIハードウェアとソフトウェアスタックのほかの問題

  1. M×Kの行列とK×Nの行列の積を計算するときの浮動小数点の演算回数(掛け算と足し算を別々に数える)はおよそいくらか。
  2. N×NのFP16(2バイト)の行列どうしの積で、2つの入力を1回ずつ読み、結果を1回書くとする。演算強度(1バイトあたりの演算数)はおよそいくらか。
  3. ピーク性能100 TFLOPS、メモリ帯域2 TB/sのアクセラレータで、演算強度が10 FLOP/バイトの処理を実行する。Rooflineモデルで見積もった達成できる性能はいくらか。
  4. 入力が1つだけ(バッチサイズ1)のとき、全結合層の計算(重みの行列とベクトルの積)がメモリ律速になりやすい理由として正しいものはどれか。
  5. 推論でバッチサイズ(まとめて処理する入力の数)を大きくすると、処理全体の効率が上がりやすい理由はどれか。