Mistral · 2024-04 · MoE sparsa

Mixtral 8x22B

Mixtral 8x22B è un Transformer a attention grouped-query (GQA) pubblicato da Mistral nel 2024-04, con 56 strati, dimensione nascosta 6144 e una finestra di contesto di 65,536 token.

Pila di strati

Attention ×56 · Strati 56

Dati chiave

LaboratorioMistral
Pubblicazione2024-04
Parametri140.6 mld
Attivi39.1 mld
Contesto65,536 tokens
AttentionGQA (48:8)
Strati56
Dim. nascosta6,144
Teste di attention48
Vocabolario32,000
Codifica posizionaleRoPE
NormalizzazioneRMSNorm
Attivazionesilu
Precisione
Classe architetturaleMixtralForCausalLM

Panoramica dell'architettura

t₁ t₂ t₃ t₄ token di ingresso Embedding · Vocabolario 32,000 → Dim. nascosta 6,144 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn Teste di attention 48 Teste KV 8 head dim 128 · RoPE Router E1 E2 E3 E4 E5 E6 E7 E8 top-2 su 8 esperti · ≈ 39.1mld parametri attivi per token × 56 blocco transformer Attention FFN / MoE RMSNorm pre-norm Normalizzazione finale · RMSNorm LM head → Vocabolario 32,000 p p p → token successivo Codifica posizionale RoPE Precisione Contesto 64K tok
Disegnato dal config.json distribuito · 56 strati / larghezza 6,144 / contesto 65,536. Schema originale di questo atlante.

Attention

attention grouped-query (GQA) — 48 q-heads / 8 kv-heads. RoPE θ=1,000,000.

FFN / MoE

È un mixture-of-experts sparso con 8 esperti instradati e routing top-2; circa 39.1 milioni di parametri sono attivi per token.

Confronto campo per campo

Confronto con il modello precedente dello stesso laboratorio; in assenza, col modello strutturalmente più vicino. La colonna rapporto = questo modello ÷ modello di confronto.

Confronto completo con il predecessore Mistral 7B
ModelloMixtral 8x22BMistral 7BRapporto
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

Architetture più simili

I valori di similarità vanno da 0 (nessuna caratteristica categoriale condivisa) a 1 (profili identici).

Campi grezzi del config.json

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