中意知识网 中意知识网

当前位置: 首页 » 常用知识 »

小米开源“Xiaomi MiMo”大模型:以7B参数超越OpenAI o1-mini

IT之家 4 月 30 日消息,小米大模型团队通过“Xiaomi MiMo”公众号宣布,今天,小米开源首个“为推理而生”的大模型 Xiaomi MiMo,联动预训练到后训练,全面提升推理能力。据介绍,MiMo 是来自全新成立不久的“小米大模型 Core 团队”的初步尝试。

在数学推理(AIME 24-25)和代码竞赛(LiveCodeBench v5)公开测评集上,MiMo 仅用7B 的参数规模,超越了 OpenAI 的闭源推理模型o1-mini 和阿里 Qwen 更大规模的开源推理模型 QwQ-32B-Preview

官方表示,MiMo 推理能力的提升,由预训练和后训练阶段中数据和算法等多层面的创新联合驱动,包括:

  • 预训练:核心是让模型见过更多推理模式
    • 数据:着重挖掘富推理语料,并合成约 200B tokens 推理数据。
    • 训练:进行了三阶段训练,逐步提升训练难度,总训练 25T tokens。
  • 后训练:核心是高效稳定的强化学习算法和框架
    • 算法:提出 Test Difficulty Driven Reward 来缓解困难算法问题中的奖励稀疏问题,并引入 Easy Data Re-Sampling 策略,以稳定 RL 训练。
    • 框架:设计了 Seamless Rollout 系统,使得 RL 训练加速 2.29 倍,验证加速 1.96 倍。

IT之家附开源地址:

  • Hugging Face:https://huggingface.co/XiaomiMiMo
  • 技术报告:https://github.com/XiaomiMiMo/MiMo/blob/main/MiMo-7B-Technical-Report.pdf

未经允许不得转载: 中意知识网 » 小米开源“Xiaomi MiMo”大模型:以7B参数超越OpenAI o1-mini