Token导航 LogoToken导航

AI 资讯 · 科技行者

MeanFlow技术让机械臂一步生成动作

更新 2026-11-04科技行者正文 5307字阅读约 17分钟

你有没有想过,为什么现在最先进的机器人,想要做一个拿杯子的动作,脑子里却要"想"十次才能动一下?

这不是夸张。目前主流的机器人基础模型,无论是英伟达的GR00T,还是各家实验室的具身智能方案,内部用的都是一种叫"流匹配"的技术来生成动作。这套技术的工作方式有点像老式相机的多次曝光:先给机械臂的动作轨迹撒上一层纯噪声,然后让模型一步一步去噪,通常需要四步、十步甚至更多步骤,才能把噪声变成一个真正可执行的动作。

问题是,机器人不是在实验室里悠哉生成一张图片,它是要在真实世界里抓取、移动、避障的。每一次去噪都要重新调用一次神经网络的前向计算,这个过程不是免费的。拿英伟达的GR00T-N1.6来说,生成动作这一个环节就占掉了整个推理时间的40%左右,在算力更弱的机器人边缘设备(比如Jetson Orin芯片)上,这个比例甚至能冲到50%。

也就是说,机器人一半的"思考时间"都花在反复去噪上,而不是真正理解世界和做决策。这就是这篇论文想要解决的问题:能不能让机器人一步就把动作生成出来,而不是磨磨蹭蹭走十步?

一步生成这件事,图像生成领域已经有答案了

把时间拨回到最近两年,图像生成领域正好遇到过一模一样的难题。扩散模型生成一张图片动辄要几十步甚至上百步,于是研究者们发明了各种加速方案,从知识蒸馏到一致性模型,五花八门。

MeanFlow:一种新的训练思路,与传统流匹配预测"瞬时速度"不同,它让模型直接学习任意两个时间点之间的"平均速度",训练好之后,只要把起点和终点一代入,模型就能一步跨过去,不需要中间的那些弯弯绕绕。

这个想法听起来挺美好。既然图像生成那边验证过MeanFlow能实现一步生成,那直接把它搬到机器人动作生成上不就行了?

论文作者一开始也是这么想的。然而直接套用的结果是:训练直接崩了。

机器人的"速度场"和图片的完全不是一回事

这才是整篇论文最硬核也最有意思的发现。

研究者们做了个对比实验:分别在ImageNet图像生成任务和LIBERO机器人操作任务上,观察模型预测的"速度"是怎么随着去噪过程变化的。他们盯着一个叫"局部加速度"的指标,简单说就是相邻两个时间点之间平均速度的变化幅度。

结果让人意外。在图像生成里,这个加速度从头到尾都很平稳,像一条几乎不起波澜的直线。但在机器人动作生成里,这条线在去噪快结束的时候(也就是越接近生成最终动作的阶段)突然暴涨,峰值能达到起始值的76倍。

这意味着什么?

意味着机器人动作的生成轨迹不是一条平滑的直线,而是前半段风平浪静、后半段突然剧烈拐弯的曲线。

这就像你开车去一个陌生的地方。前面大半段路程都在高速公路上匀速巡航,方向盘几乎不用动,这时候你完全可以凭经验,闭着眼睛估算接下来十分钟车会开到哪里。可是快到目的地的最后几百米,突然是一堆窄巷子、急转弯、还有逆行的电动车冲出来,这时候如果你还用高速上那种"一把推算到底"的方式去预判路线,大概率会撞车。机器人动作生成的后半程,就是这种窄巷子路况,如果你用前半段的平稳假设去外推整个过程,模型自然会在最后关头彻底失准。

更麻烦的是第二个发现:这种加速度的波动范围(也就是不同样本之间差异有多大),会随着去噪推进越来越宽。论文进一步分析发现,这个波动范围跟训练数据里任务的多样性强相关,也就是说,机器人要做的任务种类越多(比如又要开抽屉又要拿勺子又要导航),这个后期的剧烈波动就越难预测,越容易出错。

这两个问题叠加在一起,直接导致MeanFlow在机器人身上训练时,损失函数的图像变成了"钉子地毯"——到处是尖刺,极不稳定。论文里给出的实测数据显示,最大损失值能飙到1.6乘以10的四次方,而正常训练应该是平滑的小坡度曲线。

Kinematic MeanFlow:把一步跨越拆成两步估算

既然问题出在"用一个局部观察去外推一整段剧烈变化的轨迹",那顺理成章的思路就是:别让模型一次性猜完整段,而是把这段路拆成两截,分别猜。

这正是论文提出的核心方法,叫Kinematic MeanFlow,简称K-MF。

运动学恒等式:这是整个方法的理论基石,说的是如果你在时间区间[r, t]中间选一个点c,那么从r到t的平均速度,正好等于从t到c这一段平均速度,加上从c到r这一段平均速度的加权组合,这是个数学上严格成立的恒等式,不是近似。

