DeepSeek · 2024-05 · MoE épars

DeepSeek-V2

DeepSeek-V2 est un Transformer à attention latente multi-tête (MLA) publié par DeepSeek en 2024-05, avec 60 couches, une dimension cachée de 5120 et une fenêtre de contexte de 163,840 tokens.

Pile de couches

Attention ×60 · Couches 60

Faits clés

LaboratoireDeepSeek
Publication2024-05
Paramètres236 Md
Actifs21 Md
Contexte163,840 tokens
AttentionMLA · kv_lora_rank 512
Couches60
Dimension cachée5,120
Têtes d'attention128
Vocabulaire102,400
Encodage positionnelRoPE
NormalisationRMSNorm
Activationsilu
Précisionbf16
Classe d'architectureDeepseekV2ForCausalLM

Vue d'ensemble de l'architecture

t₁ t₂ t₃ t₄ tokens d’entrée Embedding · Vocabulaire 102,400 → Dimension cachée 5,120 Q · 128 Wᴰᴷⱽ ↓ c_KV · 512 kᴿ · RoPE 64 Wᵁᴷ/Wᵁⱽ ↑ ⊕kᴿ Attention Cache KV/token ≈ 512+64 contre 2·5,120 en dense → ≈ 1/17.8 Routeur E1 E2 E3 E4 E5 E6 E7 E8 +152… Expert partagé ×2 top-6 sur 160 experts · ≈ 21Md de paramètres actifs par token × 60 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 160K tok
Dessiné d’après le config.json livré · 60 couches / largeur 5,120 / contexte 163,840. Schéma original de cet atlas.

Attention

attention latente multi-tête (MLA). kv_lora_rank=512. RoPE θ=10,000.

FFN / MoE

C’est un mélange d’experts épars : 160 experts routés avec routage top-6 ; environ 21Md 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 DeepSeekMoE 16B
ModèleDeepSeek-V2DeepSeekMoE 16BRatio
model_typedeepseek_v2deepseek
architecturesDeepseekV2ForCausalLMDeepseekForCausalLM
hidden_size51202048×2.50
num_hidden_layers6028×2.14
num_attention_heads12816×8.00
num_key_value_heads12816×8.00
intermediate_size1228810944×1.12
hidden_actsilusilu
n_routed_experts16064×2.50
num_experts_per_tok66≈1
n_shared_experts22≈1
moe_intermediate_size15361408×1.09
kv_lora_rank512
qk_rope_head_dim64
q_lora_rank1536
rope_theta1000010000≈1
max_position_embeddings1638404096×40.00
rms_norm_eps0.0000010.000001≈1
vocab_size102400102400≈1
torch_dtypebfloat16bfloat16
attention_biasfalsefalse
attention_dropout00
aux_loss_alpha0.001
bos_token_id100000100000≈1
eos_token_id100001100001≈1
first_k_dense_replace11≈1

Architectures les plus proches

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

Champs bruts du config.json

41 champs
architecturesDeepseekV2ForCausalLM
attention_biasfalse
attention_dropout0
aux_loss_alpha0.001
bos_token_id100000
eos_token_id100001
first_k_dense_replace1
hidden_actsilu
hidden_size5120
initializer_range0.02
intermediate_size12288
kv_lora_rank512
max_position_embeddings163840
model_typedeepseek_v2
moe_intermediate_size1536
moe_layer_freq1
n_group8
n_routed_experts160
n_shared_experts2
norm_topk_probfalse
num_attention_heads128
num_experts_per_tok6
num_hidden_layers60
num_key_value_heads128
pretraining_tp1
q_lora_rank1536
qk_nope_head_dim128
qk_rope_head_dim64
rms_norm_eps0.000001
rope_theta10000
routed_scaling_factor16
scoring_funcsoftmax
seq_auxtrue
tie_word_embeddingsfalse
topk_group3
topk_methodgroup_limited_greedy
torch_dtypebfloat16
transformers_version4.39.3
use_cachetrue
v_head_dim128
vocab_size102400