推論サーバーとは 問い合わせをまとめてGPUに流すソフト、処理量2〜4倍の例
推論サーバーは、AIへの大量の問い合わせを受け付け、GPUで効率よく処理して答えを返すソフトです。公開の推論ソフトvLLMの論文は、従来のソフトに比べ処理量を2〜4倍に増やせたと報告しています。
- 推論サーバーは、AIへの問い合わせを集めてGPUに流し、答えを返す役のソフトです。
- 大規模言語モデルでは、途中の計算結果を残す「KVキャッシュ」の置き場所の管理が性能を左右します。
- vLLMは、KVキャッシュを細かい区画で管理する方法で、処理量を従来の2〜4倍にしたと論文で報告しました。
- エヌビディアのDynamoは、Triton Inference Serverの後継です。GB200 NVL72でGPU1個あたりの生成量が30倍超になったとしています。
チャットのAIに質問すると、その文章はデータセンターの推論サーバーに届きます。推論サーバーは、同じ時刻に届く多数の問いをまとめ、GPUに計算させ、答えを1語ずつ返します。モデルそのものを速くするソフトとは別の層で、交通整理を受け持ちます。
前半と後半で性質が違う
大規模言語モデルの推論は2段に分かれます。前半の「プリフィル」は、入力の文章をまとめて読み込む段で、計算の量が多くなります。後半の「デコード」は、答えを1語ずつ作る段で、毎回メモリから重みと途中の結果を読むため、メモリの速さで決まります。
途中の結果は「KVキャッシュ」としてGPUのメモリに残します。会話が長いほど、利用者が多いほど大きくなります。推論サーバーは、どの問いのKVキャッシュをどこに置くかを管理し、多くの問いを同時に処理できるようにします。
主なソフト
| 名前 | 出どころ | 特徴と示された数字 |
|---|---|---|
| vLLM | 米カリフォルニア大学バークレー校の研究から公開 | KVキャッシュを区画で管理。処理量は従来の2〜4倍(論文) |
| SGLang | 公開のソフト | AMDのROCm 10.1版も対応の一覧に入れる |
| Dynamo | エヌビディア(2025年3月18日発表) | 前半と後半を別のGPUに分担。GB200 NVL72でGPUあたりの生成量30倍超 |
vLLMの論文によると、従来のソフトでは、KVキャッシュのメモリが細切れや重複で無駄になり、同時に処理できる問いの数が限られていました。vLLMは、パソコンのOSがメモリを扱うように、KVキャッシュを小さな区画に分けて管理します。同じ待ち時間のまま、処理量はFasterTransformerやOrcaと比べ2〜4倍になりました。vLLMは今、PyTorch Foundationの事業の一つです。
エヌビディアのDynamo
エヌビディアは2025年3月18日、公開の推論の枠組み「Dynamo」を発表しました。会社の説明では、従来のTriton Inference Serverの後継です。中ではTensorRT-LLM、vLLM、SGLangを動かせます。
Dynamoは、前半と後半を別々のGPUに分け、それぞれに合った数のGPUを割り当てます。過去の問いのKVキャッシュがあるGPUに次の問いを回す仕組みや、使わないKVキャッシュを安いメモリや記憶装置に移す仕組みも持ちます。発表時、Hopper世代でLlamaのモデルを動かす場合、同じGPUの数で性能を2倍にできるとしました。発表文には、AWS、コアウィーブ、グーグル・クラウド、メタ、マイクロソフトのAzure、ネビウスなどが利用や対応の企業として並びました。
現在の製品ページでは、GB300 NVL72とDynamoの組み合わせで、専門家混合型(MoE)のモデルの処理量がHopper世代の最大50倍になるとしています。
関連する用語
- KVキャッシュ:言語モデルが答えを作る途中の結果を残したもの。
- プリフィル/デコード:推論の前半(入力を読む)と後半(1語ずつ作る)。
- バッチ処理:複数の問いをまとめて1回で計算すること。
- TensorRT-LLM:エヌビディアの言語モデル向けの推論ライブラリ。
エヌビディアは2026年8〜10月期の決算を11月17日(米国時間)に発表します。
この記事は公開資料をもとに作成しています。売買をすすめるものではありません。誤りを見つけたら訂正します(編集方針)。
関連する記事
TensorRTとは 学習済みモデルをGPU向けに作り直す推論の道具
TensorRTは、学習済みのAIのモデルを、エヌビディアのGPUで速く答えを出せる形に変えるソフトの一式です。エヌビディアは、CPUだけで動かす場合に比べ推論を36倍速くできるとしています。
メモリ容量の壁とは 4,050億の重みは16ビットで810GB、H100の10倍超
メモリ容量の壁は、AIのモデルの大きさに半導体のメモリの量が追いつかない問題です。メタのLlama 3.1 405Bは16ビットで約810GBとなり、H100の80GBの10倍を超えます(計算)。
量子化とは 学習後か学習中か、グーグルの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月に開所式を開きました。