像世界模型一样思考,像VLA一样行动:把世界模型表征蒸馏进紧凑机器人策略 Think Like a World Model, Act Like a VLA: Distilling World-Model Representations into Compact Robot Policies
离线缓存世界模型特征做余弦对齐,0.8B小VLA零推理成本获得大模型级精度
前置知识
VLA(视觉-语言-动作模型)
把预训练视觉-语言骨干与动作生成头结合的机器人控制模型:输入相机图像、语言指令与本体状态,直接输出动作序列,代表工作有 OpenVLA、π0、GR00T N1。它继承了互联网级预训练的语义理解,但训练目标里没有“世界如何响应动作”这一项,鲁棒性主要取决于数据覆盖。
本文的学生就是 0.8B 的 QwenGR00T VLA,方法是在其标准训练目标上加一个对齐项;不先理解 VLA 的训练范式,就无法定位论文改动在哪里、部署图为何不变。
世界模型(World Model)
被训练去预测环境如何随时间演化的模型,通常联合预测未来帧、状态与动作;因为必须内化时间结构与因果后果,其内部表征比纯动作策略更物理接地,迁移与抗扰动更好。但滚动未来一步的代价远高于直接预测动作:DreamZero 在 H100 上一次推理需 3 秒、45.9GB 显存。
全文出发点是“世界模型贵在生成、富在表征”这一矛盾:表征值得要,生成进不了控制回路。理解这对张力才能看懂为什么作者选择只偷特征、不搬生成器。
流匹配(Flow Matching)
一种生成式建模方式:从噪声 $\epsilon$ 出发,沿线性插值路径 $a_\tau=\tau\epsilon+(1-\tau)a_{t:t+K}$ 回归速度场 $\epsilon-a_{t:t+K}$,少数几步积分即可得到动作块。GR00T 风格的动作专家即用流匹配做速度回归,推理只需几个流步骤,是 VLA 高效部署的关键。
论文的基础动作损失 $\mathcal{L}_{act}$ 就是流匹配速度回归,且蒸馏后的策略推理时只跑四个流步骤——理解它才能理解“部署成本与基线完全一致”这句话的确切含义。
表征级知识蒸馏(FitNets / REPA)
经典蒸馏转移教师的输出分布;FitNets 证明让中间层模仿教师的“提示”(hints)能传递更多信息;REPA 把扩散 Transformer 的中间特征对齐到冻结自监督编码器,用逐点余弦损失大幅加速生成训练。本文同属此谱系,但教师换成预测未来的世界模型,特征因此自带时间与因果结构。
本文对齐项在形式上与 FitNets/REPA 同类(投影后的余弦损失加停止梯度),理解这条谱系才能分清:真正的新意不在损失函数本身,而在教师身份与零成本训练经济学。
余弦对齐与停止梯度
只约束两个向量方向一致、不约束模长的损失 $\mathcal{L}_{align}=1-\cos\big(p_\phi(f_c^S),\,\mathrm{sg}[f_c^T]\big)$,其中 $\mathrm{sg}[\cdot]$ 表示对教师目标停止梯度。方向对齐允许学生保留动作目标所需的额外结构,师生骨干无需共享特征空间或维度,一个 MLP 投影器即可适配任意宽度组合。
这一项就是论文方法的全部落地载体:配合两层 MLP 投影器 $p_\phi$,实现任意 $(D_s,D_t)$ 师生配对,也是教师无关性消融(换三个异构教师都有效)的技术基础。
LIBERO 与 RoboCasa-GR1 基准
LIBERO 是终身学习操作基准,含 SPATIAL/OBJECT/GOAL/LIBERO-10 四个套件,每任务评 50 回合、每套件 500 回合,报告平均成功率;RoboCasa-GR1 是 GR1 人形机器人的 24 个家庭任务环境,每环境 1000 条演示、共 602 万步,单 ego 相机、58 维状态。两者因流匹配采样与渲染器差异存在跨 GPU 方差。
主结果数字全部来自这两个仿真基准;作者特意用 2 种子×2 GPU 共四次评估取均值报标准差,正是为了应对这两个基准已知的复现不稳定问题。
研究动机
标准 VLA 训练只把观测映射为动作,损失里没有任何一项刻画世界如何响应这些动作,因此策略可以很好拟合训练分布,却学不到时间接地、因果一致的表征;其鲁棒性退化为数据覆盖问题——对视角偏移、光照变化、遮挡等训练中欠代表的扰动很脆弱,适应新任务或新本体还需大量额外数据。另一条路线的世界模型/世界动作模型(WAM)被迫内化时间结构与因果后果,表征更物理接地、迁移更好,但代价是部署不可行:滚动未来远贵于预测动作,DreamZero 在 H100 上一次推理需 3 秒、占用 45.9GB 显存(消费级 RTX 5090 只有 32GB,根本装不下),而 π0.5 在 RTX 5090 上也需 65ms。于是形成结构性矛盾:世界模型知识最丰富却进不了控制回路;VLA 部署便宜,却缺少那份接地。
本文的目标是论文的问题是:轻量 VLA 能否继承大型世界模型的接地表征,而完全不支付其推理成本?作者给出肯定答案,并设定了严格的目标约束——在普通 VLA 训练上只增加一个特征对齐项,让 0.8B 学生在 LIBERO、RoboCasa-GR1 两个仿真基准与单臂/双臂两台真机上追平数倍参数的模型;同时训练时不把教师装进显存(蒸馏运行的能耗与成本必须等于基线运行),训练结束丢弃对齐头,部署的策略在结构上与未蒸馏基线完全一致(同参数量、同四个流步骤),推理保持 32ms、1.86GB 跑在 RTX 5090 上。这样所有收益都只能归因于表征本身,而非额外容量或测试时计算。
与已有工作不同的是,独特切入是把“接地”与“生成”显式分离:作者观察到世界模型对物理场景的知识已经存在于其内部特征中,“生成未来”只是产出这些特征的目标函数,不是特征使用的必要属性,因此学生只需在特征层面被监督,无需自己滚动未来。与 FitNets、REPA 等表征蒸馏的本质差别在教师的身份:普通判别性编码器的特征不携带“知道未来”的归纳偏置,而世界模型的特征天然含有时间与因果结构。工程上再压缩成本:教师冻结后其目标只依赖训练帧,可离线一次性提取并缓存为内存映射文件,训练时零教师前向;对齐采用方向一致(余弦)而非逐值复制,学生得以保留动作目标所需的额外结构,师生骨干也无需共享特征空间或维度——整个蒸馏从训练端和部署端都“不可见”。
核心方法
方法的直觉是“把老师的知识抄成小抄,上课时不带老师”。整体分两条线:离线阶段,把 Cosmos 3 全模态世界模型的理解塔(Cosmos3-Nano,抽出其中的 Qwen3-VL-8B 推理器)在全部训练帧上前向一次,取 LM 第 24 层图像 token 隐状态、按相机视角均值池化成每视角一个 $D_t=4096$ 维向量,以(轨迹 id,基索引)为键写入内存映射缓存,LIBERO 数据约花 4 张 GPU 一小时。在线阶段,0.8B 的 QwenGR00T 学生(Qwen3-VL 家族骨干 + GR00T 流匹配动作专家)照常训练,仅额外把各视角图像 token 的池化特征经两层 MLP 投影到 4096 维,与缓存目标做余弦损失;总目标为 $\mathcal{L}_{act}+\lambda_{align}\mathcal{L}_{align}$,$\lambda_{align}=0.5$。训练结束丢弃投影器,部署图与未蒸馏基线逐位相同:一次骨干 prefill 加四个流步骤,在 RTX 5090 上 32ms、1.86GB。
核心创新是“表征可以继承,生成机制可以丢弃”:世界模型物理接地的原因是预测未来的目标逼它内化时间结构与因果后果,而部署时昂贵的只是“滚动未来”这个动作,不是“知道未来”这份知识。与已有方法的区别有三点。第一,教师的身份:FitNets/REPA 对齐普通判别性编码器,本文对齐预测未来的世界模型;消融中 Fast-WAM(视频 DiT)与 V-JEPA2-AC(自监督编码器+动作条件预测器)同样有效,说明迁移的是世界模型表征的广谱先验,而非两个特定网络间的脆弱对齐。第二,零成本纪律:教师离线缓存、训练零教师前向、投影器用完即弃、部署图与基线同构(连流步数都相同),收益在构造上不可能来自容量或测试时计算。第三,刻意排除动作通道:师生不共享动作参数化,任何依赖教师动作目标的配方都无法做到教师无关,故教师的贡献全部经由内部状态进入。
方法步骤详情
第一步,离线教师预计算:从 Cosmos 3 检查点抽出 Qwen3-VL-8B 推理器,对全部训练帧做确定性前向,取 LM 第 24 层图像 token 隐状态,按相机视角均值池化得 $f_c^T\in\mathbb{R}^{4096}$,按键写入内存映射缓存,LIBERO 上 4 GPU 约 1 小时。第二步,学生训练:骨干输出前缀隐状态,用互不相交的位置集 $\mathcal{I}_c$ 池化各视角图像 token 得 $f_c^S$,经两层 MLP 投影器 $p_\phi$ 映到教师宽度,计算 $\mathcal{L}_{align}=1-\cos(p_\phi(f_c^S),\mathrm{sg}[f_c^T])$;动作专家照常做流匹配速度回归 $\mathcal{L}_{act}$,仅用真值演示动作监督;联合优化 $\mathcal{L}_{act}+0.5\mathcal{L}_{align}$,4×A100、DeepSpeed ZeRO-2、有效批 256。第三步,部署:删除投影器,只做一次 prefill 加四个流步骤,RTX 5090 上 32ms、1.86GB。
技术新颖性
新颖性有四层。其一,问题分解:把“世界模型的接地”与“世界模型的生成”解耦并证明前者可离线继承,直接反驳 DreamZero、LingBot-VA 等“先预测未来帧、再用其做动作条件”的 WAM 路线——它们把接地与生成绑在一起,因而支付推理税。其二,训练经济学:常规在线蒸馏需同时装载师生,本配方的蒸馏运行成本与基线完全相同;一个缓存还能服务多个学生,因为投影器对任意 $(D_s,D_t)$ 组合自动适配,师生甚至可以住在互不兼容的软件环境里。其三,教师无关性:Cosmos3-Nano(全模态 VLM)、Fast-WAM(视频 DiT)、V-JEPA2-AC(编码器-预测器)在训练目标、架构、特征宽度(4096/3072/1024)上全异却全部有效,证明信号是世界模型表征的普遍性质而非特定教师。其四,评估纪律:以 2 种子×2 GPU 四次评估报均值与标准差,直面 LIBERO/RoboCasa 已知的跨设备复现问题,这在同类论文中少见。
实验结果
LIBERO:蒸馏后 0.8B 平均成功率 97.9±0.5%,比未蒸馏 95.3±0.7% 高 2.6 点,超过 7B OpenVLA-OFT(97.1)与 6B Fast-WAM(97.6)。RoboCasa-GR1 24 环境:48.2±2.1% 升至 50.5±2.3%,越过 4B 的 QwenFAST(39.0)、QwenPI(43.9)、QwenOFT(48.8)与 Isaac-GR00T(47.6/48.2),距 4B QwenGR00T(54.8±2.0)仅 4.3 点,参数量只有五分之一。真机:单臂水果 93.3% 追平 4B π-style;鸡蛋 60.0%(未蒸馏 46.7%、4B 70.0%),收回 7 回合差距中的 4 个;双臂水果 46.7%(40.0%、53.3%)。教师消融:V-JEPA2-AC 96.5(+1.2)、Fast-WAM 96.9(+1.6)、Cosmos3-Nano 97.9(+2.6),均高于 95.3。规模消融:0.8B 50.3→52.8、InternVL 1B 50.9→53.3、4B 56.8→58.4。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LIBERO 四套件平均成功率 | SRavg (%),2 种子×2 GPU 四次评估取均值 | 97.9±0.5(0.8B 蒸馏学生) | 同结构未蒸馏 0.8B 学生 95.3±0.7;7B OpenVLA-OFT 97.1;6B Fast-WAM 97.6 | 对自身基线 +2.6 点;以 0.8B 参数超过 7B 基线 0.8 点 |
| RoboCasa-GR1 全部 24 环境平均成功率 | SR (%),每环境 20 回合,2 种子×2 GPU 四次评估取均值 | 50.5±2.3(0.8B 蒸馏学生) | 同结构未蒸馏 0.8B 48.2±2.1;4B 同架构 QwenGR00T 54.8±2.0 | 对自身基线 +2.3 点;0.8B 距 4B 同架构模型仅 4.3 点,且超过全部 4B 的 QwenFAST/QwenPI/QwenOFT |
| 真机 AgileX Nero 水果抓放(单臂) | 成功率(30 回合) | 93.3%(28/30) | 未蒸馏 0.8B 83.3%;4B π-style 策略 93.3% | 对自身基线 +10.0 点,追平 4B 策略 |
| 真机 AgileX Nero 鸡蛋抓放(单臂) | 成功率(30 回合) | 60.0%(18/30) | 未蒸馏 0.8B 46.7%;4B π-style 策略 70.0% | 对自身基线 +13.3 点,收回与 4B 差距(7 个回合)中的 4 个 |
| 真机 TRIP-Bag 双臂水果交接 | 成功率(30 回合) | 46.7%(14/30) | 未蒸馏 0.8B 40.0%;4B π-style 策略 53.3% | 对自身基线 +6.7 点;跨本体、相机与时延变化后增益仍保持 |
| 教师消融(LIBERO) | SRavg (%) | Cosmos3-Nano 97.9±0.5 / Fast-WAM 96.9±0.8 / V-JEPA2-AC 96.5±0.3 | 未蒸馏学生 95.3 | 三个架构、目标、特征宽度(4096/3072/1024)全异的教师分别 +2.6 / +1.6 / +1.2 点,全部有效 |
局限与改进
作者承认增益“温和但一致”:仿真上 2.3–2.6 点,真机 6.7–13.3 点但仍系统性低于 4B 模型;在 RoboCasa 上距最强已发表方法(ACE-Ego-0 72.8%、PhysBrain 64.5%)还有约 14–22 点差距。评估层面,LIBERO/RoboCasa 对 GPU 型号敏感,同一检查点换卡可漂移数点(作者因此用 2 种子×2 GPU 四次评估取均值缓解),而学生规模与层数消融因算力限制只在单张 A100 上跑四分之一日程且只评一次,绝对值低于主表、结论只到排序级;层数扫描非单调(末层 49.8% 最好、2/3 深度 L16 反而最差 45.9%),作者自己也不确定这是“对挂接位置宽容”还是短日程分辨率不足。我的补充观察:LIBERO 基线已达 95.3%,天花板效应削弱 +2.6 的含金量;四类失败模式全是放置与抓握类接触错误(松手过早、夹到篮沿、鸡蛋打滑、中途掉落),恰是纯视觉表征先验最帮不上的地方;教师特征逐帧独立提取,并未利用世界模型的时序预测能力本身;缓存需随数据集重新提取,且依赖教师推理的确定性。
独立分析的弱点
第一,接触密集任务收益有限:鸡蛋任务仅从 46.7% 提到 60.0%,失败原因(打滑、中途脱落)属于接触动力学,纯图像特征对齐不携带力与接触信息;改进方向是把对齐目标换成动作条件化的未来特征(V-JEPA2-AC 结果略好已是暗示),或对齐多帧时序特征以隐式编码动力学。第二,层数选择非单调且未被解释(L16 最差 45.9%、末层 49.8% 最好),改进方向是做跨数据集的层选择元研究,或同时对齐多层做加权和。第三,消融统计强度不足:层数与规模消融均为单卡单次、四分之一日程,“排序级”结论可能不稳,应加多种子重复并报告方差。第四,LIBERO 近饱和(基线 95.3%),主结论过度依赖 RoboCasa 与真机;应引入扰动类基准(如作者引文 [5] 的鲁棒性研究)直接检验“接地”是否兑现为抗扰动能力。第五,教师间差异(+1.2 到 +2.6)与蒸馏增益同量级,但“哪种世界模型特征更好”缺乏可解释判据,可探查特征几何性质(时序可预测性、各向异性)与下游增益的相关性。
未来方向
作者在结论中把增益归因于“广谱表征先验”,这暗示的直接延伸是:把配方铺到更多本体(移动操作、灵巧手、更多双臂任务)与 StarVLA 其他动作头家族(QwenFAST/QwenPI/QwenOFT),并做更大跨度的学生规模扫描。基于本文成果可继续推进:其一,把对齐目标从单帧视觉特征升级为动作条件化的未来特征,让“思考像世界模型”名副其实,有望攻克接触类失败;其二,在 RL 或在线微调中保留对齐项作正则,检验先验是否加速探索、稳定训练;其三,多教师集成——三个教师特征的凸组合或一致性约束,观察是否超过单教师上限;其四,研究缓存目标对数据增广的敏感性,以及“部分在线”提取的折中方案;其五,从理论上解释为何方向(余弦)而非幅值对齐有效、对齐深度为何非单调,为层数选择给出原则性判据。
复现评估
复现条件中等偏上。有利面:学生基于开源的 StarVLA 框架(模块化 Qwen 骨干 + 可换动作头);三个教师均为开放权重(NVIDIA Cosmos 3、Wan2.2-TI2V-5B、V-JEPA 2);教师缓存实现简单(内存映射、每视角一个池化向量),LIBERO 上 4 GPU 约 1 小时即可生成;评估协议(每任务 50 回合×4 套件、2 种子×2 GPU)写得足够细,且有项目页 thaw-vla.trung-dt.com。门槛:训练需 4×A100-80GB + DeepSpeed ZeRO-2、有效批 256;RoboCasa-GR1 全量数据 2.4 万回合、602 万步,训练成本不低;真机部分需要 AgileX Nero 与 TRIP-Bag 平台、约 30 分钟(单臂)与 1 小时(双臂)的遥操作数据,一般实验室难以完整复刻。另需注意仿真跨 GPU 方差:复现时必须按作者协议做多次评估取均值,否则单次结果可能偏离报告值数点。总体判断:仿真主结果复现性较好,真机结果基本只能依赖作者部署。
论文图表
四类失败模式(圆圈标出失败点): 水果在到达容器前掉落(转移失败); 合爪过早、夹在篮沿而非水果上(困难抓取失败); 鸡蛋从爪中滑脱; 蛋中途掉落后滚走。作者指出这些全部是放置/接触错误而非识别错误——策略总能找对目标物体并伸手去够。
界定了方法的适用边界:表征先验改善场景理解,但对接触动力学(打滑、时机、力度)几乎无帮助,是分析局限性与设计后续工作的关键证据。
LIBERO 四套件(SPATIAL/OBJECT/GOAL/LIBERO-10)成功率汇总:VLA 策略从 7B OpenVLA 76.5 到 S2-VLA 98.2、Qwen-RobotManip 99.2;世界模型策略 Fast-WAM(6B)97.6、Light-WAM(2B)97.2;本文未蒸馏 0.8B 为 95.3±0.7,蒸馏后 97.9±0.5(99.3/99.2/99.4/93.8),均报四次评估的均值与标准差。
主结果表:显示 +2.6 点增益让 0.8B 学生越过几乎所有 <4B 模型,并逼近或超过 6B/7B 的 VLA 与 WAM,是“小模型大精度”主张的直接证据。
RoboCasa-GR1 24 环境成功率:4B 级基线包括 QwenFAST 39.0、QwenPI 43.9、Isaac-GR00T N1.6 47.6、Isaac-GR00T N1.5 48.2、QwenOFT 48.8、VP-VLA 53.8、TwinBrainVLA 54.6、QwenGR00T† 54.8±2.0、ABot-M0 58.3、PhysBrain 64.5、ACE-Ego-0 72.8;0.8B 未蒸馏 48.2±2.1,蒸馏后 50.5±2.3。
贡献隔离表:同一学生开关对齐项的直接对照(+2.3),并给出与同架构 4B 模型的精确差距(4.3 点)以及与最强已发表方法的剩余差距,诚实呈现方法的能力范围。
真机三任务每格 30 回合成功率:4B π-style 策略在 AgileX Nero 水果/鸡蛋与 TRIP-Bag 双臂水果上分别为 93.3/70.0/53.3;未蒸馏 0.8B 为 83.3/46.7/40.0;本文蒸馏 0.8B 为 93.3/60.0/46.7——单臂水果追平 4B,鸡蛋与双臂任务收回大部分差距。
证明仿真增益迁移到真实硬件,且在换本体、换相机、换控制器时延的多重变化下仍保持,是“增益是方法的性质而非某个特定设置”论断的关键支撑。
学生规模与骨干族消融(单张 A100 训练):Qwen3.5-VL 0.8B 50.3→52.8、InternVL 1B 50.9→53.3、Qwen3-VL 4B 56.8→58.4;每个规模与骨干加对齐项都提升约 1.6–2.5 点。
证明配方对学生宽度与骨干族即插即用(投影器自动适配维度),收益不是某个特定学生架构的巧合,支撑“广谱表征先验”的主张。
对齐层深度扫描(四分之一日程、单次评估):1/3 深度 L8 为 47.5%、1/2 深度 L12 为 49.4%、2/3 深度 L16 为 45.9%、末层 L24 为 49.8%;排序非单调,末层最好且因复用动作头已消费的表征而实现最省。
展示方法对挂接位置的宽容度,同时诚实报告未解释的非单调性;对实现者而言“选末层”既最好又最便宜,是复现时最重要的工程结论。
三个异构教师对比:Cosmos3-Nano(全模态 VLM,16B 检查点中的 8B 塔,LM 第 24 层,Dt=4096)达 97.9±0.5;Fast-WAM(Wan2.2-TI2V-5B 视频 DiT,DiT 第 15/30 层,Dt=3072)达 96.9±0.8;V-JEPA2-AC(ViT-g/16 编码器 1.0B + 动作条件预测器 0.3B,预测器 norm 层,Dt=1024)达 96.5±0.3;未蒸馏控制为 95.3。
支撑全文最核心的科学主张:有效的是“世界模型表征”这一普遍类别,而非与 Cosmos3-Nano 的特定对齐——教师选择只影响增益幅度(+1.2 到 +2.6),不决定成败。