DeepSeek · 2024-05 · 稀疏 MoE

DeepSeek-V2

DeepSeek-V2 是 DeepSeek 於 2024-05 發布的 多頭潛在注意力(MLA) Transformer,共 60 層,隱藏維度 5120,上下文視窗 163,840 個 token。

層堆疊

注意力 ×60 · 層數 60

關鍵參數

實驗室DeepSeek
發布時間2024-05
總參數236 B
激活參數21 B
上下文163,840 tokens
注意力MLA · kv_lora_rank 512
層數60
隱藏維度5,120
注意力頭128
詞表102,400
位置編碼RoPE
正規化RMSNorm
激活函數silu
訓練精度bf16
架構類別DeepseekV2ForCausalLM

架構概覽

t₁ t₂ t₃ t₄ 輸入 token Embedding · 詞表 102,400 → 隱藏維度 5,120 Q · 128 Wᴰᴷⱽ ↓ c_KV · 512 kᴿ · RoPE 64 Wᵁᴷ/Wᵁⱽ ↑ ⊕kᴿ Attention KV 缓存/token ≈ 512+64,稠密为 2·5,120 → 约 1/17.8 路由器 E1 E2 E3 E4 E5 E6 E7 E8 +152… 共享专家 ×2 160 专家选 top-6 · 每 token 约激活 21B 参数 × 60 Transformer 块 注意力 前饋 / MoE RMSNorm pre-norm 最終正規化 · RMSNorm LM head → 詞表 102,400 p p p → 下一 token 位置編碼 RoPE 訓練精度 BF16 上下文 160K tok
依據隨權重發布的 config.json 繪製 · 60 層 / 寬 5,120 / 上下文 163,840。本圖為本站原創示意圖。

注意力

多頭潛在注意力(MLA). kv_lora_rank=512. RoPE θ=10,000.

前饋 / MoE

它是稀疏混合專家模型:160 個路由專家、top-6 路由;每個 token 約激活 21B 參數。

欄位級對比

對比對象為同實驗室的上一代模型;無前代時取結構最相似的模型。倍數欄 = 本模型數值 ÷ 對比模型數值。

與前代 DeepSeekMoE 16B 的全欄位對比
模型DeepSeek-V2DeepSeekMoE 16B倍數
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

結構最相似的模型

相似度取值範圍 0(無共同類別特徵)到 1(結構輪廓完全一致)。

原始 config 欄位

41 個欄位
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