模型架构 / 专家与路由 原理笔记进阶 3 分钟阅读 查看源文件 ↗

DeepSeek-V4:路由评分与残差连接#

本页内容4 个部分

这篇笔记讨论 router 的打分尺度与残差连接。架构事实以 DeepSeek-V4 技术报告 为准;下面对函数的比较是数学分析,不能单独解释整个模型的性能变化。

查看概览图
ROUTER × RESIDUAL

Router 的尺度与排序,
残差流的混合与约束

把单调函数、选择 bias 与输出权重分开分析;函数无上界并不意味着路由必然更尖锐。

AFFINITYsi,t=SCORING φφ(ht⊤ei)

e_i 是第 i 个专家的 centroid / gate 向量(可学习);φ 改变打分尺度与混合权重;没有专家 bias 时,严格单调变换保持 top-K 排名。

V3Sigmoid

有上界,正数端趋近 1

hᵀeφ
  • 天花板 = 1
    饱和
  • 负数端落在 (0, 0.5)z → −∞ 时约为 exp(z)
换尺度 φ: Sigmoid
→ √Softplus
V4Sqrt(Softplus)

无上界,正数端约为 √z

hᵀeφ
  • 无上界
    改尺度
  • 负数端非常接近 0 但不为 0z → −∞ 时约为 exp(z/2),衰减比 sigmoid 慢
01选专家

按选择分数取 top-K

带负载均衡 bias 时,按 s + b 选择。V4-Pro 为 384 选 6;hash-routed 层另有机制。

02加权

以 si,t 作 gating 权重

选中专家使用未加 bias 的 s 归一化加权,再乘 routed scaling factor;硬稀疏性来自 top-K mask。

归一化位置优点缺点
Pre-Norm
x + F(Norm(x))
保留直接残差梯度路径关注深层表示变化;不保证所有梯度不消失
Post-Norm
Norm(x + F(x))
残差路径经过归一化初始化与学习率更敏感,不是必然梯度消失
HYPER-CONNECTIONS
扩展并约束残差混合

Hyper-Connections 扩展残差流;mHC 约束连接映射,兼顾表示能力与多层传播稳定性。

V3 · φ = SigmoidV4 · φ = √SoftplusV4-Pro · 384 选 6

1. Router:选择分数与混合权重分开看#

对隐藏态 hth_t 和可学习的专家向量 eie_i,先计算

zi,t=ht⊤ei,si,t=ϕ(zi,t).z_{i,t}=h_t^\top e_i,\qquad s_{i,t}=\phi(z_{i,t}).

V3 使用 sigmoid;V4 使用 softplus⁡(z)=log⁡(1+ez)\sqrt{\operatorname{softplus}(z)}=\sqrt{\log(1+e^z)}。Router 还包括负载均衡 bias、候选专家限制、hash routing 等机制。以带校正 bias 的路由为例:

It=TopK⁡i(si,t+bi),gi,t=αsi,t∑j∈Itsj,t(i∈It).\mathcal I_t=\operatorname{TopK}_i(s_{i,t}+b_i),\qquad g_{i,t}=\alpha\frac{s_{i,t}}{\sum_{j\in\mathcal I_t}s_{j,t}} \quad(i\in\mathcal I_t).

bib_i 用于选择,混合权重仍由未加 bias 的 sis_i 归一化得到;α\alpha 是 routed scaling factor。这个表达式不描述 hash-routed 层。V4-Pro 的公开配置为 384 个 routed experts、每 token 激活 6 个,另有 shared expert;不能套用 V3 的“256 选 8”。

2. 更换单调函数,不会自动改变 top-K 排名#

两个函数都严格单调递增。因此,对同一组 logits,若不加入专家相关 bias 或其他选择约束,二者的 top-K 集合相同(忽略有限精度并列)。函数变化会影响相对尺度、归一化权重及梯度;加入 bib_i 后,尺度与 bias 的相互作用才可能改变选择。

正数端:

σ(z)→1,log⁡(1+ez)∼z(z→+∞).\sigma(z)\to1,\qquad \sqrt{\log(1+e^z)}\sim\sqrt z \quad(z\to+\infty).

后者没有输出上界,但不能据此说路由必然更“尖锐”,需要比较具体 logits 下归一化后的权重。

负数端:

σ(z)∼ez,log⁡(1+ez)∼ez/2(z→−∞).\sigma(z)\sim e^z,\qquad \sqrt{\log(1+e^z)}\sim e^{z/2} \quad(z\to-\infty).

所以 sqrt-softplus 衰减得更慢,不是更强地抑制负 logits。例如 z=−10z=-10 时,两者约为 4.54×10−54.54\times10^{-5} 和 6.74×10−36.74\times10^{-3}。专家输出的硬稀疏性来自 top-K mask,而非这两个始终为正的函数。

3. Hyper-Connections:扩展残差路径#

归一化位置 梯度路径 需要关注的现象
Pre-Norm:x+F(Norm⁡(x))x+F(\operatorname{Norm}(x)) 保留直接的残差梯度路径 层间更新的相对大小、深层表示变化;不保证所有参数梯度都不消失
Post-Norm:Norm⁡(x+F(x))\operatorname{Norm}(x+F(x)) 残差路径也经过归一化 Jacobian 初始化、学习率和 warmup 更敏感;并非必然梯度消失

Hyper-Connections 扩展残差流及其混合方式;V4 使用的 mHC 进一步约束连接映射,以控制多层传播的稳定性。它不能简化成“Pre-Norm 必坍缩,Post-Norm 必消失”。具体映射与约束见 mHC 笔记。

相关资料:DeepSeek-V3、On Layer Normalization in the Transformer Architecture。

延伸阅读#

相关阅读路线

理解 Transformer 与架构变体 ↗