Theme 2 / 6
AI データセンター・ファブリックとネットワーク
大規模学習・推論クラスタ向けに設計されたネットワーク層を扱う。少数の同期した高帯域フローが支配的なAIトラフィックの特性が、トポロジー、トランスポート、運用、テレメトリのすべてを一般的なクラウドネットワークとは異なる設計へと導く。
全体像
AIデータセンター・ファブリックは、大規模学習・推論クラスタ専用に構築されたネットワーキング層である。通常のクラウドネットワークとの決定的な違いは、支配的なトラフィックが多数のテナントによる独立した小フローの集合ではなく、collective communication・パラメータ交換・エキスパートルーティング・リモート状態移動によって生成される、少数の同期したバースト性の高帯域フローである点にある。
重要な視点の転換は、ネットワークがML実行エンジンの一部になったということである。学習の進捗は最も遅い参加者によって律速されるため、輻輳の小さな発生・パケットロス・ルーティングの不均衡・障害特定の遅延が、数千デバイス規模でGPU利用率全体を低下させうる。これによりファブリック設計は単なる汎用インフラ問題ではなく、AIシステム設計そのものの第一級の問題になる。
AIファブリックは4層が協調設計されるべき対象として捉えられる。(1) バックエンドネットワーク: GPU間通信専用に分離されたネットワーク。Metaは分散学習を独立したバックエンドネットワークに分離し、一般トラフィックとは独立に進化させている。(2) トポロジーと経路選択: Alibaba HPNの2-tierデュアルプレーン設計やMetaのバックエンド集約は、ECMPハッシュ偏極を減らし、低エントロピーな学習トラフィックの経路選択を予測可能にする。(3) トランスポートと輻輳制御: ロスレス/準ロスレスEthernet、RoCE、PFC、DCQCN的制御、collective対応のペーシングが、UECのような標準化やAI特化トランスポートと競合する。(4) 運用と可観測性: 1万GPU超の規模では、可観測性・障害の局所化・ストラグラー診断そのものがファブリックの一部になる。MegaScaleは学習の安定性維持のために深い可観測性が不可欠だとしている。
学習ネットワークは単一のファブリックではなく、GPUカーネルからcollectiveライブラリ、NICトランスポート、ロードバランシング、輻輳制御、トポロジー、テレメトリ、運用へと続くスタックである。研究の重心は「ネットワークを速くする」ことから「学習ジョブを予測可能にする」ことへ移っており、これには学習ランタイムとネットワークを一体で設計することが求められる。collectiveは依存構造を露出し、トランスポートは信頼性と制御を露出し、テレメトリはGPU数か月分の浪費を招く前に障害を局所化できるだけの状態を露出する必要がある。
設計上の代表的パターンとして、専用バックエンドファブリック、低エントロピートラフィックの偏極回避、collective対応の輻輳制御、複数建屋を束ねるバックエンド集約、光学・再構成可能ファブリック、障害を意識したファブリック運用などが挙げられる。これらはいずれも、GPU帯域の伸びと電気スイッチングの物理的・電力的限界を折り合わせるための試みである。
主要な概念
AIクラスタのネットワークトポロジー
AIクラスタトポロジーは、GPU・NIC・スイッチ・光回線・ラック・建屋・バックエンド集約層の物理的・論理的配置であり、同期型MLジョブでは小さな経路の不均衡が全体のGPUアイドル時間につながるため第一級の設計変数となる。バンド幅ドメインの大きさ、プレーン/レール数、静的Clos対スケジュールドファブリック対光回路交換の選択、トポロジー情報の学習フレームワークへの露出方法、障害ドメインの整合が中心的な問いである。専用バックエンドファブリック、デュアルプレーン/レール指向レイアウト、建屋横断のバックエンド集約、再構成可能・光学ファブリック、トポロジー認識スケジューリングといった設計パターンが確立しつつある。
データセンター・トランスポートプロトコル
トランスポートプロトコルは、マイクロ秒レイテンシ・高帯域・incast・バースト・マルチテナンシー・ハードウェアオフロードの下でファブリックをどう使うかを決める、RDMA・ストレージ・RPC・AI collectiveの基盤層である。研究の系譜は、DCTCP的なECN輻輳フィードバックの実運用化から、ロスレスEthernetの前提の再検証、PPTのような現実的デプロイ可能トランスポート、Falconのようなハードウェアトランスポートの再登場、SIRDやeTranによる送信者/受信者主導制御とカーネル拡張性の拡大、そして商用RNICでのパケットスプレーによる負荷分散のトランスポート層プリミティブ化へと続く。
再構成可能データセンターネットワーク
再構成可能ネットワークは、光回路やスケジュールド接続を用いてトポロジーや経路可用性を時間とともに変化させる。固定的な電気Closファブリックが、少数の大規模ワークロードに予測可能な高帯域を提供するには電力・スイッチ容量の面で高コストになることが動機である。RotorNetのマイクロ秒級光再構成、Uniform-Cost Multi-Path Routingによる経路選択、NegotiaToRのオンデマンド再構成の簡素化、Shaleのようなオブリビアスでスケーラブルな設計、再構成に適応するクレジットベーストランスポート、そしてPhotonic RailsやInfiniteHBDなどAI向けの高帯域再構成が研究の流れである。
RoCE と RDMA for AI Clusters
RDMAは低CPUオーバーヘッド・低レイテンシのリモートメモリアクセスセマンティクスを学習ジョブに提供し、RoCEはそれをEthernet上で実現することで広いベンダーエコシステムと既存運用の再利用を可能にする。AI学習でRDMAが重要な理由は同期性にあり、数千GPUにまたがるcollective通信を待つ学習ステップでは、一つのランクの輻輳・損失・再送・遅い経路がジョブ全体の効率を損なう。バックエンドネットワーク分離、ロスレス/準ロスレス運用(PFCとHOLブロッキングのトレードオフ)、collective対応トランスポート、トランスポートの一部としてのテレメトリが鍵となる論点であり、Ethernetの開放性とInfiniBand的予測可能性の緊張関係が背景にある。
Collective Communication
Collective communicationは、AllReduce・AllGather・ReduceScatter・Broadcast・All-to-Allなど分散MLワークロードが用いるグループ通信操作の集合であり、モデル並列・データ並列学習が繰り返しテンソルを同期しトークンをルーティングするため、AIデータセンターのネットワーク負荷の中心を占める。通常のRPCトラフィックと異なり依存グラフが構造化されており、全ランクがステップを完了するまで進捗しないため、ネットワークの分散がジョブレベルの遅延に直結する。TACCLのようなアルゴリズム合成、multi-commodity flowとしての定式化、対称性の活用、資源を意識したスケジューリング、そしてMoEのAll-to-Allをエッジケースではなく第一級のボトルネックとして扱う視点が重要なパターンである。
SmartNIC と DPU オフロード
SmartNICとDPUは、仮想スイッチング・ストレージオフロード・セキュリティ・テレメトリ・制御プレーン支援・シャッフル・サービスメッシュ・リソース管理といったインフラ処理を、ホストCPUからプログラム可能なNIC側コアへ移す。利点はホストCPUサイクルの節約とライン速度でのパケット処理だが、リスクとしてNIC自体が独自の資源制約・プログラミングモデル・スケジューラ・障害モード・マルチテナント分離問題を持つ「第2のコンピュータ」になる点が挙げられる。ライン速度対柔軟性、分離対利用率、オフロード対可観測性、制御プレーン対データプレーン、専用化対ポータビリティが主要な設計上の緊張である。
技術・手法
- Collective Communication Scheduling
- all-reduce・all-to-all・broadcast・reduce-scatterなどが経路と時間をどう使うかを決める手法群。TACCLのアルゴリズム合成からMCCSのサービス化、ResCCLの資源効率化、SyCCLの対称性活用、コフロー的なスケジューリング、AutoCCLの自動チューニングへと発展している。
- Collective-Aware Congestion Control
- ML collectiveの同期フェーズやランク依存関係の知識を用いてネットワーク負荷を管理する輻輳制御。collectiveフェーズ情報をトランスポート/ランタイムに露出し、ランク依存に基づいてフローをペーシングし、RoCE/RDMAの輻輳制御と協調させることで、平均リンク利用率ではなくテールステップ時間の改善を狙う。
- RDMA Reliability
- RDMAの低レイテンシ・低CPUなデータパスを維持しつつ、損失・輻輳・仮想化・マルチテナンシー・大規模障害の下で堅牢にする問題。制御されたロスレスファブリックから、ロッシーファブリック向けRDMA、パケット粒度のソフトウェア制御へと研究系譜が発展し、単一の不安定な経路が学習ストラグラーやチェックポイント失敗を生むため、パケット到達率ではなくジョブ進捗とテールステップ時間で評価すべきとされる。
- RoCE Monitoring
- RDMA-over-Ethernetファブリックのテレメトリと診断層。サブシステム異常検知(Collie)、ホストローカル診断(Hostping)、サービス対応プロービング(R-Pingmesh)、PFC由来の異常追跡(Hawkeye)を経て、単なるメトリクスではなく「どのジョブ・ランク群・経路・デバイスを修復すべきか」という実行可能な局所化結果を出すことが設計原則とされる。
- Network Telemetry
- パケット/フローカウンタ、キュー信号、プロービング、サンプリングトレース、プログラマブルスイッチ信号、OpenTelemetry的分散トレースを含む測定基盤。カバレッジ対オーバーヘッド、粒度対ストレージコスト、ネットワーク中心対サービス中心の視点、汎用トレーシング対ネットワーク特化信号、そして事後分析だけでなく制御への入力としてのテレメトリという緊張関係がある。
主要論文
| 論文名 | 年 | 一言貢献 |
|---|---|---|
| Alibaba HPN | 2024 | 周期的・バースト的・低エントロピーなLLM学習トラフィック向けに2-tierデュアルプレーンのデータセンターネットワークアーキテクチャを再設計。 |
| RDMA over Ethernet at Meta | 2024 | 大規模分散AI学習向けの本番RoCEバックエンドネットワークを、設計・ルーティング・トランスポート・運用まで包括的に記述。 |
| R-Pingmesh | 2024 | サービス対応のend-to-endプロービングでRoCEネットワークの障害を切り分け、数万RNIC規模で6か月運用し問題特定精度85%を達成。 |
| MegaScale | 2024 | 1万GPU超のLLM学習を支える本番システムで、アルゴリズム・システム全体の協調設計と深い可観測性の必要性を示す。 |
| AutoCCL | 2025 | NCCL的な低レベルcollectiveパラメータを自動チューニングし、追加ハードウェアなしで反復あたり1.07〜1.32倍の学習時間改善を達成。 |
| White-Boxing RDMA | 2025 | BlueField-3のデータパスアクセラレータを用い、ハードウェアRDMAトランスポートにパケット粒度のソフトウェア制御を露出。 |
| Revisiting RDMA Reliability for Lossy Fabrics | 2025 | PFCに依存しないスイッチ/RNIC協調設計のDCPアーキテクチャを提案し、既存のロスレス/ロッシーRDMAより1.6〜2.1倍の性能改善。 |
| MixNet | 2025 | 分散MoE学習向けに、リージョナルな光回路交換と電気ファブリックを組み合わせ、実行時にトポロジーをエキスパートトラフィックに合わせて再構成。 |
トレードオフ
レイテンシとスループットは、しばしば異なるバッチング・キューイング・配置の選択の下で改善する。危険なのは、平均スループットを最適化しつつテールレイテンシやユーザ体感の応答性を悪化させてしまうことである。LLMサービングでは大バッチがスループットを上げる一方でデコードレイテンシとTTFTが悪化し、ストレージでは書き込みバッチングが効率を上げる一方で読み取りやチェックポイント復旧が停滞しうる。トランスポートでは深いキューがリンク利用率を上げる一方でレイテンシを増大させ、collective communicationでは帯域効率とテールステップ時間が衝突しうる。設計原則としては、平均スループットとテール/進捗レイテンシを常に分離して評価すべきであり、AI学習ではステップ時間、サービングではTTFTとTBTが該当する指標となる。
未解決問題・研究の方向
- 輻輳制御の責務をcollectiveライブラリ・NIC・スイッチ・スケジューラの間でどう分担すべきか。
- Ethernetベースのファブリックは、脆弱なロスレス前提に頼らずにInfiniBand並みの予測可能な学習進捗を実現できるか。
- ジョブ配置は、特定のハードウェア世代に過学習することなくトポロジー情報をどう露出すべきか。
- リンクレベルの症状をランクレベルの学習症状にマッピングするテレメトリ抽象化はどうあるべきか。
- 光学・再構成可能ファブリックが電気的な過剰プロビジョニングより安価になるのはどのような条件か。
- トランスポートは、ワークロードごとの専用化なしにRPC・ストレージ・RDMA・AI collectiveを共通に支えられるか。
- ロジックをNICハードウェアとプログラム可能なホストソフトウェアのどちらに置くべきか、その境界線はどこにあるか。
- パケットスプレーは順序制御・輻輳制御・可観測性と両立できるか。
- 研究ロードマップとしては、AI学習向けEthernet/RoCEの予測可能性、collective対応トランスポートの置き場所、トポロジー認識配置、光学・再構成可能ファブリックの損益分岐点、ファブリック診断のランク粒度化が主要スレッドとして挙げられており、評価指標には学習ステップ時間・テールステップ時間・通信起因のGPUアイドル時間・障害局所化時間・障害あたりの損失GPU時間が用いられる。