avatar

NVIDIA NeMoがMoE訓練の速いレイヤーを狙う

速さの数字の裏にNVIDIAスタック依存がついて回る

avatar@NVIDIAAI
27 days ago

TL;DR:

  • NeMo AutoModelはHugging Faceの普通の読み込み経路にNVIDIAカーネルを直接差し込んでいる
  • 訓練コードを変えなくてもMoEモデルで実際に速くなる
  • 一度この速さに乗ると他のスタックへ戻る手間がどんどん増える

NVIDIAはMoE訓練の話をモデル構造から高速実行レイヤー誰が握るかに移した。Transformers v5で基本的な拡張はできたもののNeMo AutoModelはNVIDIAのカーネルと並列化を多くの人が使うfrom_pretrainedにそのまま入れている。

大規模にファインチューニングする研究所や企業は実質的に遅い実行を受け入れるかNVIDIAスタックに乗るかの二択を迫られる。

速度改善の数字ははっきりしている。Expert ParallelismとDeepEP、TransformerEngineカーネルでスループットが3.4から3.7倍になりメモリ使用量は29から32パーセント減る。通常のロード呼び出しのまま動くので表面的にはただのアップグレードに見える。

ただ実務では違う。一度この高速経路を標準にすると後から別スタックに移る時間とコストがかさむ。他のフレームワークが徐々に二番手に見えてくる。

チップ側の力を持たない純粋なソフトウェアプロジェクトは大規模MoEでさらに不利になる。短期で結果を出したい企業にとってもNVIDIAの経路にロックインする理由が増える。

市場はまだ互換性を保ったまま得られる速度改善がどれだけ早く標準になるかを過小評価している。素のFSDPや独自のDeepSpeed構成は簡単な導入で二桁改善が出る選択肢の前では見劣りしやすい。

X上の反応は本質ではない。研究所がこれを実際のファインチューニングパイプラインに組み込むかどうかが大事だ。現時点でHugging Faceや競合から目立った反発はなくこの変化の大きさはまだ十分に認識されていない。

NVIDIAはMoE訓練の性能基準を早い段階で押さえに来た。フルスタックに乗るチームと投資家は持続的な優位を取りやすい。一方で純粋なオープンソース選択肢に依存する参加者は時間とともに周縁へ押し出されるリスクがある。