本文へスキップ
IFAI

研究

NVIDIAがHelix並列化を発表:数百万トークンのリアルタイム推論で同時実行数を32倍に

公開 拡散指数 31

NVIDIAはBlackwell向けの新技術「Helix Parallelism」を公開しました。アテンションとFFNの並列戦略を分離することでDRAM帯域のボトルネックを解消し、一定の遅延要件下で同時接続ユーザー数を最大32倍に引き上げます。今後主要な推論フレームワークへ統合予定です。

NVIDIAがHelix並列化を発表:数百万トークンのリアルタイム推論で同時実行数を32倍に
画像:developer.nvidia.com

数か月分のチャット履歴、百科事典並みの判例集、あるいは巨大なコードベースをまるごとAIに読み込ませると、誰もがある壁にぶつかります。それは、出力が信じられないほど遅くなるという問題です。

超長文コンテキストの推論では、1トークン生成するごとに数百万トークン分の過去ログ(KVキャッシュ)をDRAMから読み出しつつ、巨大なFFN(フィードフォワードネットワーク)の重みもロードしなければなりません。従来のテンソル並列(TP)では、LlamaのGQAやDeepSeekのMLAといったアーキテクチャにおいてKVキャッシュが複数GPU間で重複保持されてしまい、メモリ帯域が即座にパンクしてしまいます。

NVIDIAは7月8日、Blackwellアーキテクチャ向けに設計された新たな並列化手法「Helix Parallelism(ヘリックス並列)」を発表しました。数百万トークン級のコンテキストを持つモデルでも、リアルタイムなデコードを可能にする技術です。

アテンションとFFNの分離運用

DNAの二重螺旋構造に着想を得たHelixの最大の特徴は、Transformer層内のアテンション処理とFFN処理で並列化戦略を完全に切り離し、同じGPU群を使い回す点にあります。

アテンションフェーズでは、KV並列(KVP)を用いてシーケンス方向にキャッシュを分割し、KVヘッド数を超えない範囲でテンソル並列を適用します。これにより、キャッシュの重複保持を完全に排除します。局所的なFlashAttentionの計算後、GPU間で結果を持ち寄るAll-to-All通信を行いますが、この通信量はバッチサイズに比例するだけで、コンテキスト長が数百万トークンに伸びても膨らみません。

さらに、バッチ間で計算と通信を重ね合わせるパイプライン手法(HOP-B)により、通信レイテンシを計算の裏側に隠蔽します。

アテンションが終わると、同じGPU群がそのままFFNフェーズ向けに再構成されます。高密度モデルなら1次元のテンソル並列、DeepSeekのようなMoEモデルならテンソル並列とエキスパート並列(EP)の2次元グリッドに切り替わります。新しいトークンの書き込みも複数GPUにラウンドロビン方式で分散させ、メモリの偏りを防ぎます。

同一遅延で同時実行ユーザー数が最大32倍に

Blackwellシステム(NVLink 72およびFP4演算)を想定したシミュレーションにおいて、100万トークンのコンテキストを持つDeepSeek-R1(671B)を処理したところ、Helixは圧倒的な数値を示しました。

一定のレイテンシ枠を維持した場合、対応可能な同時接続ユーザー数は最大32倍(GPUあたりのスループットが32倍)に向上しました。

また、低負荷で純粋な応答速度を求める環境でも、トークン間の最小遅延を最大1.5倍短縮(高速化)できています。

なぜ重要か

私たちは、Helixが長文AIエージェントの商業化における最大のブレークスルーの一つになると考えています。

長文対応モデルは実用的価値が高い反面、ホスティング費用と待ち時間の長さが最大の普及障壁でした。1台のクラスタでさばけるユーザー数が32倍になるなら、法務リサーチやリポジトリ全体を常時把握するコーディングアシスタントの運用コストは劇的に下がります。

また、この技術はNVIDIAのハードウェア優位性を一段と強固にします。Helixの頻繁なAll-to-All通信と動的な構成変更は、BlackwellのNVLink 72による超広帯域インターコネクトがあって初めて成立します。単体チップの演算性能で追随しようとする競合他社にとって、システム全体の通信帯域で殴り込みをかけるこのアプローチは非常に崩しにくい壁となります。

ただし、今回のデータは網羅的なシステムシミュレーションに基づく結果であり、本番環境の予測不能なトラフィック下での実測値ではありません。今後TensorRT-LLMなどの推論フレームワークへ実装された際に、どこまでシミュレーション通りの効率を発揮できるかが次の注目点です。

情報源 (1)

  1. [1]
    developer.nvidia.com(新しいタブで開きます)原文developer.nvidia.com

毎日のAIピックアップ