本模型仓库采用 MIT 许可证授权。
目前,大多数成功的强化学习(RL)研究成果,包括开源项目,都依赖于规模相对较大的基础模型,例如 32B 模型,尤其在提升代码推理能力方面。此外,业界普遍认为,在小模型中同时实现数学和代码能力的均衡且同步提升颇具挑战。然而,我们认为,经 RL 训练的推理模型其效果取决于基础模型本身蕴含的推理潜能。要充分释放语言模型的推理潜能,不仅需要关注后训练阶段,还需制定针对推理任务的预训练策略。
在本研究中,我们提出了 MiMo-7B 系列模型,这是一系列从零开始训练、专为推理任务设计的模型。基于 MiMo-7B-Base 进行的 RL 实验表明,我们的模型具备卓越的推理潜能,甚至超越了规模大得多的 32B 模型。此外,我们在冷启动的 SFT 模型上进行了 RL 训练,得到了 MiMo-7B-RL。该模型在数学和代码推理任务上均表现出优异性能,可与 OpenAI o1-mini 相媲美。
我们开源了 MiMo-7B 系列,包括基础模型、SFT 模型、基于基础模型训练的 RL 模型以及基于 SFT 模型训练的 RL 模型的 checkpoint。我们相信本报告及所开源的模型将为开发强大的推理型大语言模型提供宝贵见解,造福更广泛的社区。
预训练:为推理而生的基础模型
训练后方案:开创推理模型新范式
强化学习基础设施
MiMo-7B 的 MTP 层在预训练和 SFT 期间进行调优,并在 RL 期间冻结。使用一个 MTP 层进行投机解码时,接受率约为 90%。
模型可在 https://huggingface.co/XiaomiMiMo 和 https://www.modelscope.cn/organization/XiaomiMiMo 获取
| 模型 | 描述 | 下载(HuggingFace) | 下载(ModelScope) |
|---|---|---|---|
| MiMo-7B-Base | 具有卓越推理潜力的基础模型 | 🤗 XiaomiMiMo/MiMo-7B-Base | 🤖️ XiaomiMiMo/MiMo-7B-Base |
| MiMo-7B-RL-Zero | 基于基础模型训练的 RL 模型 | 🤗 XiaomiMiMo/MiMo-7B-RL-Zero | 🤖️ XiaomiMiMo/MiMo-7B-RL-Zero |
| MiMo-7B-SFT | 基于基础模型训练的 SFT 模型 | 🤗 XiaomiMiMo/MiMo-7B-SFT | 🤖️ XiaomiMiMo/MiMo-7B-SFT |
| MiMo-7B-RL | 基于 SFT 模型训练的 RL 模型,性能优越,可媲美 OpenAI o1-mini | 🤗 XiaomiMiMo/MiMo-7B-RL | 🤖️ XiaomiMiMo/MiMo-7B-RL |
| 基准测试 | GPT-4o-0513 | Claude-3.5-Sonnet-1022 | OpenAI o1-mini | QwQ-32B-Preview | R1-Distill-Qwen-14B | R1-Distill-Qwen-7B | MiMo-7B-RL |
|---|---|---|---|---|---|---|---|
| 通用能力 | |||||||
| GPQA Diamond (Pass@1) | 49.9 | 65.0 | 60.0 | 54.5 | 59.1 | 49.1 | 54.4 |
| SuperGPQA (Pass@1) | 42.4 | 48.2 | 45.2 | 43.6 | 40.6 | 28.9 | 40.5 |
| DROP (3-shot F1) | 83.7 | 88.3 | 83.9 | 71.2 | 85.5 | 77.0 | 78.7 |
| MMLU-Pro (EM) | 72.6 | 78.0 | 80.3 | 52.0 | 68.8 | 53.5 | 58.6 |
| IF-Eval (Prompt Strict) | 84.3 | 86.5 | 84.8 | 40.4 | 78.3 | 60.5 | 61.0 |
| 数学能力 | |||||||
| MATH-500 (Pass@1) | 74.6 | 78.3 | 90.0 | 90.6 | 93.9 | 92.8 | 95.8 |
| AIME 2024 (Pass@1) | 9.3 | 16.0 | 63.6 | 50.0 | 69.7 | 55.5 | 68.2 |
| AIME 2025 (Pass@1) | 11.6 | 7.4 | 50.7 | 32.4 | 48.2 | 38.8 | 55.4 |
| 代码能力 | |||||||
| LiveCodeBench v5 (Pass@1) | 32.9 | 38.9 | 53.8 | 41.9 | 53.1 | 37.6 | 57.8 |
| LiveCodeBench v6 (Pass@1) | 30.9 | 37.2 | 46.8 | 39.1 | 31.9 | 23.9 | 49.3 |
MiMo-7B 系列
| 基准测试 | MiMo-7B-Base | MiMo-7B-RL-Zero | MiMo-7B-SFT | MiMo-7B-RL |
|---|---|---|---|---|
| 数学能力 | ||||
| MATH500 (Pass@1) | 37.4 | 93.6 | 93.0 | 95.8 |
| AIME 2024 (Pass@1) | 32.9 | 56.4 | 58.7 | 68.2 |
| AIME 2025 (Pass@1) | 24.3 | 46.3 | 44.3 | 55.4 |
| 代码能力 | ||||
| LiveCodeBench v5 (Pass@1) | 32.9 | 49.1 | 52.3 | 57.8 |
| LiveCodeBench v6 (Pass@1) | 29.1 | 42.9 | 45.5 | 49.3 |
[!IMPORTANT] 评估在
temperature=0.6条件下进行。AIME24 和 AIME25 为 32 次重复测试的平均得分。LiveCodeBench v5(20240801-20250201)、LiveCodeBench v6(20250201-20250501)、GPQA-Diamond 和 IF-Eval 为 8 次重复测试的平均得分。MATH500 和 SuperGPQA 为单次测试结果。
感谢 SGLang 团队的贡献,我们在 24 小时内就在 SGLang 主流版本中支持了 MiMo,MTP 功能即将上线。
示例脚本
# Install the latest SGlang from main branch
python3 -m uv pip install "sglang[all] @ git+https://github.com/sgl-project/sglang.git/@main#egg=sglang&subdirectory=python"
# Launch SGLang Server
python3 -m sglang.launch_server --model-path XiaomiMiMo/MiMo-7B-RL-Zero --host 0.0.0.0 --trust-remote-code详细使用方法可参见 SGLang 文档。MTP 功能也将在 24 小时内支持。
示例脚本
from vllm import LLM, SamplingParams
model_path = "/path/to/MiMo"
llm = LLM(
model=model_path,
trust_remote_code=True,
num_speculative_tokens=1,
disable_log_stats=False
)
sampling_params = SamplingParams(temperature=0.6)
conversation = [
{
"role": "system",
"content": ""
},
{
"role": "user",
"content": "Write an essay about the importance of higher education.",
},
]
outputs = llm.chat(conversation,
sampling_params=sampling_params,
use_tqdm=False)
for output in outputs:
prompt = output.prompt
generated_text = output.outputs[0].text
print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
print("=" * 80)您可以将 registry/register_mimo_in_vllm.py 复制到您的目录中,并通过以下方式导入
import register_mimo_in_vllm
from vllm import LLM, SamplingParams
model_path = "/path/to/MiMo"
llm = LLM(
model=model_path,
trust_remote_code=True,
# num_speculative_tokens=1,
disable_log_stats=False
)
sampling_params = SamplingParams(temperature=0.6)示例脚本
from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer
model_id = "XiaomiMiMo/MiMo-7B-RL-Zero"
model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained(model_id)
inputs = tokenizer(["Today is"], return_tensors='pt')
output = model.generate(**inputs, max_new_tokens = 100)
print(tokenizer.decode(output.tolist()[0]))我们尚未在其他推理引擎上验证MiMo,欢迎基于Huggingface仓库中的模型定义进行贡献 💻。
@misc{coreteam2025mimounlockingreasoningpotential,
title={MiMo: Unlocking the Reasoning Potential of Language Model -- From Pretraining to Posttraining},
author={{Xiaomi LLM-Core Team}},
year={2025},
eprint={2505.07608},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2505.07608},
}如有任何问题,请通过mimo@xiaomi.com与我们联系,或提交issue。