利用这个恒等式,K-MF把原本需要在整个大区间上估算一次的"时间导数项",拆成了两个子区间上分别估算的导数项。一个负责早期阶段(去噪刚开始,路况平稳),一个专门负责晚期阶段(去噪接近结束,路况复杂)。

这就好比你原本要独自开车从北京跑到广州,中途完全不休息不换人,路上任何一点疲劳驾驶的失误,都会在后面几千公里里被放大成致命的偏航。现在换成接力赛:第一个司机专门负责高速平直路段,开到武汉就交接给第二个司机,第二个司机只需要集中精力应付后半段进城的复杂路况。如果不这样分段,一个人既要应付平稳路段的长时间专注,又要在最后关头突然切换到高难度路况,出错率自然高。K-MF做的就是这件事:在中间点c设一个"交接站",让两段路分别被专门的估算负责,而不是让一次估算扛下全程。

那么这个交接点c到底该设在哪儿?论文试了三种策略。

第一种是随机采样,让c在区间内随机取值,指望用多样化的切分方式让模型见多识广。第二种是固定在中点,简单粗暴,每次都从正中间切一刀。第三种是可学习策略,用一个只有三层的小型神经网络,根据当前的t和r自动算出最合适的c应该设在哪里。

实验结果有点出乎意料:随机采样反而是三种策略里表现最差的。固定中点效果明显更好,而可学习策略进一步小幅超越固定中点,是三者中综合表现最优的。

这个结果其实挺反直觉的。按常理说,让模型见识更多样的切分方式,应该有利于泛化,就像教小孩算术,见过越多种类的题目理应学得越扎实。可现实是,胡乱切分反而带来了更多噪声,没能帮上忙。

这说明了一件事:K-MF真正起作用的地方,不是因为它提供了"更丰富的训练样本多样性",而是因为它找到了一个更精准的"估算锚点"去压制误差放大。多样性不是重点,精准的定位才是。

可学习的交接点:小网络怎么决定"在哪里换人"

既然固定中点已经不错,为什么还要费劲训练一个额外的小网络?

论文给出的设计考量挺有意思。这个小型生成器(代码里叫Gφ,一个仅有三层的轻量MLP)在训练的不同阶段,需要完成不同的任务。训练初期,模型自己对"整体时间导数"的估算还很不准,这时候生成器应该让拆分后的两段估算尽量区别于那个本来就不准的整体估算,相当于故意制造差异化、避免被带偏。而到了训练后期,当两段分开估算的"接力赛机制"已经跑顺了,生成器就应该让拆分估算逐渐和整体估算趋同,保持结果的自洽性。

论文用了一个随训练进度线性变化的参数γ来控制这个权重的转变,让生成器在训练的前半段和后半段扮演不同的角色。

实验里还有个挺耐人寻味的观察:这个可学习生成器算出来的交接点位置,和"区间宽度"(也就是t减去r)有很强的相关性。区间越宽,生成器的策略越明显地偏离中点;数据集规模越大(比如从2000条轨迹的LIBERO换成8.7万条轨迹的Fractal数据集),生成的交接点分布反而更集中。这暗示着,任务越复杂,模型越需要精确的锚点定位,而不是随意摆动。

这就好比一个资深的急诊科医生在分诊。病人少、病情简单的小诊所,护士随便按个先来后到分诊也不会出大错。可是到了大型医院,病患种类五花八门,这时候分诊台就得有经验丰富的医生专门盯着,根据病情轻重精确判断谁该优先处理,交接点的"随意"与"精确"之间的差距,会随着场景复杂度的提升而被急剧放大。

一步生成,效果不降反升

说了这么多理论,实际效果到底怎么样?

论文在两种训练范式下做了系统验证:一种是在预训练好的GR00T-N1.6上做微调,另一种是在SimVLA-S上从零开始训练。结果显示,K-MF在只用一步生成的情况下,综合表现持平甚至超过了需要四步甚至十步的传统流匹配方法。

具体数字上,在LIBERO基准的四个任务套件(空间关系、物体识别、目标变化、长序列任务)上,GR00T-N1.6用K-MF一步生成拿到了97.9%的平均成功率,而传统流匹配四步生成是97.3%。SimVLA-S从零训练的情况下,K-MF一步生成是95.1%,流匹配十步生成是94.2%。

更有说服力的是跨场景验证。论文还在真实世界采集的Fractal数据集(谷歌机械臂,8.7万条轨迹)和BridgeData V2数据集(WidowX机械臂,5.3万条轨迹)上做了测试,涵盖"拿取""移动""开抽屉""关抽屉"等桌面操作任务,以及人形机器人Unitree G1的点到点导航任务。结果显示K-MF在绝大多数任务上都优于传统流匹配,比如在Fractal的"开抽屉"任务上,成功率从35.5%直接跳到了50.5%,提升了15个百分点。

