Mistral · 2024-04 · スパースMoE

Mixtral 8x22B

Mixtral 8x22B は Mistral が 2024-04 に公開した グループ化クエリ注意(GQA) Transformer です。56 層、隠れ次元 6144、コンテキスト長 65,536 トークン。

層スタック

注意機構 ×56 · 層数 56

主要スペック

開発元Mistral
公開日2024-04
総パラメータ140.6 B
活性パラメータ39.1 B
コンテキスト65,536 tokens
注意機構GQA (48:8)
層数56
隠れ次元6,144
注意ヘッド48
語彙数32,000
位置エンコーディングRoPE
正規化RMSNorm
活性化関数silu
学習精度
アーキテクチャクラスMixtralForCausalLM

アーキテクチャ概要

t₁ t₂ t₃ t₄ 入力トークン Embedding · 語彙数 32,000 → 隠れ次元 6,144 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn 注意ヘッド 48 KVヘッド 8 head dim 128 · RoPE ルーター E1 E2 E3 E4 E5 E6 E7 E8 8 エキスパートから top-2 · トークンごとに約 39.1B パラメータ活性 × 56 Transformer ブロック 注意機構 FFN / MoE RMSNorm pre-norm 最終正規化 · RMSNorm LM head → 語彙数 32,000 p p p → 次のトークン 位置エンコーディング RoPE 学習精度 コンテキスト 64K tok
同梱の config.json に基づき作図 · 56 層 / 隠れ次元 6,144 / コンテキスト 65,536。図は本サイトのオリジナルです。

注意機構

グループ化クエリ注意(GQA) — 48 q-heads / 8 kv-heads. RoPE θ=1,000,000.

FFN / MoE

スパースなMixture-of-Expertsモデルで、8 個のルーティングエキスパートと top-2 ルーティングを持ち、トークンごとに約 39.1B のパラメータが活性化します。

フィールド単位の比較

同一開発元の前世代モデルと比較します。前世代がない場合は構造最相似モデルと比較。倍数欄 = 本モデル ÷ 比較対象。

前世代 Mistral 7B との全フィールド比較
モデルMixtral 8x22BMistral 7B倍数
model_typemixtralmistral
architecturesMixtralForCausalLMMistralForCausalLM
hidden_size61444096×1.50
num_hidden_layers5632×1.75
num_attention_heads4832×1.50
num_key_value_heads88≈1
intermediate_size1638414336×1.14
hidden_actsilusilu
num_experts_per_tok2
rope_theta100000010000×100.00
max_position_embeddings6553632768×2.00
sliding_window4096
rms_norm_eps0.000010.00001≈1
vocab_size3200032000≈1
torch_dtypebfloat16bfloat16
attention_dropout0
bos_token_id11≈1
eos_token_id22≈1
initializer_range0.020.02≈1
num_local_experts8
output_router_logitsfalse
router_aux_loss_coef0.001
tie_word_embeddingsfalsefalse
transformers_version4.38.04.34.0.dev0
use_cachetruetrue

構造が最も近いモデル

類似度は 0(共通のカテゴリ特徴なし)から 1(構造プロファイルが完全に一致)の範囲です。

config.json の全フィールド

24 フィールド
architecturesMixtralForCausalLM
attention_dropout0
bos_token_id1
eos_token_id2
hidden_actsilu
hidden_size6144
initializer_range0.02
intermediate_size16384
max_position_embeddings65536
model_typemixtral
num_attention_heads48
num_experts_per_tok2
num_hidden_layers56
num_key_value_heads8
num_local_experts8
output_router_logitsfalse
rms_norm_eps0.00001
rope_theta1000000
router_aux_loss_coef0.001
tie_word_embeddingsfalse
torch_dtypebfloat16
transformers_version4.38.0
use_cachetrue
vocab_size32000