Token导航 LogoToken导航

0.8B机器人小模型偷师大模型逼近14B性能

更新时间 2026-10-17来源 搜狐科技正文 5592字阅读约 18分钟1 张图片
0.8B机器人小模型偷师大模型逼近14B性能

有一件事我一直觉得挺反直觉的。

按理说,机器人要学会干活,应该是练得越多越聪明。可现实是,那些能操控机械臂抓水果、叠衣服的AI模型,练了几十万条数据后依然会在光线一变、角度一换、桌子上多了个杯子的时候突然失手。这不是数据不够多的问题,因为哪怕再喂它十倍的数据,这种脆弱性依然存在。问题出在更底层的地方:这些模型压根不知道"世界会怎么反应"。

这篇来自威斯康星大学麦迪逊分校和伊利诺伊大学香槟分校团队的论文,就是冲着这个structural问题去的。他们的答案挺巧妙:不去喂更多数据,而是去偷一点"世界观"过来。

**核心问题:会做动作的模型,不代表懂物理**

先说清楚两个概念。

VLA模型*:Vision-Language-Action,视觉-语言-动作模型,简单说就是看一眼场景、听一句指令,直接吐出机械臂该往哪动的模型。它的训练方式是看着大量人类示范,学"看到这个画面,该做这个动作",本质上是一个映射函数,输入观测,输出动作,中间没有对"如果我这么做了,世界会变成什么样"的建模。

这就是VLA的天生短板。它学的是"对应关系",不是"因果关系"。就像一个学生做题只背答案,不理解解题过程,题型稍微变一下就懵了。论文里提到,OpenVLA这类早期开源模型用7B参数的Llama-2做骨干,把连续动作离散成token,一路预测下去,性能不错,但对环境扰动很敏感——遮挡多一点、灯光换一下,成功率就往下掉。

与之相对的是另一条路子:世界模型*:World Model,训练目标不是直接输出动作,而是预测"未来的画面会长什么样",往往同时预测未来帧、未来状态、未来动作。这个额外的负担逼着模型必须理解时间怎么流动、动作会导致什么后果,所以它的内部表征天然比只会输出动作的模型更"懂物理"。

已经有研究证明,这类基于世界模型构建的策略(论文里叫World Action Model,WAM)在新任务上的迁移能力更强,扰动下的表现也更稳。但代价极其高昂:DreamZero这个14B的世界动作模型,在H100显卡上跑一次推理要3秒,显存占用45.9GB,普通消费级显卡RTX 5090(32GB显存)根本装不下。相比之下,π0.5这类正常VLA只要65毫秒,本文的0.8B学生模型更是只要32毫秒,显存占用仅1.86GB。

这就是矛盾所在。世界模型脑子好使,但跑起来太慢太贵,没法真正部署到机械臂上实时控制。VLA跑得快,但脑子里缺一根筋。

这就好比你想雇一个既懂心理学又能做体力活的保安。心理学教授能一眼看穿谁图谋不轨,但让他去扛麻袋,他磨磨蹭蹭半天挪不动一步;壮汉扛麻袋一把好手,但对可疑人物毫无察觉。如果非要在两者之间选一个,你永远得牺牲一头。研究者们想问的是:能不能把心理学教授脑子里的判断力,想办法"移植"到壮汉脑子里,而不需要壮汉真的去读四年心理学?

**方法核心:只学"怎么想",不学"怎么生成"**

论文提出的答案是一个听起来简单到有点朴素的想法:世界模型之所以懂物理,是因为它的内部特征里已经藏着这些知识;生成未来画面只是产生这些特征的手段,而不是知识本身。既然知识已经在特征里了,那就没必要让学生模型也去学着生成未来画面,直接让学生的内部特征去"对齐"老师的内部特征就够了。

这个操作在整篇论文里被称作表征对齐*:Representation Alignment,不要求学生输出和老师一样的动作或画面,只要求学生某一层的内部特征在方向上和老师的特征保持一致。

具体怎么做的?拆开看,其实是三步走。

第一步,把世界模型当作一个纯粹的"标注机器",提前跑一遍训练数据,把每一帧画面对应的内部特征抽出来,存进一个缓存文件里。这一步完成后,老师模型就可以彻底关掉,不再需要占用任何显存。论文里说,在LIBERO这个基准上,用四张GPU跑这个预处理大约只要一个小时。

第二步,训练学生VLA模型的时候,正常走它自己的动作预测流程,用真实的示范动作做监督,这部分完全不变。

第三步,额外加一个损失项:把学生模型某一层(论文选的是最后一层)的图像token特征,做个均值池化,再通过一个两层MLP投影器*:一种小型神经网络,负责把学生特征的维度转换成和老师特征一样的维度,方便两边做比较,投影到老师特征的维度上,然后计算两者的余弦相似度损失*:cosine loss,衡量两个向量方向是否一致的一种损失函数,值域在0到2之间,越接近0说明两个向量指向越一致。这个损失只要求方向一致,不要求数值完全相同,所以两个模型完全不需要用一样的网络结构或者一样的特征维度。