推理速度方面,这才是这篇论文真正想解决的痛点所在。在英伟达L40显卡和Jetson Orin边缘设备上,无论是用普通的eager模式还是编译优化后的compile模式,K-MF都把动作生成这一环节的延迟砍掉了67.5%到74.4%,整个端到端推理延迟降低了30.3%到54.9%。换句话说,机器人原本要花两秒钟"想"出一个动作,现在可能一秒不到就搞定了。

当然,天下没有免费的午餐。K-MF因为需要同时估算两段子区间的导数,训练时间比传统流匹配要增加大约27%到33%,GPU显存占用也略有上升。但考虑到推理速度带来的巨大收益,论文认为这个训练成本是完全可以接受的。

对比其他同类方案,差距在哪里

论文没有回避和其他同期工作的横向对比。他们选了三个代表性的对照方法:直接套用原版MeanFlow、专注于提升强化学习稳定性的MVP方法、以及用课程学习思路做区间划分的α-Flow方法。

实测发现,原版MeanFlow想要勉强跑通,需要一整套"精心调校"的训练技巧,包括梯度裁剪、渐进式时间步采样(训练初期先让模型学习瞬时速度,再逐步过渡到学习平均速度)、以及足够多的训练轮次。即便这样伺候着,MeanFlow也只能实现两步生成,一步生成直接宣告失败。而且它对"切分时间点"选在哪里特别敏感,论文的敏感性实验显示,把两步生成的切分点从0.5挪到0.1,成功率能从接近90%暴跌到接近0。

α-Flow和MVP倒是不需要这套复杂的调教流程,分别靠区间划分的课程设计和瞬时速度约束撑住了训练稳定性。但一旦逼它们做到真正的一步生成,MVP直接失效,α-Flow的成功率也明显掉队。

只有K-MF,在没有任何额外调参技巧的情况下,稳稳地实现了一步生成,而且效果不输多步方法。

写在后面

读完这篇论文,最让我意外的不是K-MF这个方法本身有多精巧,而是那个关于"速度场形状"的发现。研究者没有一上来就设计新算法,而是先花时间去画图、去对比,弄清楚机器人动作生成和图像生成到底哪里不一样。这种"先搞清楚病因再开药方"的研究路径,其实比直接套用现成方案更费时间,但也更靠谱。

论文里还有个细节我觉得值得单独拎出来说:那个关于误差放大的实验。研究者故意在不同的早期时间点,往模型输入里掺进1%的随机扰动,然后观察这个扰动对最终导数估计造成的影响能被放大多少倍。结果是,越早引入扰动,到最后被放大的倍数越夸张,能到120倍左右;而越晚引入扰动,放大效应反而越温和。这说明机器人动作生成这件事,容错窗口是前松后紧的,早期犯点小错没关系,系统会自己纠正,但越靠近终点,哪怕一个很小的偏差都可能被滚雪球式地放大。这跟很多人对"深度学习模型越往后越精细"的直觉认知,其实是拧着的。

还有一个让我多想了一会儿的点:论文发现任务越多样、数据集越大,速度场的那个"波动范围"就越宽,而且这个增长趋势一开始接近对数增长,到了某个阶段又变成饱和的幂律增长。这其实暗示了一件更大的事:随着机器人基础模型继续往更大规模、更多任务类型扩展,一步生成这件事可能会变得更难,而不是更容易。论文作者自己也在结尾坦率地说,他们的方法目前还没有覆盖全身控制和灵巧操作这些更难的场景。这不是一个已经终结的故事,倒像是刚刚掀开了一角。

Q&A

Q1:Kinematic MeanFlow是什么?

A:Kinematic MeanFlow(K-MF)是论文提出的一种让机器人基础模型一步生成动作的新方法,它基于运动学恒等式把时间导数拆成两段分别估算,解决了原版MeanFlow直接用在机器人上会训练崩溃的问题。

Q2:为什么MeanFlow直接用在机器人动作生成上会失败?

A:因为机器人动作的速度场在去噪后期会突然剧烈波动(局部加速度能飙到初始值的76倍),而且不同样本之间的波动差异也会越来越大,这让模型很难准确估算关键的时间导数项,导致训练出现大量损失尖刺甚至彻底崩溃。

Q3:Kinematic MeanFlow能带来多大的推理速度提升?

A:在GR00T-N1.6模型上测试,K-MF能把动作生成部分的延迟降低67.5%到74.4%,整体端到端推理延迟降低30.3%到54.9%,同时在大多数任务上的成功率不输甚至超过传统多步生成方法。

资讯来源:由AI资讯编辑整理自互联网公开内容,版权归原作者所有,未经许可,不得转载。