Mistral · 2024-04 · MoE épars

Mixtral 8x22B

Mixtral 8x22B est un Transformer à attention groupée (GQA) publié par Mistral en 2024-04, avec 56 couches, une dimension cachée de 6144 et une fenêtre de contexte de 65,536 tokens.

Pile de couches

Attention ×56 · Couches 56

Faits clés

LaboratoireMistral
Publication2024-04
Paramètres140.6 Md
Actifs39.1 Md
Contexte65,536 tokens
AttentionGQA (48:8)
Couches56
Dimension cachée6,144
Têtes d'attention48
Vocabulaire32,000
Encodage positionnelRoPE
NormalisationRMSNorm
Activationsilu
Précision
Classe d'architectureMixtralForCausalLM

Vue d'ensemble de l'architecture

t₁ t₂ t₃ t₄ tokens d’entrée Embedding · Vocabulaire 32,000 → Dimension cachée 6,144 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn Têtes d'attention 48 Têtes KV 8 head dim 128 · RoPE Routeur E1 E2 E3 E4 E5 E6 E7 E8 top-2 sur 8 experts · ≈ 39.1Md de paramètres actifs par token × 56 bloc transformer Attention FFN / MoE RMSNorm pre-norm Norme finale · RMSNorm LM head → Vocabulaire 32,000 p p p → token suivant Encodage positionnel RoPE Précision Contexte 64K tok
Dessiné d’après le config.json livré · 56 couches / largeur 6,144 / contexte 65,536. Schéma original de cet atlas.

Attention

attention groupée (GQA) — 48 q-heads / 8 kv-heads. RoPE θ=1,000,000.

FFN / MoE

C’est un mélange d’experts épars : 8 experts routés avec routage top-2 ; environ 39.1Md de paramètres sont actifs par token.

Comparaison champ à champ

Comparé au modèle précédent du même labo ; à défaut, au modèle structurellement le plus proche. La colonne ratio = ce modèle ÷ le modèle comparé.

Comparaison complète avec le prédécesseur Mistral 7B
ModèleMixtral 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

Architectures les plus proches

La similarité varie de 0 (aucune caractéristique catégorielle commune) à 1 (profils identiques).

Champs bruts du config.json

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