AIスタートアップd-Matrixが、AI推論ワークロードの「メモリ・ウォール」問題を抜本的に解決しうる次世代チップアーキテクチャ「Raptor」を発表した。同社は、この技術が次世代高帯域幅メモリ(HBM)の「HBM4」と比較しても桁違いの性能目標を掲げており、これによってAIハードウェアの進化を加速させることが可能と謳っている。果たしてこれはどのような技術なのだろうか?

AD

メモリが律速するAI推論の現実

生成AIの進化が指数関数的な様相を呈する中、我々はAIコンピューティングにおける深刻なボトルネックに直面している。それは演算性能、すなわちFLOPsの不足ではない。問題の本質は「メモリ」にある。1994年に提唱された「メモリ・ウォール」という概念は、プロセッサの性能向上ペースにメモリ帯域幅の進化が追いつかないという、古くて新しい課題を指し示す。この30年来の課題が、巨大化するAIモデルによって、かつてないほど深刻な形でデータセンターの前に立ちはだかっているのだ。

特に、全AIワークロードの85%以上を占めると予測される「推論」フェーズにおいて、この問題は顕著である。モデルのパラメータや中間データをプロセッサに供給するメモリ帯域が、システム全体の性能を決定づける律速段階となっている。この状況に対し、業界のデファクトスタンダードであるNVIDIAは、広帯域メモリ(HBM)をGPUに近接実装することで解を提示してきた。しかし、HBMはコスト、消費電力、そして供給量のすべてにおいて大きな課題を抱える。

この根源的な課題に対し、第一原理思考に基づいた全く新しいアプローチで挑むのが、2019年設立のスタートアップ、d-Matrixである。同社は、AI推論ワークロードに特化した次世代チップアーキテクチャ「Raptor」と、その中核をなす革新的な3D DRAM技術「3DIMC (3D stacked digital in-memory compute)」を発表。現行のHBM4と比較して「10倍のメモリ帯域幅」と「10倍のエネルギー効率」という、極めて野心的な目標を掲げた。これは単なる漸進的な改善ではない。メモリとコンピュートの関係性を再定義し、AIインフラの経済性を根底から覆す可能性を秘めた、アーキテクチャレベルでの挑戦状なのだ。

3DIMCアーキテクチャの構造と原理

d-Matrixのアプローチの核心は、「コンピュートとメモリの物理的距離を極限まで縮小し、データ移動そのものをなくす」という思想にある。これを実現するのが、Digital In-Memory Compute (DIMC)と3D積層技術の融合である3DIMCだ。

Digital In-Memory Compute (DIMC): メモリセルが演算する世界

従来のフォン・ノイマン型アーキテクチャでは、データはメモリから演算ユニットへバスを経由して転送され、演算結果が再びメモリに書き戻される。このデータ移動が、レイテンシと消費電力の主たる要因だ。

d-Matrixが採用するDIMCは、この原則を覆す。同社のDIMCエンジンは、標準的なSRAM(Static RAM)セルに演算用のトランジスタを追加で実装することで、メモリセル自体が演算能力を持つ構造を実現している。具体的には、AI推論、特にTransformerモデルで多用される「行列ベクトル積」演算を、メモリセルアレイの内部で直接実行する。これによって得られるメリットは以下の通りだ。

  • データロードの削減: モデルの重み(パラメータ)はSRAMセルに保持されたままで、入力データ(アクティベーション)が入力されると、セル内部で乗算・累算処理が行われる。これにより、重みデータを演算器にロードするという最もオーバーヘッドの大きいプロセスが原理的に不要となる。
  • 並列処理: d-MatrixのApolloコンピュートコアは、8つのDIMCユニットを内包し、それぞれが64×64の行列乗算を並列実行できる。これは、Transformerのアテンション計算のような大規模な並列演算に対して極めて高い親和性を持つアーキテクチャである。
  • データフォーマットへの最適化: 推論ワークロードでは、学習ほどの精度は要求されない。d-Matrixのアーキテクチャは、INT8やINT4、ブロック浮動小数点(Block Floating Point)といった低精度データフォーマットをネイティブにサポートしており、メモリフットプリントと演算コストをさらに削減する設計となっている。

このDIMCは、いわば究極のカーネルフュージョン(複数の演算を一つのカーネルに統合してメモリアクセスを削減する技術)をハードウェアレベルで実装したものと解釈できる。ソフトウェアによる最適化の限界を超え、物理法則のレベルでデータ移動を削減するアプローチである。

