DeepSeek · 2024-01 · MoE sparsa

DeepSeekMoE 16B

DeepSeekMoE 16B è un Transformer a attention multi-head (MHA) pubblicato da DeepSeek nel 2024-01, con 28 strati, dimensione nascosta 2048 e una finestra di contesto di 4,096 token.

Pila di strati

Attention ×28 · Strati 28

Dati chiave

LaboratorioDeepSeek
Pubblicazione2024-01
Parametri16.2 mld
Attivi2.6 mld
Contesto4,096 tokens
AttentionMHA (16:16)
Strati28
Dim. nascosta2,048
Teste di attention16
Vocabolario102,400
Codifica posizionaleRoPE
NormalizzazioneRMSNorm
Attivazionesilu
Precisionebf16
Classe architetturaleDeepseekForCausalLM

Panoramica dell'architettura

t₁ t₂ t₃ t₄ token di ingresso Embedding · Vocabolario 102,400 → Dim. nascosta 2,048 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn Teste di attention 16 Teste KV 16 head dim 128 · RoPE Router E1 E2 E3 E4 E5 E6 E7 E8 +56… Esperto condiviso ×2 top-6 su 64 esperti · ≈ 2.6mld parametri attivi per token × 28 blocco transformer Attention FFN / MoE RMSNorm pre-norm Normalizzazione finale · RMSNorm LM head → Vocabolario 102,400 p p p → token successivo Codifica posizionale RoPE Precisione BF16 Contesto 4K tok
Disegnato dal config.json distribuito · 28 strati / larghezza 2,048 / contesto 4,096. Schema originale di questo atlante.

Attention

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

FFN / MoE

È un mixture-of-experts sparso con 64 esperti instradati e routing top-6; circa 2.6 milioni di parametri sono attivi per token.

Confronto campo per campo

Confronto con il modello precedente dello stesso laboratorio; in assenza, col modello strutturalmente più vicino. La colonna rapporto = questo modello ÷ modello di confronto.

Confronto completo con il predecessore DeepSeek LLM 7B
ModelloDeepSeekMoE 16BDeepSeek LLM 7BRapporto
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

Architetture più simili

I valori di similarità vanno da 0 (nessuna caratteristica categoriale condivisa) a 1 (profili identici).

Campi grezzi del config.json

30 campi
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