Mistral · 2024-04 · 稀疏 MoE

Mixtral 8x22B

Mixtral 8x22B 是 Mistral 於 2024-04 發布的 分組查詢注意力(GQA) Transformer,共 56 層,隱藏維度 6144,上下文視窗 65,536 個 token。

層堆疊

注意力 ×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₄ 輸入 token 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 · 每 token 约激活 39.1B 参数 × 56 Transformer 块 注意力 前饋 / MoE RMSNorm pre-norm 最終正規化 · RMSNorm LM head → 詞表 32,000 p p p → 下一 token 位置編碼 RoPE 訓練精度 上下文 64K tok
依據隨權重發布的 config.json 繪製 · 56 層 / 寬 6,144 / 上下文 65,536。本圖為本站原創示意圖。

注意力

分組查詢注意力(GQA) — 48 q-heads / 8 kv-heads. RoPE θ=1,000,000.

前饋 / MoE

它是稀疏混合專家模型:8 個路由專家、top-2 路由;每個 token 約激活 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 欄位

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