MiMo-V2-Flash 技术报告

MiMo-V2-Flash Technical Report

小米 MiMo 团队 Xiaomi MiMo Team · Xiaomi · 2026-01-06 · arXiv:2601.02780 ↗ · 被引 99

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了 MiMo-V2-Flash,一种混合专家(MoE)模型,总参数量 309B,激活参数量 15B,专为快速、强推理和智能体能力而设计。MiMo-V2-Flash 采用混合注意力架构,交错使用滑动窗口注意力(SWA)和全局注意力,滑动窗口大小为 128 个 token,混合比例为 5:1。该模型在 27 万亿个 token 上使用多 token 预测(MTP)进行预训练,原生支持 32k 上下文长度,随后扩展至 256k。为了高效扩展训练后计算,MiMo-V2-Flash 引入了一种新颖的多教师同策略蒸馏(MOPD)范式。在该框架中,领域专家教师(例如通过大规模强化学习训练)提供密集的 token 级奖励,使学生模型能够完美掌握教师专业知识。尽管 MiMo-V2-Flash 的总参数量仅为 DeepSeek-V3.2 和 Kimi-K2 的 1/2 和 1/3,但其性能可与这些顶级开源模型相媲美。在推理过程中,通过将 MTP 重新用作投机解码的草稿模型,MiMo-V2-Flash 在三个 MTP 层下实现了高达 3.6 的接受长度和 2.6 倍的解码加速。我们开源了模型权重和三层 MTP 权重,以促进开放研究和社区合作。

We present MiMo-V2-Flash, a Mixture-of-Experts (MoE) model with 309B total parameters and 15B active parameters, designed for fast, strong reasoning and agentic capabilities. MiMo-V2-Flash adopts a hybrid attention architecture that interleaves Sliding Window Attention (SWA) with global attention, with a 128-token sliding window under a 5:1 hybrid ratio. The model is pre-trained on 27 trillion tokens with Multi-Token Prediction (MTP), employing a native 32k context length and subsequently extended to 256k. To efficiently scale post-training compute, MiMo-V2-Flash introduces a novel Multi-Teacher On-Policy Distillation (MOPD) paradigm. In this framework, domain-specialized teachers (e.g., trained via large-scale reinforcement learning) provide dense and token-level reward, enabling the student model to perfectly master teacher expertise. MiMo-V2-Flash rivals top-tier open-weight models such as DeepSeek-V3.2 and Kimi-K2, despite using only 1/2 and 1/3 of their total parameters, respectively. During inference, by repurposing MTP as a draft model for speculative decoding, MiMo-V2-Flash achieves up to 3.6 acceptance length and 2.6x decoding speedup with three MTP layers. We open-source both the model weights and the three-layer MTP weights to foster open research and community collaboration.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 18)

阅读逐段中英对照全文 →