这个设计里有两个细节特别值得琢磨。

一个是"冻结老师"。训练过程中老师模型从头到尾不会被加载进显存,因为它的输出已经提前算好存起来了。这意味着蒸馏这件事的训练成本,和不做蒸馏的普通训练成本几乎一样,唯一多出来的开销就是读一个缓存文件加算一个余弦损失。

另一个是"方向对齐而非数值复制"。因为只要求方向一致,学生模型完全可以保留它自己动作预测所需要的额外结构,不会被老师的特征"污染"或"绑架"。这也是为什么后面消融实验里换了三种完全不同架构的老师模型都能生效,因为对齐的是一种抽象的"物理直觉",不是某个特定网络的具体数值。

这套操作特别像补课。你请一个物理学家来给孩子讲题,但你并不要求孩子把物理学家的解题步骤一字不差地抄下来,你只要求孩子最后得出的"物理直觉"和物理学家的直觉方向一致就行。物理学家讲完课就可以回家了,不需要每次考试都坐在孩子旁边盯着。如果非要孩子完全复刻物理学家的解题过程,孩子反而可能学不会自己独立思考,考试时物理学家不在场就抓瞎。而这套方案恰恰保证了:物理学家(世界模型)教完就走,孩子(VLA学生模型)自己独立上考场(真实部署),发挥依然稳定。

论文里主力用的老师模型是Cosmos3-Nano*:英伟达发布的Cosmos 3世界模型中提取出的一个8B参数的理解塔(属于16B总参数的混合专家架构中的一部分),从中读取第24层的图像token隐藏状态,作为教学信号。这个塔本身是确定性的,不涉及扩散生成那种随机采样过程,所以每次抽取的特征都是稳定的。

学生模型的主力配置是一个0.8B参数的QwenGR00T策略*:由Qwen3-VL视觉语言骨干网络加上GR00T风格的flow-matching动作专家组成的模型架构,flow-matching是一种通过学习"噪声到目标"的速度场来生成动作序列的技术,这个学生在部署时只需要跑一次骨干前向计算加四步flow采样,推理速度和不做蒸馏的版本完全一样,因为对齐用的投影器和损失计算在推理阶段被整个丢弃了。

**实验结果:小身板打出大力气**

数字最能说明问题。

在LIBERO基准(一个包含空间、物体、目标、长时序四大类任务的机器人操作测试集)上,这个0.8B的蒸馏学生达到了97.9%的成功率,比同样架构但没做蒸馏的版本高出2.6个百分点。作为对比,表格里其他参数量在4B以下的模型,成绩普遍卡在78.7%到95.3%之间,没有一个能摸到98%。

在RoboCasa-GR1人形机器人操作基准上,24个任务环境,成功率从48.2%提升到50.5%,涨了2.3个百分点。这个提升让0.8B的小模型反超了QwenFAST、QwenPI、QwenOFT这几个4B的同门师兄,也超过了两个版本的Isaac-GR00T,距离同架构的4B版本QwenGR00T(54.8%)只差4.3个百分点,而参数量只有对方的五分之一。

真实机械臂上的测试更有说服力。团队在一台AgileX Nero单臂机械臂上做了两个任务:一个是把水果捡进篮子,一个是把鸡蛋从筐里挪进篮子。前者是相对容易的抓取放置,后者因为鸡蛋光滑易滑手,对抓取力度要求更苛刻。单臂捡水果任务,0.8B蒸馏模型达到93.3%的成功率(30次里成功28次),和一个4B的π风格模型打平,比未蒸馏的对照组高出3次。到了更难的捡鸡蛋任务,差距才真正显现出来:未蒸馏版本掉到46.7%,蒸馏版本追回了4次成功,达到60.0%,而4B模型是70.0%。

接着团队又把水果任务搬到一台双臂平台TRIP-Bag上,改造成一个需要两只手臂配合完成的"接力递水果"任务:右臂抓水果,在空中递给左臂,左臂把水果放进袋子。这个改动同时换了机身、摄像头、控制延迟,用来验证蒸馏带来的收益到底是方法本身的效果,还是只在某一套硬件上凑巧管用。结果显示同样的排序关系保留了下来,未蒸馏版本40.0%,蒸馏版本46.7%,4B模型53.3%。

失败的案例也很有意思,团队专门分析了失败模式,发现问题几乎都出在最后几厘米的精细操作上:机械臂能准确找到目标物体,伸手过去方向也对,但松手时机不对,要么提前松开掉在半路,要么因为物体和容器边缘在画面里挨得太近而抓到了容器边缘而不是物体本身。鸡蛋任务还多了一种失败:手指已经合拢抓住了蛋,抬起来的过程中蛋从指缝滑走。这类失败本质上是接触力控制的问题,不是场景理解的问题,团队坦白说这恰恰是他们预期这套方法帮助最小的地方,因为世界模型对齐提升的是"理解物理场景"的能力,不是精细化的力控技巧。

这个观察其实挺诚实的,一个方法不可能包治百病,能讲清楚它在哪里不起作用,反而更让人相信它在起作用的地方是真的起作用。