3D積層技術: 帯域幅と容量を垂直方向に拡張

DIMCがデータ移動の「回数」を減らす技術だとすれば、3D積層はデータ移動の「距離」を縮め、「経路」を増やす技術である。d-Matrixは、この二つを組み合わせることで相乗効果を狙う。

同社が「Pavehawk」と呼ぶ、ラボで現在稼働中の最初の3DIMCシリコンは、この構造を実証するための重要なマイルストーンだ。

  • 積層構造: Pavehawkは、ベースとなるロジックダイの上に、複数のDRAMダイを垂直に積層する構造を持つ。ロジックダイはTSMCのN5プロセスで製造され、DIMCエンジンや制御回路が集積されている。その上に積層されるのは、高価なHBMではなく、コスト効率に優れた汎用のLPDDR5メモリである。
  • 超短距離インターコネクト: ロジックダイとDRAMダイは、シリコン貫通ビア(TSV)などの微細な配線技術によって直接接続される。これにより、従来のパッケージ基板を経由する配線と比較して、信号の伝送距離が桁違いに短くなる。これは、信号の減衰が少なく、より高速かつ低消費電力でのデータ転送を可能にすることを意味する。
  • 帯域幅の飛躍的向上: 垂直方向にDRAMダイを積層することで、ロジックダイとの接続インターフェースの数を水平方向の制約から解放する。結果として、極めて広大なバス幅を実現でき、これがHBM4を凌駕するメモリ帯域の根拠となっている。

このアーキテクチャは、プロセッサとメモリが別々のチップとしてパッケージ上で接続されていた従来の構成から、単一の3D構造体へと進化させるものだ。これは、メモリがプロセッサの「外部記憶」ではなく、「内部構造の一部」となるパラダイムへの移行を示唆している

チップレットアーキテクチャの戦略的価値

d-Matrixの製品は、Corsairプラットフォームから一貫してチップレットアーキテクチャを採用している。この選択は、3DIMCの実現と市場投入において極めて戦略的な意味を持つ。

モノリシックな巨大チップを設計・製造するのではなく、比較的小さな機能単位(チップレット)を組み合わせることで、開発の柔軟性とコスト効率が向上する。特にメモリ技術の進化が速いAIアクセラレータ市場において、このアプローチは大きな利点となる。

  • 迅速な技術採用: 新しい規格のDRAMが登場した際、ロジック部分の設計を大きく変更することなく、メモリチップレットのみを更新して対応できる。これにより、市場の要求に迅速に応えることが可能だ。
  • 歩留まりの向上: 小さなチップレットは、大きなモノリシックダイよりも高い歩留まりで製造できる。これは、最終製品のコスト競争力に直結する。
  • スケーラビリティ: 複数のコンピュートチップレットとメモリチップレットをインターポーザ上で組み合わせることで、小規模なエッジデバイスから大規模なデータセンター向けまで、多様な製品ラインナップを効率的に展開できる。

次世代アーキテクチャ「Raptor」は、このチップレット設計とPavehawkで実証された3DIMC技術を完全に統合したものになる。これは、単一の技術的ブレークスルーに依存するのではなく、複数の先進技術を有機的に結合させた、極めて洗練されたシステムアーキテクチャと言えるだろう。

AD

HBM4に対する「10倍」の技術的妥当性

d-Matrixが掲げる「HBM4比で10倍の帯域と電力効率」という目標は、にわかには信じがたいほどの野心的な数値だ。しかし、これまで見てきたアーキテクチャの原理に立ち返れば、その主張の技術的根拠が見えてくる。

帯域幅の比較分析

HBM4の仕様はまだ確定していないが、2048ビットのインターフェース幅と高速な転送レートにより、1スタックあたり2TB/sを超える帯域幅が目標とされている。これに対し、3DIMCはどのようにして10倍、すなわち20TB/sクラスの帯域を目指すのか。

その鍵は、前述の通り「接続密度の違い」にある。HBMはマイクロバンプを介してインターポーザに接続されるが、3DIMCではより微細なダイ間の直接接続技術を用いることで、桁違いの数のI/Oを確保できる可能性がある。これは、チップの面積あたりに確保できる帯域幅(Bandwidth Density)において、原理的にHBMを上回るポテンシャルを持つことを意味する。さらに、DIMCによってメモリ内部で演算が完結するため、実効帯域幅(Effective Bandwidth)の利用効率が極めて高い。外部へのデータ転送がボトルネックにならず、内部の膨大な帯域をロスなく演算に活用できるのだ。

