DeepSeek · 2024-01 · MoE épars

DeepSeekMoE 16B

DeepSeekMoE 16B est un Transformer à attention multi-tête (MHA) publié par DeepSeek en 2024-01, avec 28 couches, une dimension cachée de 2048 et une fenêtre de contexte de 4,096 tokens.

Pile de couches

Attention ×28 · Couches 28

Faits clés

LaboratoireDeepSeek
Publication2024-01
Paramètres16.2 Md
Actifs2.6 Md
Contexte4,096 tokens
AttentionMHA (16:16)
Couches28
Dimension cachée2,048
Têtes d'attention16
Vocabulaire102,400
Encodage positionnelRoPE
NormalisationRMSNorm
Activationsilu
Précisionbf16
Classe d'architectureDeepseekForCausalLM

Vue d'ensemble de l'architecture

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

Attention

attention multi-tête (MHA) — 16 q-heads / 16 kv-heads. RoPE θ=10,000.

FFN / MoE

C’est un mélange d’experts épars : 64 experts routés avec routage top-6 ; environ 2.6Md 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 DeepSeek LLM 7B
ModèleDeepSeekMoE 16BDeepSeek LLM 7BRatio
model_typedeepseekllama
architecturesDeepseekForCausalLMLlamaForCausalLM
hidden_size20484096×0.500
num_hidden_layers2830×0.933
num_attention_heads1632×0.500
num_key_value_heads1632×0.500
intermediate_size1094411008×0.994
hidden_actsilusilu
n_routed_experts64
num_experts_per_tok6
n_shared_experts2
moe_intermediate_size1408
rope_theta1000010000≈1
max_position_embeddings40964096≈1
rms_norm_eps0.0000010.000001≈1
vocab_size102400102400≈1
torch_dtypebfloat16bfloat16
attention_biasfalse
attention_dropout0
bos_token_id1000001×100000.00
eos_token_id1000012×50000.50
first_k_dense_replace1
initializer_range0.020.02≈1
moe_layer_freq1
norm_topk_probfalse
pretraining_tp11≈1

Architectures les plus proches

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

Champs bruts du config.json

30 champs
architecturesDeepseekForCausalLM
attention_biasfalse
attention_dropout0
bos_token_id100000
eos_token_id100001
first_k_dense_replace1
hidden_actsilu
hidden_size2048
initializer_range0.02
intermediate_size10944
max_position_embeddings4096
model_typedeepseek
moe_intermediate_size1408
moe_layer_freq1
n_routed_experts64
n_shared_experts2
norm_topk_probfalse
num_attention_heads16
num_experts_per_tok6
num_hidden_layers28
num_key_value_heads16
pretraining_tp1
rms_norm_eps0.000001
rope_theta10000
scoring_funcsoftmax
tie_word_embeddingsfalse
torch_dtypebfloat16
transformers_version4.36.0
use_cachetrue
vocab_size102400