Mistral · 2024-04 · Sparsames MoE

Mixtral 8x22B

Mixtral 8x22B ist ein Grouped-Query-Attention (GQA)-Transformer, veröffentlicht von Mistral im 2024-04, mit 56 Schichten, Hidden-Größe 6144 und einem Kontextfenster von 65,536 Tokens.

Schichtstapel

Attention ×56 · Schichten 56

Kernfakten

LaborMistral
Veröffentlicht2024-04
Parameter140.6 Mrd.
Aktiv39.1 Mrd.
Kontext65,536 tokens
AttentionGQA (48:8)
Schichten56
Hidden-Dim.6,144
Attention-Köpfe48
Vokabular32,000
PositionscodierungRoPE
NormierungRMSNorm
Aktivierungsilu
Präzision
ArchitekturklasseMixtralForCausalLM

Architekturüberblick

t₁ t₂ t₃ t₄ Eingabetokens Embedding · Vokabular 32,000 → Hidden-Dim. 6,144 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn Attention-Köpfe 48 KV-Köpfe 8 head dim 128 · RoPE Router E1 E2 E3 E4 E5 E6 E7 E8 Top-2 von 8 Experten · ≈ 39.1 Mrd. Parameter pro Token aktiv × 56 Transformer-Block Attention FFN / MoE RMSNorm pre-norm Finale Norm · RMSNorm LM head → Vokabular 32,000 p p p → nächstes Token Positionscodierung RoPE Präzision Kontext 64K tok
Gezeichnet nach der mitgelieferten config.json · 56 Schichten / Breite 6,144 / Kontext 65,536. Originaldiagramm dieses Atlas.

Attention

Grouped-Query-Attention (GQA) — 48 q-heads / 8 kv-heads. RoPE θ=1,000,000.

FFN / MoE

Es handelt sich um ein sparsames Mixture-of-Experts mit 8 gerouteten Experten und Top-2-Routing; etwa 39.1 Mrd. Parameter sind pro Token aktiv.

Feldweiser Vergleich

Verglichen mit dem vorherigen Modell desselben Labors; fehlt dieses, mit dem strukturell nächsten Modell. Die Spalte Ratio = dieses Modell ÷ Vergleichsmodell.

Vollfeldvergleich mit dem Vorgänger Mistral 7B
ModellMixtral 8x22BMistral 7BRatio
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

Strukturell ähnlichste Modelle

Ähnlichkeitswerte reichen von 0 (keine gemeinsamen kategorialen Merkmale) bis 1 (identische Profile).

Rohe Config-Felder

24 Felder
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