エネルギー効率 (pJ/bit) の考察

エネルギー効率の改善は、データ移動距離の短縮にほぼ直結する。GPUとHBMパッケージ内でのデータ移動ですら数ピコジュール/ビット(pJ/bit)のエネルギーを消費するのに対し、3DIMCにおけるダイ間の垂直方向のデータ移動は、その数分の一、あるいは一桁小さいエネルギーで済む可能性が高い。

さらに重要なのは、DIMCによってそもそも不要になるデータ移動の存在だ。重みパラメータを外部メモリから読み出すという、AIアクセラレータにおける最も電力消費の大きい処理の一つが根こそぎ削減される。この効果は絶大であり、「10倍」という目標も、アーキテクチャの観点からは決して非現実的な数字ではない。

「我々がターゲットとするのは、HBM4と比較して10倍優れたメモリ帯域幅と10倍優れたエネルギー効率だ。これらは漸進的な改善ではない。スケールでの推論に何が可能かを再定義する、ステップ関数の改善である」

– Sudeep Bhoja, d-Matrix CTO

この言葉は、d-Matrixが自社の技術に抱く自信の表れであり、メモリ・ウォール問題に対する彼らの根本的な解決策が、既存技術の延長線上にはないことを明確に示している。

市場への影響とd-Matrixが直面する課題

d-Matrixの技術が持つポテンシャルは計り知れないが、それが市場で成功を収めるかは別の問題である。特に、NVIDIAがCUDAという強力なソフトウェア・エコシステムで築き上げた牙城はあまりにも高い。

HBMエコシステムへの挑戦

NVIDIAのGPUアーキテクチャは、HBMとの緊密な連携を前提に最適化され続けてきた。サプライチェーン、パッケージング技術、そして何よりもCUDAライブラリ群は、HBMの特性を最大限に引き出すように設計されている。d-Matrixの3DIMCは、この巨大なエコシステム全体に対する挑戦となる。

d-Matrixのアプローチが成功すれば、HBMの供給不足やコスト高騰に悩むハイパースケーラーやクラウドプロバイダーにとって、極めて魅力的な代替案となり得る。特に、推論専用インフラのTCO(総所有コスト)を劇的に削減できる可能性は、彼らがd-Matrixの技術を評価する強い動機となるだろう。

ソフトウェアスタックという最大の障壁

しかし、いかに優れたハードウェアであっても、それを使いこなすソフトウェアがなければただのシリコンの塊に過ぎない。d-Matrixにとって最大の課題は、独自のアーキテクチャをサポートするコンパイラ、ライブラリ、そして開発者ツールといったソフトウェアスタックを構築し、普及させることにある。

NVIDIAのCUDAが持つ強みは、単なるAPIの集合体ではない。長年にわたって蓄積された最適化のノウハウ、膨大な数の対応アプリケーション、そして世界中の開発者コミュニティそのものが参入障壁となっている。d-Matrixがこの壁を乗り越えるには、PyTorchやTensorFlowといった主要なAIフレームワークとのシームレスな統合はもちろんのこと、特定のワークロードにおいてCUDAを凌駕する圧倒的な性能を提示し、開発者が「乗り換える価値がある」と判断するだけのインセンティブを提供する必要がある。このソフトウェア開発の成否が、d-Matrixの未来を左右すると言っても過言ではない。

推論特化からその先へ

d-Matrixは現在、戦略的に「推論」市場にフォーカスしている。これは、市場規模の大きさと、技術的な特化による優位性を築きやすいという点で、スタートアップとしては極めて賢明な戦略である。

Pavehawkでの実証を経て市場に投入される次世代アーキテクチャ「Raptor」は、まず大規模言語モデル(LLM)の推論市場におけるTCO削減の切り札として、その価値を問われることになるだろう。ここで確固たる地位を築くことができれば、その先にはさらに大きな可能性が広がる。3DIMCのアーキテクチャは、原理的には推論に限定されるものではない。将来的には、より広範なメモリ律速型のコンピューティング課題、例えば科学技術計算やゲノム解析などへの応用も視野に入ってくるだろう。

この技術的選択が、競合他社の戦略、特に次世代GPUのメモリサブシステムの設計思想にどのような影響を与えるかが今後の注目ポイントとなるだろう。d-Matrixの挑戦は、AIアクセラレータの進化の方向性を左右する上で非常に重要な事例となりそうだ。


Sources