メモリ容量の壁とは 4,050億の重みは16ビットで810GB、H100の10倍超
メモリ容量の壁は、AIのモデルの大きさに半導体のメモリの量が追いつかない問題です。メタのLlama 3.1 405Bは16ビットで約810GBとなり、H100の80GBの10倍を超えます(計算)。
- モデルの重みと、答えを作る途中の結果(KVキャッシュ)の両方が、GPUのメモリに収まる必要があります。
- 4,050億個の重みは16ビットで約810GBになり、H100の80GBでは11基以上が要ります(計算)。
- GPU1基のメモリは、H100の80GBからRubinの最大288GBへ増えましたが、モデルと会話の長さも伸びています。
- 対処は、複数のGPUへの分割、数値の桁の削減、安いメモリや記憶装置への退避の3つです。
AIのモデルを動かすには、重みをすべてメモリに置く必要があります。さらに、言語モデルは会話の途中の結果(KVキャッシュ)もメモリに残します。合わせた大きさが、半導体のメモリに入りきらない。これがメモリ容量の壁です。
何がメモリを使うのか
メモリの使い道は大きく3つです。1つ目はモデルの重みです。重みの数に、1つあたりのバイト数を掛けた大きさになります。2つ目はKVキャッシュで、入力の文章が長いほど、同時に相手をする利用者が多いほど増えます。3つ目は学習の時だけに要るもので、勾配や、最適化の計算に使う値です。学習では重みの数倍の領域が必要になります。
| 項目 | 大きさ |
|---|---|
| Llama 3.1 405B(16ビット) | 約810GB |
| Llama 3.1 405B(8ビット) | 約405GB |
| DeepSeek-V3(6,710億個、8ビット) | 約671GB |
| H100のメモリ | 80GB |
| H200のメモリ | 141GB |
| Rubinのメモリ(最大) | 288GB |
| GB300 NVL72(72基の合計) | 約20TB |
GPUのメモリはエヌビディアの製品ページと技術ブログから。モデルの大きさは、重みの数×1つあたりのバイト数で計算。
DeepSeek-V3は1語ごとに370億個の重みしか動かさない専門家混合型(MoE)ですが、メモリには6,710億個すべてを置く必要があります。計算が減っても、容量の問題は残ります。
3つの対処
1つ目は分割です。モデルを複数のGPUに分けて置き、GPU同士を速い道でつなぎます。エヌビディアのGB200 NVL72は72基のGPUを毎秒130テラバイトの道でつなぎ、1つの大きなGPUのように扱えるようにしました。
2つ目は数値の桁を減らすことです。16ビットを4ビットにすれば、重みの大きさはおおむね4分の1になります。エヌビディアの4ビット形式NVFP4は、目盛りを含めてFP16の約3.5分の1の大きさです。
3つ目は退避です。使っていないKVキャッシュを、GPUのメモリから安いメモリやSSDに移し、必要な時に戻します。エヌビディアの推論ソフトDynamoは、この仕組みを持っています。
メモリの種類を使い分ける動き
エヌビディアは2025年9月、長い文章の読み込みに特化したGPU「Rubin CPX」を発表しました。高価なHBMではなく、128GBのGDDR7を積みます。100万語規模の入力を扱う用途向けで、2026年末に出す予定です。これを組み込んだラック「Vera Rubin NVL144 CPX」は、速いメモリを100TB持ち、ラック全体のメモリ帯域は毎秒1.7ペタバイトとしています。
メモリの量そのものを増やす開発も進みます。SKハイニックスはHBM4の開発を2025年9月に終えたと発表しました。Rubinは1基に最大288GBのHBM4を積みます。H100の80GBの3.6倍です(計算)。
関連する用語
- KVキャッシュ:言語モデルが答えを作る途中の結果を残したもの。
- メモリ帯域:1秒間に運べるデータの量。容量とは別の数字。
- モデル並列:モデルを複数のGPUに分けて置く方法。
- GDDR7:グラフィックス向けのメモリ。HBMより安く、帯域は狭い。
Rubin CPXについて、エヌビディアは1億ドルの投資ごとに50億ドルのトークン収入を生めると説明しています。
この記事は公開資料をもとに作成しています。売買をすすめるものではありません。誤りを見つけたら訂正します(編集方針)。
関連する記事
メモリ帯域とは 端子の数×速さで決まる、HBM4は端子を2,048本に倍増
メモリ帯域は、メモリと計算の回路の間で1秒間に運べるデータの量です。エヌビディアのGPU1基の帯域は、H100の毎秒3.35テラバイトから、Rubinの毎秒22テラバイトに広がります。
モデル並列とは 重みの行列を割ってGPUで分担、Megatronは83億の重みを512基で
モデル並列は、1つのAIのモデルを複数のGPUに切り分けて置き、協力して計算する方法です。エヌビディアの研究者は2019年、83億個の重みのモデルを512基のGPUで学習し、1基の場合に比べ76%の効率を保ったと報告しました。
量子化とは 学習後か学習中か、グーグルのGemma 3は重みを54GBから14.1GBに
量子化は、AIのモデルが使う数値を16ビットから8ビットや4ビットに減らし、メモリと計算を軽くする方法です。グーグルは2025年4月、学習の段階から4ビットに慣らしたGemma 3 27Bを公開し、重みの大きさを54GBから14.1GBにしました。
建設のリードタイムとは TSMC熊本は着工から開所まで約1年10か月、アリゾナは発表から量産まで約4年半
建設のリードタイムとは、工場やデータセンターの着工から稼働までにかかる時間です。TSMCの熊本の第1工場は2022年4月に着工し、2024年2月に開所式を開きました。