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