罗福莉沉寂半年官宣小米强化学习!直播新模型训练过程,一小时烧3万美元

关注
奖励曲线、显卡故障全公开。

编者按:本文来自微信公众号 量子位(ID:QbitAI),作者:梦晨,创业邦经授权转载。

什么情况?小米新模型的强化学习训练,直接开了现场直播。

点进去一看,这哪是训练现场啊,这就是烧钱现场,一眨眼就是10美刀,一分钟就是4000多元人民币出去了。

图片

当然烧钱的速度可能不是均匀的,从pro模型开始训练算起36小时,两款模型已经花了超过108万美元,平均每小时3万美元。

要是别人干这事得喊一句误闯天家,但是小米干这事只能喊一句误闯米家了。

在这个页面上,训练花了多少钱、跑了多少步、奖励曲线涨没涨、哪个节点的显卡出了故障,全部公开可查。

图片

见过开放权重、开放训练代码和开放训练数据的,开放训练过程的还真是没见过,围观群众纷纷大呼过瘾。

图片

图片

那么目前MiMo-V2.6的Flash和Pro两款模型训练得怎么样了?

其实还在初期,毕竟刚开始训练1天多一点。

不过已经能看出两者在RL过程中都有了比较明显的能力提升:

Pro的dynsam/avg@n从第1步的约0.565提升至0.614,Flash则从约0.514提升至0.603;最新公开的DeepSWE v1.1离线评测中,Pro和Flash分别达到62.24和60.77。(对比DeepSeek-Flash v1.1是74.2%)

Flash从更低的起点快速追近,Pro目前只保持小幅领先。不过Pro训练完一个Step更慢,所以最新的评分还没出来。

图片

自4月份开源MiMo-v2.5后,小米沉默了近半年。

现在负责人罗福莉出来解释,这期间只研究了一件事:强化学习能扩展到多远。

图片

小米公开如何Scaling强化学习

传统的预训练scaling很容易理解:更多数据 × 更多参数 × 更多算力 → 更强模型

现在前沿模型越来越关注RL能不能也这样scaling?

也就说如果持续增加每轮生成多少轨迹;模型面对多少种真实任务环境;为判断这些轨迹好不好投入多少计算;

模型能力是否还能持续提高?

小米给出的答案是沿三个维度扩展:训练计算量、环境/执行框架、评分计算3量。

计算量的Scaling

第一个维度最直观,就是把RL本身的计算规模做大。

MiMo-V2.6系列每个训练Step大约会处理20亿Token,配置为1568个Prompt × 每个Prompt 16条Rollout。

也就是说,同一道题并不是只让模型回答一次,而是让它尝试多条不同的解题和行动轨迹,再从这些尝试中获得强化学习信号。

1568×16意味着一轮就可能产生超过2.5万条Rollout。而对于Agent任务来说,一条Rollout又远不只是“一问一答”:模型可能要经历几十轮思考、工具调用、环境反馈和再次行动,因此最终一个Step的Token数可以达到数十亿级别。

图片

更重要的是,整个系统采用了Fully Async,也就是完全异步的执行方式。

传统意义上可以把一次RL训练想象成一条整齐的流水线:先把所有样本生成完,再统一评分,然后训练模型,最后开始下一轮。

但大规模Agent RL很难这样等待。

在MiMo的系统里,不同任务可以同时处于不同阶段:有的Agent还在环境中执行任务,有的已经完成、等待判分,有的正在计算Reward,还有新的任务正在进入系统。生成、执行、评分和训练不再严格排队,而是组成了一套持续运转的异步流水线。

图片

环境的Scaling

第二个扩展方向是Environments and Harnesses,也就是训练环境和执行框架的扩展。

MiMo-V2.6做的是Multi-task Agentic RL:代码、通用任务、视觉、Chat等不同类型的任务,可以被混合到同一次RL Run中训练,而且这些任务还可以运行在不同的Harness里。

例如一个编程Agent的Harness可能允许模型打开终端、修改代码、执行测试、阅读报错,然后继续修改;另一个视觉Agent面对的则可能是一套完全不同的工具、环境反馈和成功条件。

因此,MiMo想扩展的除了“题目数量”,还有模型能够进入多少种环境、使用多少种工具、解决多少种类型的问题。

这也是为什么直播页面会专门展示Batch Composition:这一栏实际上是在告诉外界,每一个训练Step中,模型正在从什么样的任务和环境里获取经验。

图片

评分的Scaling

第三个Scaling方向则更容易被忽略:Grader Compute,也就是给打分本身增加算力。

强化学习需要奖励,但复杂Agent任务的奖励 并不像数学选择题那样容易得到。

代码任务还可以运行Test Case:100个测试通过了多少个,就可以形成相对客观的反馈。

但面对更开放的Agent任务,仅仅判断“最终成功还是失败”往往远远不够,负责判断模型做得好不好的评分者,同样开始消耗越来越多计算资源。

而这里还有一个更加关键的问题Credit Assignment,也就是信用分配。

假设一个Agent为了完成任务连续执行了40步:搜索资料、打开网页、阅读信息、编写代码、运行测试、发现错误、修改代码,最终成功完成任务。

如果系统最后只告诉模型一句“成功,Reward=1”,这个学习信号其实非常粗糙。

因为模型并不知道:前面的40步中,究竟哪些动作真正帮助了成功?哪些步骤其实毫无必要?哪一次修改扭转了整个任务?又有哪些动作虽然最后没有导致失败,却实际上是不好的选择?

MiMo在这次训练中特别提到了Agentic In-group Credit Assignment。目前还没有公布具体算法,但结合“同一个Prompt生成16条Rollout”的训练方式,可以理解其基本目标:利用同组多条Agent轨迹及其结果,获得比简单的最终成败更加细致的强化学习信号。

同一组任务尝试,哪些行为应该获得更多鼓励、哪些应该减少鼓励,以及如何把这种判断转成训练信号。

图片

这样一来,MiMo所说的三个Scaling方向实际上构成了一套完整体系:

扩展计算量,让模型产生更多经验;扩展环境和Harness,让模型获得更加多样的经验;更多评分计算量,则负责从这些海量经验中提取更加准确的学习信号。

当预训练的Scaling已经发展多年之后,能否把模型与环境交互、产生经验、评价经验再到学习经验的整个RL循环,也变成一台可以持续扩大规模的机器?

不过前来围观的大佬ViT大佬给出更直接的翻译:三件事,背后都是算力。

图片

参考链接
[1] https://mimo.xiaomi.com/rl
[2]https://x.com/_LuoFuli/status/2100296686719610932

本文为专栏作者授权创业邦发表,版权归原作者所有。文章系作者个人观点,不代表创业邦立场,转载请联系原作者。如有任何疑问,请联系editor@cyzone.cn。

反馈
联系我们
推荐订阅