スパース化とは 使わない計算を省く方法、MoEは6,710億の重みのうち370億だけ動かす
スパース化は、AIのモデルのうち値がゼロの部分や使わない部分を計算から外す方法です。エヌビディアのA100は、4つの値のうち2つをゼロにする形で、16ビットの計算性能を毎秒312兆回から624兆回に倍増させました。
- スパース化は、ゼロや使わない部分の計算を省いて、AIを速く軽くする方法です。
- 重みの一部をゼロにする方法と、問いごとにモデルの一部だけを動かす方法があります。
- エヌビディアはA100から、4つの値のうち2つをゼロにすると計算が2倍速くなる回路を載せました。
- DeepSeek-V3は6,710億個の重みを持ち、1語ごとに動かすのは370億個です(MoE)。
スパース(sparse)は「まばら」という意味です。AIのモデルの重みには、ゼロに近く答えにほとんど影響しないものが多く含まれます。その部分を計算から外せば、時間も電力も減らせます。これがスパース化です。
2つのやり方
1つ目は、重みの一部をゼロにする「枝刈り」です。小さな重みを選んでゼロにし、少し学習し直して精度を戻します。ただ、ゼロがばらばらに散らばっていると、半導体はどこを飛ばせばよいかを毎回調べる必要があり、思ったほど速くなりません。
そこでエヌビディアは、ゼロの並び方に決まりを設けました。連続する4つの値のうち、必ず2つをゼロにする「2対4」の形です。GPUは、ゼロでない値とその位置の目印だけを保存し、半分の計算で同じ結果を出します。
2つ目は、問いごとにモデルの一部だけを動かす方法です。代表が専門家混合型(MoE)のモデルです。中に多数の「専門家」の小さな部品を持ち、1語ごとに数個だけを選んで計算します。重みの総数は大きくても、1回の計算量は小さくなります。
A100で始まった「2対4」
エヌビディアの2021年7月20日の技術解説によると、A100の16ビットの計算性能は、ゼロを使わない場合で毎秒312兆回、2対4の形で624兆回です。精度もほぼ保たれました。
| モデル・課題 | 指標 | 元のモデル | スパース化後 |
|---|---|---|---|
| ResNet-50(画像分類) | 正解率 | 76.1 | 76.2 |
| ResNeXt-101(画像分類) | 正解率 | 79.3 | 79.3 |
| Transformer(英独翻訳) | BLEU | 28.2 | 28.5 |
| BERT-Large(読解) | F1 | 91.9 | 91.9 |
エヌビディアのモデルを軽くする道具「Model Optimizer」は、量子化や枝刈りと並んで、スパース化を扱います。推論用の道具TensorRTで、スパース化したモデルを速く動かせます。
MoEは「動かす部分」を絞る
中国のDeepSeek(深度求索)が公開したDeepSeek-V3は、MoEの代表例です。公開の資料によると、重みの総数は6,710億個で、1語ごとに動かすのは370億個です。全体の約5.5%にあたります(計算)。学習には14.8兆語分のデータを使い、エヌビディアのH800で延べ278.8万GPU時間かかりました。
MoEでは、計算は減っても、すべての専門家の重みをメモリに置いておく必要があります。メモリの量はモデル全体の大きさで決まり、計算の量は動かす部分で決まる、というずれが生まれます。多数の専門家を複数のGPUに分けて置く「専門家並列」という分担も使われます。
関連する用語
- 枝刈り:小さな重みをゼロにしてモデルを軽くすること。
- MoE(専門家混合):多数の部品から一部だけを選んで動かすモデルの形。
- 量子化:数値の桁を減らして軽くする方法。スパース化と組み合わせられる。
エヌビディアの製品ページは、GB300 NVL72で推論ソフトDynamoを使うと、MoEのモデルの処理量がHopper世代の最大50倍になるとしています。
この記事は公開資料をもとに作成しています。売買をすすめるものではありません。誤りを見つけたら訂正します(編集方針)。
関連する記事
量子化とは 学習後か学習中か、グーグルのGemma 3は重みを54GBから14.1GBに
量子化は、AIのモデルが使う数値を16ビットから8ビットや4ビットに減らし、メモリと計算を軽くする方法です。グーグルは2025年4月、学習の段階から4ビットに慣らしたGemma 3 27Bを公開し、重みの大きさを54GBから14.1GBにしました。
メモリ帯域とは 端子の数×速さで決まる、HBM4は端子を2,048本に倍増
メモリ帯域は、メモリと計算の回路の間で1秒間に運べるデータの量です。エヌビディアのGPU1基の帯域は、H100の毎秒3.35テラバイトから、Rubinの毎秒22テラバイトに広がります。
モデル並列とは 重みの行列を割ってGPUで分担、Megatronは83億の重みを512基で
モデル並列は、1つのAIのモデルを複数のGPUに切り分けて置き、協力して計算する方法です。エヌビディアの研究者は2019年、83億個の重みのモデルを512基のGPUで学習し、1基の場合に比べ76%の効率を保ったと報告しました。
建設のリードタイムとは TSMC熊本は着工から開所まで約1年10か月、アリゾナは発表から量産まで約4年半
建設のリードタイムとは、工場やデータセンターの着工から稼働までにかかる時間です。TSMCの熊本の第1工場は2022年4月に着工し、2024年2月に開所式を開きました。