Qwen · 2024-06 · 稠密

Qwen2 72B

Qwen2 72B 是 Qwen 于 2024-06 发布的 分组查询注意力(GQA) Transformer,共 80 层,隐藏维度 8192,上下文窗口 32,768 个 token。

层堆栈

注意力 ×80 · 层数 80

关键参数

实验室Qwen
发布时间2024-06
总参数71.4 B
上下文32,768 tokens
注意力GQA (64:8)
层数80
隐藏维度8,192
注意力头64
词表152,064
位置编码RoPE
归一化RMSNorm
激活函数silu
训练精度bf16
架构类名Qwen2ForCausalLM

架构概览

t₁ t₂ t₃ t₄ 输入 token Embedding · 词表 152,064 → 隐藏维度 8,192 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn 注意力头 64 KV 头 8 head dim 128 · RoPE FFN · SwiGLU FFN 维度 29,568 × 80 Transformer 块 注意力 前馈 / MoE RMSNorm pre-norm 最终归一化 · RMSNorm LM head → 词表 152,064 p p p → 下一 token 位置编码 RoPE 训练精度 BF16 上下文 32K tok
依据随权重发布的 config.json 绘制 · 80 层 / 宽 8,192 / 上下文 32,768。本图为本站原创示意图。

注意力

分组查询注意力(GQA) — 64 q-heads / 8 kv-heads. RoPE θ=1,000,000.

前馈 / MoE

它是稠密模型:全部 71.4B 参数都参与每个 token 的计算。

字段级对比

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

与前代 Qwen1.5 7B 的全字段对比
模型Qwen2 72BQwen1.5 7B倍数
model_typeqwen2qwen2
architecturesQwen2ForCausalLMQwen2ForCausalLM
hidden_size81924096×2.00
num_hidden_layers8032×2.50
num_attention_heads6432×2.00
num_key_value_heads832×0.250
intermediate_size2956811008×2.69
hidden_actsilusilu
rope_theta10000001000000≈1
max_position_embeddings3276832768≈1
sliding_window13107232768×4.00
rms_norm_eps0.0000010.000001≈1
vocab_size152064151936≈1
torch_dtypebfloat16bfloat16
attention_dropout00
bos_token_id151643151643≈1
eos_token_id151645151643≈1
initializer_range0.020.02≈1
max_window_layers8028×2.86
tie_word_embeddingsfalsefalse
transformers_version4.40.14.37.0
use_cachetruetrue
use_sliding_windowfalsefalse

结构最相似的模型

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

原始 config 字段

23 个字段
architecturesQwen2ForCausalLM
attention_dropout0
bos_token_id151643
eos_token_id151645
hidden_actsilu
hidden_size8192
initializer_range0.02
intermediate_size29568
max_position_embeddings32768
max_window_layers80
model_typeqwen2
num_attention_heads64
num_hidden_layers80
num_key_value_heads8
rms_norm_eps0.000001
rope_theta1000000
sliding_window131072
tie_word_embeddingsfalse
torch_dtypebfloat16
transformers_version4.40.1
use_cachetrue
use_sliding_windowfalse
vocab_size152064