DeepSeek-V4:路由评分与残差连接#
本页内容4 个部分
这篇笔记讨论 router 的打分尺度与残差连接。架构事实以 DeepSeek-V4 技术报告 为准;下面对函数的比较是数学分析,不能单独解释整个模型的性能变化。
查看概览图
Router 的尺度与排序,
残差流的混合与约束
把单调函数、选择 bias 与输出权重分开分析;函数无上界并不意味着路由必然更尖锐。
e_i 是第 i 个专家的 centroid / gate 向量(可学习);φ 改变打分尺度与混合权重;没有专家 bias 时,严格单调变换保持 top-K 排名。
有上界,正数端趋近 1
-
饱和
天花板 = 1 -
负数端落在 (0, 0.5)z → −∞ 时约为 exp(z)
→ √Softplus
无上界,正数端约为 √z
-
改尺度
无上界 -
负数端非常接近 0 但不为 0z → −∞ 时约为 exp(z/2),衰减比 sigmoid 慢
按选择分数取 top-K
带负载均衡 bias 时,按 s + b 选择。V4-Pro 为 384 选 6;hash-routed 层另有机制。
以 si,t 作 gating 权重
选中专家使用未加 bias 的 s 归一化加权,再乘 routed scaling factor;硬稀疏性来自 top-K mask。
x + F(Norm(x))保留直接残差梯度路径关注深层表示变化;不保证所有梯度不消失Norm(x + F(x))残差路径经过归一化初始化与学习率更敏感,不是必然梯度消失Hyper-Connections 扩展残差流;mHC 约束连接映射,兼顾表示能力与多层传播稳定性。
1. Router:选择分数与混合权重分开看#
对隐藏态 和可学习的专家向量 ,先计算
V3 使用 sigmoid;V4 使用 。Router 还包括负载均衡 bias、候选专家限制、hash routing 等机制。以带校正 bias 的路由为例:
用于选择,混合权重仍由未加 bias 的 归一化得到; 是 routed scaling factor。这个表达式不描述 hash-routed 层。V4-Pro 的公开配置为 384 个 routed experts、每 token 激活 6 个,另有 shared expert;不能套用 V3 的“256 选 8”。
2. 更换单调函数,不会自动改变 top-K 排名#
两个函数都严格单调递增。因此,对同一组 logits,若不加入专家相关 bias 或其他选择约束,二者的 top-K 集合相同(忽略有限精度并列)。函数变化会影响相对尺度、归一化权重及梯度;加入 后,尺度与 bias 的相互作用才可能改变选择。
正数端:
后者没有输出上界,但不能据此说路由必然更“尖锐”,需要比较具体 logits 下归一化后的权重。
负数端:
所以 sqrt-softplus 衰减得更慢,不是更强地抑制负 logits。例如 时,两者约为 和 。专家输出的硬稀疏性来自 top-K mask,而非这两个始终为正的函数。
3. Hyper-Connections:扩展残差路径#
| 归一化位置 | 梯度路径 | 需要关注的现象 |
|---|---|---|
| Pre-Norm: | 保留直接的残差梯度路径 | 层间更新的相对大小、深层表示变化;不保证所有参数梯度都不消失 |
| Post-Norm: | 残差路径也经过归一化 Jacobian | 初始化、学习率和 warmup 更敏感;并非必然梯度消失 |
Hyper-Connections 扩展残差流及其混合方式;V4 使用的 mHC 进一步约束连接映射,以控制多层传播的稳定性。它不能简化成“Pre-Norm 必坍缩,Post-Norm 必消失”。具体映射与约束见 mHC 笔记。
相关资料:DeepSeek-V3、On Layer Normalization in the Transformer Architecture。
延伸阅读#
- 稀疏注意力:选择与压缩机制比较:不同稀疏方案如何选择参与注意力计算的信息?