本文へ移動

Ai2 が大規模 MoE 向けの学習基盤「Olmo-core 3」を公開 専門家の数を8から128に増やしても処理量の落ちは5%未満だとしている

カテゴリ共通のイメージ画像

Ai2(Allen Institute for AI)が、混合専門家(MoE)型の大規模モデルを学習させるための基盤「Olmo-core 3」を公開した。専門家の数を8から128に増やしても学習の処理量の落ちは5%未満で、1兆パラメータ超の構成でも測ったとしている。

  • Ai2(Allen Institute for AI)が2026年10月1日、混合専門家(MoE)型の大規模モデルを学習させるための基盤「Olmo-core 3」を公開した。次の世代の Olmo を動かす中核の仕組みで、誰でも使えるよう公開するとしている
  • ある測定では、1トークンあたりに選ぶ専門家を4つに保ったまま専門家の数を8から128に増やし、1トークンあたりの有効パラメータを約32億に固定したまま、全体のパラメータ容量を46億から470億へ増やした。このとき学習の処理量の落ちは5%未満だったとしている
  • 前の実装は FSDP(完全分割データ並列)で小さなバッチごとに重みを集め直していたが、今回は DDP(分散データ並列)に変え、専門家を GPU に置いたままデータを送る形にした。NVIDIA B300 を8枚使った予備の測定では、470億パラメータの MoE が1 GPU あたり毎秒52,000トークンで、前の実装の19,400から約2.7倍になったとしている
  • NVIDIA B300 を512枚使い、1トークンあたり583.6億パラメータが動く1.2兆パラメータのモデルも測り、最も高いときで1 GPU あたり858 TFLOP/s だった。別の通信方式を試した構成では合計2.38兆パラメータに届いたが、短時間の容量の確認で、継続的な学習の性能ではないと書いている
  • 低い精度の数値形式 MXFP8 にも対応した。NVIDIA B300 を4枚使った条件をそろえた測定で、BF16 と比べて学習の処理量が約21%高く、使用メモリの最大値は103GiB から95GiB に下がったとしている
  • 料金の話は発表には無い(研究者や開発者が自分の MoE を学習させるのに使えるとしている)。ライセンスの種類も発表の文章には書かれておらず、「完全に公開されている」「GitHub で公開している」という説明にとどまる。技術報告書には、専門家ごとの学習率を下げても結果は良くならなかったなど、うまくいかなかった試みも載せたとしている

確かめた一次資料

この記事は、上の一次資料の原文を読んで書いています。数字は原文のとおりで、計算した値はそう書いています。