一句话定位

昆仑万维天工的 146B MoE(16 专家),由 Skywork-13B 稠密 checkpoint upcycling 而来,提出 gating logit 归一化与自适应辅助损失系数两项训练技巧。

摘要

Skywork-MoE 是 146B 参数、16 专家的高性能 MoE LLM,由已有的 Skywork-13B 稠密 checkpoint 初始化(upcycling)。系统对比 upcycling 与从头训练两种初始化方式,结论是选择取决于已有稠密 checkpoint 质量与 MoE 训练预算。提出两项创新技巧:(1) gating logit normalization——提升专家多样化;(2) adaptive auxiliary loss coefficients——允许逐层调节辅助损失系数。实验验证两者有效。

关键技术细节(带数字)

  • 规模:146B 总参,16 个专家(top-2 激活,约 22B 激活)。
  • 初始化:从 Skywork-13B 稠密 checkpoint upcycling。
  • 技巧 1:gating logit normalization(门控 logit 归一化,提升专家差异化)。
  • 技巧 2:adaptive auxiliary loss coefficients(逐层自适应辅助损失系数)。
  • 研究结论:upcycling vs from-scratch 的选择取决于稠密底座质量与 MoE 训练预算。

原始链接

一手源存档(sources/)

  • skywork-moe.pdf (PDF 不入 git,走 HF bucket)