DeepSeek · 2024-01 · 稀疏 MoE

DeepSeekMoE 16B

DeepSeekMoE 16B 是 DeepSeek 于 2024-01 发布的 多头注意力(MHA) Transformer,共 28 层,隐藏维度 2048,上下文窗口 4,096 个 token。

层堆栈

注意力 ×28 · 层数 28

关键参数

实验室DeepSeek
发布时间2024-01
总参数16.2 B
激活参数2.6 B
上下文4,096 tokens
注意力MHA (16:16)
层数28
隐藏维度2,048
注意力头16
词表102,400
位置编码RoPE
归一化RMSNorm
激活函数silu
训练精度bf16
架构类名DeepseekForCausalLM

架构概览

t₁ t₂ t₃ t₄ 输入 token Embedding · 词表 102,400 → 隐藏维度 2,048 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn 注意力头 16 KV 头 16 head dim 128 · RoPE 路由器 E1 E2 E3 E4 E5 E6 E7 E8 +56… 共享专家 ×2 64 专家选 top-6 · 每 token 约激活 2.6B 参数 × 28 Transformer 块 注意力 前馈 / MoE RMSNorm pre-norm 最终归一化 · RMSNorm LM head → 词表 102,400 p p p → 下一 token 位置编码 RoPE 训练精度 BF16 上下文 4K tok
依据随权重发布的 config.json 绘制 · 28 层 / 宽 2,048 / 上下文 4,096。本图为本站原创示意图。

注意力

多头注意力(MHA) — 16 q-heads / 16 kv-heads. RoPE θ=10,000.

前馈 / MoE

它是稀疏混合专家模型:64 个路由专家、top-6 路由;每个 token 约激活 2.6B 参数。

字段级对比

对比对象为同实验室的上一代模型;无前代时取结构最相似的模型。倍数列 = 本模型数值 ÷ 对比模型数值。

与前代 DeepSeek LLM 7B 的全字段对比
模型DeepSeekMoE 16BDeepSeek LLM 7B倍数
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

结构最相似的模型

相似度取值范围 0(无共同类别特征)到 1(结构画像完全一致)。

原始 config 字段

30 个字段
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