DeepSeek · 2024-05 · スパースMoE

DeepSeek-V2 Lite

DeepSeek-V2 Lite は DeepSeek が 2024-05 に公開した マルチヘッド潜在注意(MLA) Transformer です。27 層、隠れ次元 2048、コンテキスト長 163,840 トークン。

層スタック

注意機構 ×27 · 層数 27

主要スペック

開発元DeepSeek
公開日2024-05
総パラメータ15.6 B
活性パラメータ2.5 B
コンテキスト163,840 tokens
注意機構MLA · kv_lora_rank 512
層数27
隠れ次元2,048
注意ヘッド16
語彙数102,400
位置エンコーディングRoPE
正規化RMSNorm
活性化関数silu
学習精度bf16
アーキテクチャクラスDeepseekV2ForCausalLM

アーキテクチャ概要

t₁ t₂ t₃ t₄ 入力トークン Embedding · 語彙数 102,400 → 隠れ次元 2,048 Q · 16 Wᴰᴷⱽ ↓ c_KV · 512 kᴿ · RoPE 64 Wᵁᴷ/Wᵁⱽ ↑ ⊕kᴿ Attention KV キャッシュ/トークン ≈ 512+64(デンス 2·2,048 → 約 1/7.1) ルーター E1 E2 E3 E4 E5 E6 E7 E8 +56… 共有エキスパート ×2 64 エキスパートから top-6 · トークンごとに約 2.5B パラメータ活性 × 27 Transformer ブロック 注意機構 FFN / MoE RMSNorm pre-norm 最終正規化 · RMSNorm LM head → 語彙数 102,400 p p p → 次のトークン 位置エンコーディング RoPE 学習精度 BF16 コンテキスト 160K tok
同梱の config.json に基づき作図 · 27 層 / 隠れ次元 2,048 / コンテキスト 163,840。図は本サイトのオリジナルです。

注意機構

マルチヘッド潜在注意(MLA). kv_lora_rank=512. RoPE θ=10,000.

FFN / MoE

スパースなMixture-of-Expertsモデルで、64 個のルーティングエキスパートと top-6 ルーティングを持ち、トークンごとに約 2.5B のパラメータが活性化します。

フィールド単位の比較

同一開発元の前世代モデルと比較します。前世代がない場合は構造最相似モデルと比較。倍数欄 = 本モデル ÷ 比較対象。

前世代 DeepSeek-V2 との全フィールド比較
モデルDeepSeek-V2 LiteDeepSeek-V2倍数
model_typedeepseek_v2deepseek_v2
architecturesDeepseekV2ForCausalLMDeepseekV2ForCausalLM
hidden_size20485120×0.400
num_hidden_layers2760×0.450
num_attention_heads16128×0.125
num_key_value_heads16128×0.125
intermediate_size1094412288×0.891
hidden_actsilusilu
n_routed_experts64160×0.400
num_experts_per_tok66≈1
n_shared_experts22≈1
moe_intermediate_size14081536×0.917
kv_lora_rank512512≈1
qk_rope_head_dim6464≈1
q_lora_rank1536
rope_theta1000010000≈1
max_position_embeddings163840163840≈1
rms_norm_eps0.0000010.000001≈1
vocab_size102400102400≈1
torch_dtypebfloat16bfloat16
attention_biasfalsefalse
attention_dropout00
aux_loss_alpha0.0010.001≈1
bos_token_id100000100000≈1
eos_token_id100001100001≈1
first_k_dense_replace11≈1

構造が最も近いモデル

類似度は 0(共通のカテゴリ特徴なし)から 1(構造プロファイルが完全に一致)の範囲です。

config.json の全フィールド

40 フィールド
architecturesDeepseekV2ForCausalLM
attention_biasfalse
attention_dropout0
aux_loss_alpha0.001
bos_token_id100000
eos_token_id100001
first_k_dense_replace1
hidden_actsilu
hidden_size2048
initializer_range0.02
intermediate_size10944
kv_lora_rank512
max_position_embeddings163840
model_typedeepseek_v2
moe_intermediate_size1408
moe_layer_freq1
n_group1
n_routed_experts64
n_shared_experts2
norm_topk_probfalse
num_attention_heads16
num_experts_per_tok6
num_hidden_layers27
num_key_value_heads16
pretraining_tp1
qk_nope_head_dim128
qk_rope_head_dim64
rms_norm_eps0.000001
rope_theta10000
routed_scaling_factor1
scoring_funcsoftmax
seq_auxtrue
tie_word_embeddingsfalse
topk_group1
topk_methodgreedy
torch_dtypebfloat16
transformers_version4.33.1
use_cachetrue
v_head_dim128
vocab_size102400