MoE(専門家混合)とは DeepSeek-V3は6710億のうち1トークンに370億だけを使う
MoEは、多数の専門家の中から入力ごとに一部だけを選んで計算するモデルの設計です。DeepSeek-V3は全体で6710億のパラメータを持ち、1つのトークンの計算には370億だけを使います。
- MoEは、モデルの一部を「専門家」に分け、入力ごとに選んだ専門家だけで計算する設計です。
- どの専門家を使うかは「ルーター」と呼ぶ小さな仕組みが、トークンごとに決めます。
- ミストラルAIのMixtral 8x7Bは全体467億のうち129億、DeepSeek-V3は6710億のうち370億を使います。
- 計算は減りますが、全部の専門家をメモリに置くため、メモリの容量とチップ間の通信の負担は残ります。
MoE(Mixture of Experts、専門家混合)は、AIのモデルの中に、専門家(エキスパート)と呼ぶ小さなネットワークを多数並べておき、入力ごとにその一部だけを動かす設計です。全部の数値を毎回使う「密な」モデルに対して、「疎な」モデルとも呼ばれます。
仕組み ルーターが専門家を選ぶ
言語モデルの各層には、計算の中心となる部分(フィードフォワード層)があります。MoEでは、この部分が複数の専門家に分かれた形です。トークンが層に入るたびに、ルーターが各専門家の向き不向きを点数にし、点数の高い少数の専門家だけに計算を任せます。残りの専門家は、そのトークンの計算には使われません。
考え方の出発点としてよく挙げられるのが、グーグルの研究者らが2017年1月に公開した論文です。最大で数千の専門家を並べた層を提案し、モデルの容量を1000倍超に増やしながら計算の効率をあまり落とさずに済むとしました。最大1370億パラメータのモデルを試しています。
公表されたMoEのモデル
| モデル | 公表 | 全体のパラメータ | 1トークンで使う分 | 割合(計算) |
|---|---|---|---|---|
| Mixtral 8x7B(ミストラルAI) | 2023年12月 | 467億 | 129億 | 27.6% |
| DeepSeek-V3 | 2024年12月 | 6710億 | 370億 | 5.5% |
ミストラルAIは2023年12月11日、Mixtral 8x7Bを公開しました。各層に8つの専門家を置き、ルーターがトークンごとに2つを選びます。同社は、メタのLlama 2 70Bを多くの試験で上回り、推論は6倍速いと説明しています。
DeepSeek-V3は、14.8兆トークンで学習し、学習全体に使ったのはH800を延べ278万8000GPU時間だったと技術報告で公表しました。推論に強いDeepSeek-R1も同じ6710億・370億の構成です。
半導体への影響
MoEは1トークンあたりの計算を減らしますが、どの専門家がいつ選ばれるかは事前にわからないため、全部の専門家をメモリに置いておく必要があります。DeepSeek-V3の6710億のパラメータを8ビットで置くと約671GBで、1基のGPUには収まりません(計算)。専門家を多数のGPUに分けて置くと、トークンをその専門家のいるGPUへ送る通信が増えます。大容量のHBMと、GPUどうしを結ぶ高速の配線の需要につながる設計です。
関連する用語
- パラメータ数:モデルの中の数値の個数。MoEでは全体の数と使う数の2つで示します。
- ルーター(ゲート):トークンごとに使う専門家を選ぶ仕組み。
- 密なモデル:毎回すべてのパラメータを使って計算するモデル。
DeepSeek-R1は12万8000トークンの文脈を扱えるとしています。
この記事は公開資料をもとに作成しています。売買をすすめるものではありません。誤りを見つけたら訂正します(編集方針)。
関連する記事
パラメータ数とは 4050億パラメータのモデルは16ビットで約810GBのメモリが要る
パラメータ数は、AIモデルの中の調整できる数値の個数です。メタのLlama 3.1 405Bは4050億個あり、1個を16ビット(2バイト)で持つと、数値だけで約810GBのメモリを使います(計算)。
スケーリング則とは 2020年の論文は7けた超の範囲でべき乗の関係を確かめた
スケーリング則は、モデルとデータと計算を増やすほど、言語モデルの誤りが決まった形で減るという経験則です。2020年1月の論文は、この関係が7けたを超える範囲で成り立つと報告しました。
推論(インファレンス)とは グーグルの自社モデルのAPIは1分に約320億トークンを処理
推論は、できあがったAIモデルを使って答えを出す計算です。アルファベットは10月28日、自社モデルのAPIが1分あたり約320億トークンを処理していると明らかにしました。
建設のリードタイムとは TSMC熊本は着工から開所まで約1年10か月、アリゾナは発表から量産まで約4年半
建設のリードタイムとは、工場やデータセンターの着工から稼働までにかかる時間です。TSMCの熊本の第1工場は2022年4月に着工し、2024年2月に開所式を開きました。