网站首页 文章专栏 小米开源“Xiaomi MiMo”大模型:为推理而生,以 7B 参数超越 OpenAI o1-mini
小米开源“Xiaomi MiMo”大模型:为推理而生,以 7B 参数超越 OpenAI o1-mini
发布 作者:清源 浏览量:2
小米大模型团队宣布开源首个大模型 Xiaomi MiMo,提升推理能力,超越其他模型。

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之家附开源地址:

loading