Skip to content

Kimi K2: Open Agentic Intelligence ​

原标题: Kimi K2: Open Agentic Intelligence 中文标题: Kimi K2:开放的智能体智能

一句话摘要 ​

月之暗面发布了万亿参数级别的开源 MoE 大模型 Kimi K2,通过创新的 MuonClip 优化器实现了零损失尖峰的稳定训练,并凭借大规模智能体数据合成管线和联合强化学习,在代码生成、工具调用和多步推理等智能体任务上取得了开源模型的最佳表现。


完整核心内容翻译 ​

模型架构 ​

Kimi K2 是一个1.04 万亿参数的混合专家(MoE)Transformer 模型,但每个 token 仅激活约 320 亿参数,实现了极高的计算效率。

核心架构参数:

参数数值
总参数量~1.04 万亿
每 token 激活参数~320 亿
Transformer 层数61 层
隐藏维度7,168
专家数量384 个
每 token 激活专家数8 个
专家前馈维度2,048
注意力头数64
词表大小~160K tokens
激活函数SwiGLU
稀疏因子48x

模型采用多头潜在注意力(Multi-Head Latent Attention, MLA) 机制,通过对注意力头进行分组来降低计算开销,支持高效的长上下文处理。上下文窗口通过渐进式训练从 4K 扩展到 128K token,使用自研的 YaRN 位置编码扩展方法。

MuonClip 优化器——万亿级训练的"定海神针" ​

MuonClip 是本项目最重要的工程创新之一,它将高 token 效率的 Muon 优化器与一种全新的稳定性增强机制相结合。

核心问题: 在万亿参数规模的训练中,注意力层的 QK 点积容易产生"爆炸性 logits",导致训练过程出现损失尖峰(loss spike),轻则浪费大量算力重启训练,重则导致模型彻底崩溃。传统做法需要手动监控和干预,代价高昂。

QK-Clip 机制: 当注意力 logits 超过阈值(τ ≈ 100)时,MuonClip 自动重新缩放对应的 Query/Key 权重,将最大点积限制在安全范围内。整个过程无需人工干预,logits 会在训练过程中逐步自然回落到正常范围。

实际效果: 在 15.5 万亿 token 的完整预训练过程中实现了零损失尖峰,避免了可能价值数百万美元的重训练开销。

三阶段预训练课程 ​

  1. 第一阶段: 约 10 万亿 token,4,096 token 上下文长度,恒定学习率(2e-4)
  2. 第二阶段: 约 5.5 万亿 token,同样的上下文长度,学习率衰减至 2e-5
  3. 第三阶段(长上下文扩展): 4,000 亿 token @ 4K 上下文 → 600 亿 token @ 32K 上下文,通过渐进式 YaRN 扩展至 128K

训练数据涵盖大规模多语言文本、代码和数学内容,并通过合成数据改写管线(synthetic data rephrasing pipeline)来扩充高质量内容的可用量。

后训练:从"会说话"到"会做事" ​

监督微调阶段: 覆盖问答、摘要、编程提示和工具使用示例等基础能力。

智能体数据合成管线——最具差异化的创新:

团队构建了一套精密的数据合成系统,从 GitHub 采集超过 3,000 个真实工具,并额外合成约 20,000 个工具定义,然后通过算法将工具集与任务描述配对,模拟多步执行过程,生成数万条高质量的智能体训练样本。这使 K2 获得了原生的多步规划和工具调用能力,而非简单的指令微调。

联合强化学习阶段: 结合客观奖励(单元测试通过率、数学题正确率)与基于评分标准的自我批评(self-critique)机制,模型在递减温度下对自身输出进行评分,实现迭代式能力提升。

基准测试结果 ​

代码生成领域(最强表现):

基准测试Kimi K2GPT-4.1差距
LiveCodeBench53.7%44.7%+9.0
SWE-Bench Verified(智能体模式)65.8%54.6%+11.2
OJBench27.1%19.5%+7.6
HumanEval73.2%——
MultiPL-E85.7%86.7%-1.0

高级推理:

基准测试Kimi K2对比
Humanity's Last Exam(带工具)44.9%开源最高
Humanity's Last Exam(无工具)22.3%开源最高
BrowseComp(开放域检索)60.2%远超人类基线 29.2%
MMLU78.6%—

实际能力亮点:

  • K2 Thinking 模式可在单次会话中自主执行多达 300 步连续工具调用而不偏离目标
  • 一次生成完整可运行的太空侵略者游戏,仅用约 3,500 token
  • 完成复杂软件迭代任务的计算成本约 $0.07,相比同类闭源模型约 $3

量化与部署 ​

K2 支持原生 INT4 量化(通过量化感知训练实现),可将模型从 500+ GB 压缩至约 245 GB,保留约 85% 的精度。研究者已成功在两块 Apple M3 Ultra GPU(各 48GB) 上以约 15 token/s 的速度运行 K2。


技术要点 ​

  1. MuonClip 优化器:将 Muon 优化器与 QK-Clip 机制结合,实现万亿参数模型的零损失尖峰训练,解决了大规模 MoE 训练中最棘手的工程问题。

  2. 384 专家架构:超越 DeepSeek 的 256 专家设计,验证了更高的专家数量在稀疏性约束下能提升模型表达力,每 token 仅激活 8 个专家实现 48 倍稀疏率。

  3. 智能体数据合成管线:结合 3,000+ 真实工具和 20,000+ 合成工具的大规模多步轨迹生成,赋予模型原生的规划和工具使用能力。

  4. 渐进式长上下文训练:4K → 32K → 128K 的三阶段课程式上下文扩展策略,在保持预训练稳定性的同时实现前沿上下文窗口。

  5. 消费级硬件部署:原生 INT4 量化使万亿参数模型可在双 M3 Ultra 上运行,打破了大模型对云端推理的依赖。


深度解读 ​

Kimi K2 的发布标志着开源大模型竞争进入了一个新阶段:万亿参数、智能体原生、消费级可部署。

从"对话模型"到"智能体模型"的范式转换: K2 最深远的影响不在于参数量的堆叠,而在于它将"智能体能力"从后天微调提升为先天设计。通过在后训练阶段大规模注入工具调用轨迹,K2 不是一个"被教会使用工具的对话模型",而是一个"天生就会做事的智能体模型"。300 步连续工具调用的稳定性就是最好的证明。

MuonClip 的工程意义: 在万亿参数规模的训练中,一次损失尖峰就可能浪费数百万美元的算力。MuonClip 通过一个简洁的 QK 截断机制,将这个风险降为零。这不是理论上的优雅,而是实实在在的工程突破——它直接决定了月之暗面是否有能力完成这个规模的训练。

开源策略的深层逻辑: K2 采用修改版 MIT 许可证开源,定价仅为 Claude 的五分之一。这不仅是技术自信的表现,更是一种生态战略——通过低价开源吸引开发者,将 K2 打造为智能体应用的基础设施。


延伸思考 ​

  1. 稀疏度 vs 表达力的天花板: K2 使用 384 专家、48 倍稀疏率。稀疏度还能继续提升吗?当每 token 只激活总参数的 2% 时,专家间的知识共享是否会成为瓶颈?

  2. 智能体数据合成的可扩展性: 用合成工具和模拟环境生成训练数据是优雅的方案,但合成数据与真实世界之间的差距(sim-to-real gap)在智能体场景中会如何表现?

  3. 消费级部署的实际意义: 虽然 K2 可以在 M3 Ultra 上运行,但 15 token/s 的速度对于实际智能体应用(可能需要数百步交互)来说是否足够?瓶颈可能不在模型推理,而在工具调用的延迟。

  4. 与 K1.5 的技术传承: K1.5 的核心洞察(简洁 RL 框架、长上下文扩展)在 K2 中得到了继承和放大。K3 是否会在此基础上进一步扩展 RL 的角色,例如让模型在部署后持续通过 RL 学习?


原文链接 ​


发布时间:2025年7月16日 | 作者:Kimi Team | 架构:1.04T MoE, 32B 激活参数

全球顶尖 AI 实验室博客的中文翻译与深度解读 · 210+ 篇文章 · 15+ 实验室