Moonshot AI 发布 Moonlight 模型及 Muon 优化器全套开源方案
Moonshot AI Releases Moonlight Model and Complete Muon Optimizer Open-Source Package
EVENT SUMMARY
事件摘要
2025年3月3日,Moonshot AI(月之暗面)同步发布了 Moonlight 系列模型及配套 Muon 优化器的完整开源方案。核心产品 Moonlight 是基于 Muon 优化器训练的 16B 总参数/3B 激活参数 MoE 模型,训练数据量达 5.7T tokens,采用与 DeepSeek-V3 相同的架构,兼容 VLLM 和 SGLang 等主流推理引擎,同时提供预训练版本(Moonlight)和指令微调版本(Moonlight-Instruct),上下文长度均为 8K。技术层面,团队公开了两项关键可扩展性改进:引入权重衰减和精确调整参数级更新比例以保持 RMS 一致性,使 Muon 可直接用于大规模训练而无需额外超参数调优,声称相比 AdamW 仅需约 52% 计算量即可达到同等性能。工程层面,开源了基于 ZeRO-1 风格的分布式 Muon 实现,在保持算法数学特性的同时实现最优内存效率和更低通信开销。此外,还罕见地发布了 Moonlight 和 Moonlight-A(AdamW 对照版本)的中间检查点,供研究者分析不同优化器在训练过程中的动态差异(如 SVD Entropy 和 Srank 指标变化)。完整技术报告已上传 arXiv(2502.16982),代码开源于 GitHub。
ANALYST VIEW
中立分析
为什么重要
此次发布从模型、算法改进、分布式工程实现到中间检查点形成了完整的开源研究闭环,展示了 Muon 优化器在大规模 LLM 训练中的可行性。若 Muon 在更大规模上持续验证有效,可能挑战 AdamW 作为默认训练优化器的行业地位,推动训练成本降低,并加速替代优化器研究的开源生态发展。
影响对象
面向 AI 研究人员和训练工程师:提供可直接使用的开源 MoE 模型和经过验证的 Muon 优化器改进方案;面向训练基础设施开发者:提供可参考的分布式 Muon 实现,降低工程复现门槛;面向优化器研究者:提供罕见的中间检查点数据,降低训练动态对比研究的复现成本。
证据说明
来源直接证明 Moonshot AI 发布了 Moonlight 模型权重、开源代码、分布式 Muon 实现和中间检查点;技术报告已上传 arXiv(2502.16982),GitHub 仓库(github.com/MoonshotAI/Moonlight)包含训练示例命令,均为公开可验证的学术与工程发布。
不确定性
Muon 优化器在更大规模(如 100B+ 参数)训练中的表现尚未验证;'约 2 倍计算效率提升'的结论在不同模型规模、架构和数据分布下的泛化性仍需进一步验证;分布式实现在不同 GPU 拓扑和集群规模下的性能表现未详细披露;与 DeepSpeed、Megatron 等现有分布式训练框架的集成程度未知;实际推理效率和部署成本未详细说明。
SOURCES
1 个来源 · 6 条记录
以下记录均来自同一来源,不构成多来源相互印证。
来源原文:我们将 Moonlight 与同等规模的最先进公开模型进行了对比:LLAMA3-3B 是一个使用 9 万亿 tokens 训练的 30 亿参数密集模型;Qwen2.5-3B 是一个使用 18 万亿 tokens 训练的 30 亿参数密集模型;Deepseek-v2-Lite 是一个使用 5.7 万亿 tokens 训练的 2.4 亿/ 160 亿参数混合专家模型。
来源原文:我们进行了扩展性研究,将 Muon 与 AdamW 的高性能基准进行对比,结果显示了Muon 的卓越性能。根据 Scaling Law 结果,Muon 在仅使用约 52% 训练计算量的情况下,就达到了与AdamW 相当的性能。查看技术报告:Muon is Scalable for LLM Training https://arxiv.org/abs/2502.16982
来源原文:我们已经发布了Moonlight和Moonlight-A的中间检查点(checkpoint)以支持进一步的研究工作。一些亮点:我们可以看到使用 Muon 训练的 Moonlight 模型在数学和编程方面的表现优于使用AdamW训练的Moonlight-A。我们还可以分析 Moonlight 和 Moonlight-A 中间检查点在 SVD Entropy 和 Srank 方面的不同表现。
来源原文:我们开源了内存优化且通信高效的分布式 Muon 实现。同时,我们也发布了经过预训练、指令微调的模型检查点(checkpoints)以及中间检查点,以支持后续研究。相关代码已在 https://github.com/MoonshotAI/Moonlight 仓库开源。
来源原文:我们发现了两个提升 Muon 可扩展性的关键技术:(1)引入权重衰减(weight decay);(2)精确调整每个参数的更新比例。有了这些改进,Muon可以直接用于大规模训练,无需额外的超参数调优。扩展性实验表明,相比 AdamW,在计算量最优的训练条件下,Muon 的计算效率上实现了约 2 倍提升。
来源原文:基于这些改进,我们推出了 Moonlight 模型,这是一个使用 Muon 训练的 3B/16B 参数的混合专家模型(MoE),训练数据量达 5.7T tokens。我们的模型推进了当前的帕累托前沿(Pareto frontier),与现有模型相比,用更少的训练计算量实现了更好的性能。