**消融实验:验证这不是巧合**

一个方法要让人真正信服,光靠一组好看的数字是不够的,得经得起反复折腾。论文做了四组消融实验,逐一拆解这套方法里每个部分到底是不是必要的。

第一组换学生模型的规模和骨干网络家族,从0.8B的Qwen3.5-VL,到1B的InternVL,再到4B的Qwen3-VL,三种配置下蒸馏都带来了稳定提升,涨幅在1.6到2.5个百分点之间。这说明投影器"自动适配维度"这个设计确实管用,不需要为每种学生模型单独调整对齐方式。

第二组换对齐用的层,在0.8B学生模型的四分之一深度、二分之一深度、三分之二深度、最后一层之间做扫描。结果显示最后一层效果最好(49.8%),但二分之一深度的层紧随其后(49.4%),差距在实验的误差范围内。团队解释说这说明这个方法对"具体接在哪一层"不算敏感,选最后一层纯粹是因为工程上最省事,反正动作预测头本来就要用这一层的输出,不需要额外保留中间层的激活值。

第三组是最关键的一组:换老师模型。团队特意挑了两个和主力老师Cosmos3-Nano在训练目标、网络架构、特征维度上都截然不同的对照组:Fast-WAM(基于Wan2.2视频扩散模型的世界动作模型)和V-JEPA2-AC(一种自监督视频编码器加动作条件预测器的组合)。结果三个老师全部带来提升,V-JEPA2-AC涨1.2个点,Fast-WAM涨1.6个点,Cosmos3-Nano涨2.6个点。

这一点特别重要,因为它排除了一种最让人担心的可能性:会不会这个提升只是因为Cosmos3-Nano这个特定老师碰巧和学生"八字很合",换个老师就失灵了?答案是不会。三个风格迥异的老师都管用,说明学生学到的不是某个特定网络的怪癖,而是一种更普遍的、关于物理世界的表征先验。

这就好比你怀疑一个孩子考试进步是不是因为运气好抽到了他刚好复习过的题型,于是你专门找了三份完全不同出题风格的试卷让他考,结果三次成绩都提升了。这时候你才能真正相信,这孩子是学会了知识本身,而不是蒙对了一次考题。

写在后面

读完这篇论文,最让我觉得有意思的一点,其实不是那些漂亮的成功率数字,而是它对"知识"这个东西本身的拆解方式。

我们通常默认,一个模型要"懂"什么东西,就得能"做"那件事——世界模型懂物理,是因为它能生成未来的画面;如果你想让另一个模型也懂物理,直觉上应该也让它去学生成未来画面。但这篇论文说的是,生成能力和理解能力其实是可以剥离的两件事,生成只是训练理解的一种手段,一旦理解形成了,手段就可以扔掉。这有点像你学游泳是为了在水里活动自如,但一旦学会了,你未必需要每次下水前都重新经历一遍学游泳的挣扎过程,那种"水感"已经内化成你身体的一部分了。

另一个值得琢磨的细节是失败模式那部分的坦诚。团队没有把所有失败都归咎于方法不够好,而是精确地说出了这个方法"帮不上忙"的地方在于最后几厘米的接触力控制。这种边界感其实比无限吹嘘效果更有价值,因为它给后续研究者指出了一个明确的方向:如果想解决抓取鸡蛋这类精细力控问题,可能需要一种完全不同的信号来源,而不是继续往世界模型对齐这条路上使劲。

这篇论文也没有回答的一个问题是,如果把多个不同类型的老师模型同时蒸馏进同一个学生里,会不会产生叠加效应,还是会互相打架?目前的实验只验证了单个老师换来换去都有效,但没有验证"多老师联合教学"是否可行。这大概是留给下一篇论文的空间了。

Q&A

Q1:Think Like a World Model, Act Like a VLA这篇论文的核心方法是什么?

A:核心方法是给普通VLA训练加一个特征对齐损失项,让学生模型的图像特征去模仿一个冻结的世界模型缓存下来的特征方向,不需要学生学会生成未来画面,部署时也不需要加载老师模型,推理速度和普通VLA完全一样。

Q2:为什么0.8B的小模型能追上参数量大好几倍的模型?

A:因为它通过蒸馏继承了世界模型内部对物理场景的理解能力,而不是单纯靠堆参数或堆数据。实验显示这个0.8B学生在LIBERO上达到97.9%,在RoboCasa-GR1上从48.2%提升到50.5%,真实机械臂任务上也追平了4B模型的表现。

Q3:这套蒸馏方法只对特定的世界模型老师有效吗?

A:不是。论文换了三种架构、训练目标完全不同的老师模型(Cosmos3-Nano、Fast-WAM、V-JEPA2-AC)测试,全部带来了成绩提升,说明学到的是一种普遍的物理表征先验,不是某个特定网络之间的巧合。

文章标签机器人大模型
资讯来源:由AI资讯编辑整理自互联网公开内容,版权归原作者所有,未经许可,不得转载。

继续浏览更多资讯

返回资讯目录

相关资讯

更多