Cortex 2.0:面向真实工业部署的世界模型落地 Cortex 2.0: Grounding World Models in Real-World Industrial Deployment
用视觉隐空间世界模型给VLA加上前瞻规划,实现工业机械臂零人工干预
前置知识
视觉-语言-动作模型 (VLA)
VLA是把摄像头图像和语言指令映射到机器人动作的大型序列模型,如RT-1、RT-2、π0。它从互联网级视觉-语言数据继承语义知识,实现跨任务、跨本体泛化,但本质上是反应式的:每个动作只根据当前观测选取,不评估未来后果。
本文的Cortex 2.0正是在自家VLA基础上扩展世界模型规划模块,理解VLA的反应式局限才能理解本文动机。
世界模型 (World Model)
世界模型是对环境动态的预测模型,给定当前状态和动作能想象未来状态。Dreamer在隐空间进行想象式rollout,UniSim、Cosmos在互联网视频上预训练获得物理先验。本文的世界模型在视觉隐空间用流匹配预测未来隐向量序列。
Cortex 2.0的核心是把世界模型生成的候选未来轨迹在动作前打分筛选,是理解plan-and-act范式的关键。
流匹配 (Flow Matching)
流匹配是一种生成建模方法,学习噪声到数据之间的直线插值速度场,相比扩散策略的去噪过程能大幅减少推理步数。给定插值 $x_\nu=\nu x+(1-\nu)\varepsilon$,模型预测速度 $u_\nu=x-\varepsilon$,推理时从噪声积分ODE生成。
本文的动作头和世界模型都用流匹配实现,理解其训练目标和ODE积分推理是读懂方法节的前提。
过程奖励算子 (PRO)
PRO是本文提出的稠密奖励模块,在真实部署轨迹上训练,对世界模型想象出的隐向量rollout打分。它有三个头:进度头估计任务推进量、风险头预测失败概率、终止头预测成功完成概率,聚合成复合分数选出最优轨迹。
PRO是Cortex 2.0规划环路的评分核心,理解它如何从真实数据迁移到想象轨迹是理解全文的关键。
研究动机
工业级机器人操作对可靠的长时程执行要求极高:动作不可逆,失败会沿长时程累积,一次未恢复的错误就能中断整条产线工作流。近年的VLA模型(如π0、π0.5)虽在跨任务、跨本体上展现强泛化,但本质上是反应式的——每步动作只根据当前观测选取,不显式推理未来后果。这导致它们对长时程任务的复合失效模式极为脆弱:在退货处理这类任务中,渐进滑移、卡阻、碰撞等失效往往要走几步之后才显现。仓储操作还充满托盘和包装造成的频繁遮挡、挑战RGB感知的反光与半透明表面、以及班次和站点间快速变化的物体分布。作者用四个真实任务上的实验证明,反应式基线(π0.5、Diffusion Policy、RDT-2)在杂乱、遮挡、富接触环境中频繁进入不可恢复状态,需要大量人工干预。
本文的目标是本文目标是把机器人控制从反应式的try-and-see转变为plan-and-act:在每个决策步,先由世界模型在视觉隐空间生成一组候选未来轨迹,用PRO模块针对任务进度、风险可能性和完成可能性打分,只承诺得分最高的候选轨迹,从而在行动前评估潜在未来而不是直接执行第一个可用动作。具体希望在单臂和双臂平台上,跨越四个复杂度递增的任务(拾放、物品与垃圾分拣、螺丝分拣、鞋盒拆包)超越SOTA的VLA基线,达到全部任务最高成功率,并实现零人工干预。同时希望世界模型在视觉隐空间运作从而实现跨本体(单臂、双臂、人形)的规划环路迁移,且能利用持续从live运营收集的部署数据形成闭环改进。
与已有工作不同的是,本文的独特切入角度在于把世界模型从训练时的rollout生成器(如Dreamer,会累积模型误差)转变为推理时的候选筛选器,并且让PRO评分函数落地于真实工业部署数据。关键洞察是想象出的未来必须与真实观测处于同一表征空间(视觉隐空间),这样在真实执行轨迹上学到的PRO评分函数才能直接迁移到想象轨迹。与IRASim、GPC等并行工作只在受控数据上scoring rollout不同,Cortex 2.0把世界模型训练grounding在从live运营连续收集的部署录像上(30Hz、超过5亿次操作交互),形成越好的世界模型带来越准的规划、越高的执行质量、越干净的训练信号的闭环增益循环,这是它区别于纯学术系统的工业落地视角。
核心方法
Cortex 2.0是一个四层分层VLA系统:高层VLM观测并编码场景、世界模型生成候选未来、PRO评估并排序、流匹配动作头承诺最高分轨迹。直觉上,它在动作前先在脑中想象若干条可能的未来轨迹,用一个学到的裁判(PRO)挑出最靠谱的一条,再让策略去实现它。技术路线上,机器人在每步接收多模态观测 $o_t=(I^{rgb}_t, r_t, f_t, l_t)$(RGB、机器人状态、力反馈、指令嵌入),编码为视觉隐向量 $z_t=f_{enc}(o_t)$。高层VLM产出结构化任务上下文 $s_t$;世界模型在规划horizon $H_{wm}$ 上生成 $k$ 条候选隐向量轨迹;PRO对每条打分选出 $\tau^*$,其二值化优势指标 $I_t\in\{0,1\}$ 条件化策略 $\pi_\theta$ 输出执行horizon $H_{act}$ 上的动作块。
核心创新是把世界模型的想象rollout在动作执行前用PRO筛选,且整个规划在视觉隐空间进行。与已有方法的本质区别有三:其一,世界模型不是训练时rollout生成器而是推理时前瞻规划器,通过 $\tau^*=\arg\max_{\tau_j}S_j(z_t,s_t)$ 选出最优候选,避免了Dreamer式的训练误差累积;其二,PRO在真实部署轨迹(有真实结果标签)上预训练后冻结,因为世界模型产出的隐向量与真实观测同处一个视觉隐空间,PRO的评分能直接迁移到想象隐向量上;其三,规划在视觉空间进行,同一套规划环路无需修改即可跨单臂、双臂、人形本体迁移,本体差异完全由轻量Action Mapping适配器处理。
方法步骤详情
第一步,多模态观测 $o_t$ 经VLM视觉编码器得隐向量 $z_t$,高层VLM产出任务上下文 $s_t=f_{hl-VLM}(z_t)$,编码子目标分解与语言-场景实体的grounding变量。第二步,世界模型 $f_\phi$ 在视觉隐空间用流匹配学习预测动态,训练时对每个未来隐向量 $z_{t+h}$ 采样流时间 $\sigma\sim Beta(\alpha,\beta)$($\beta\gg\alpha$偏向高噪声以便少步ODE积分即可评分),构造插值 $\tilde z^{(h)}_\sigma=\sigma z_{t+h}+(1-\sigma)\xi^{(h)}$,目标速度 $v^{(h)}=z_{t+h}-\xi^{(h)}$,用流匹配损失 $L_{WM}$ 训练。推理时对 $k$ 个候选各采独立噪声 $\xi^{(j)}$ 积分ODE生成完整候选序列 $z^{(j)}_{t+1:t+H_{wm}}$。第三步,PRO对每条rollout $\hat\tau_j$ 用时序模型产出表征,三个头分别算进度 $\Delta p^{(j)}=V_\phi(z^{(j)}_{t+H_{wm}})-V_\phi(z_t)$、风险 $\rho^{(j)}=P_\phi(fail=1|\hat\tau_j)$、终止 $d^{(j)}=P_\phi(success=1|\hat\tau_j)$,聚合为复合分 $S_j=\Delta p^{(j)}-\lambda\rho^{(j)}+\beta d^{(j)}$。第四步,选最优 $z^*_{token}=\arg\max_j S_j$,优势 $\Delta^*=S(\tau^*)-\frac{1}{k}\sum_j S(\tau_j)$ 二值化为 $I_t=1[\Delta^*>\epsilon(s_t)]$。第五步,把 $z^*_{token}$ 和 $I_t$ 投影后与任务上下文拼接为 $c_t=[s_t;z_t;W_z z^*_{token};W_I I_t]$,流匹配动作头据此生成动作块,Action Mapping适配器输出机器人专属指令。
技术新颖性
技术新颖性体现在几方面。首先是把稠密过程奖励模型PRO直接lift进规划环路:PRO在真实执行轨迹的运营telemetry(力矩、真空压力、接触信号)上监督训练,学到的进度/风险/终止信号能迁移到世界模型想象出的隐向量——这种迁移之所以成立,是因为世界模型产出的隐向量与真实观测同属一个视觉隐空间。其次是训练分两阶段并采用知识绝缘(knowledge insulation)方案:阶段一PRO单独在部署数据上预训练然后冻结,阶段二世界模型和动作头联合优化 $L_{total}=L_{FM}+\lambda_{wm}L_{WM}$,且阻断世界模型和动作头对预训练VLM骨干的梯度。第三是规划预算 $k$、horizon $H_{wm}$、$H_{act}$ 全是推理时超参不影响训练,可按任务代价灵活调整——代价高的打包用高 $k$,重抓这类廉价恢复用低 $k$。第四是把动作表述为未来状态而非原始控制指令,配合Action Mapping适配器实现跨本体部署。
实验结果
在四个真实任务上Cortex 2.0全面超越所有基线且零人工干预。实验一单臂拾放(160微调episode、16次trial):Cortex 2.0成功率0.98、平均完成时间20s、0次干预,而π0.5为0.7/49s/2次、Diffusion Policy 0.56/53s/4次、RDT-2最低0.4/63s/7次。实验二物品与垃圾分拣(双臂、每策略10 rollout、源箱10-15个物体):Cortex 2.0每操作成功率0.95、任务完成700s、0干预,而所有基线在每个rollout都需人工干预且都在15分钟限制内无法完成整个任务(π0.5 0.61/53次干预、Diffusion 0.47/59次、RDT-2 0.18/95次)。实验三螺丝分拣(双臂):Cortex 2.0每操作成功率0.98、180s、0干预,此任务相对基线性能差距最大,π0.5仅0.4/24次干预、Diffusion 0.2/16次、RDT-2完全失败0.0/50次干预。实验四鞋盒拆包(四步序列、更少演示数据):Cortex 2.0成功率0.96、58s、0干预,π0.5 0.6/103s/5次、Diffusion 0.12/52s/9次、RDT-2 0.0/62s/10次。规划预算消融显示成功率随 $k$ 从1个rollout的0.962升至30个的0.996,但每步时间从310ms线性增至9200ms,正式评估固定低延迟 $k=2$。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 单臂拾放 | 成功率(16 trial均值) | 0.98 | π0.5: 0.7 | +0.28,且干预0 vs 2 |
| 物品与垃圾分拣 | 每操作成功率(10 rollout) | 0.95 | π0.5: 0.61 | +0.34,干预0 vs 53且基线均超时 |
| 螺丝分拣 | 每操作成功率 | 0.98 | π0.5: 0.4 | +0.58,干预0 vs 24 |
| 鞋盒拆包 | 全任务成功率(四步) | 0.96 | π0.5: 0.6 | +0.36,干预0 vs 5 |
| 规划预算消融 | 成功率 vs 时间 | k=30: 0.996/9200ms | k=1: 0.962/310ms | 成功率+0.034,代价时间增约30倍 |
局限与改进
作者承认当前评估只覆盖了本体和任务族的子集,规划horizon $H_{wm}$ 和预算 $k$ 是按任务固定的,缺乏在线自适应和不确定性感知的动态预算分配。世界模型训练在计算和数据上还有很大扩展空间,更锐利的预测才能给条件化策略提供更有信息量的信号。我的额外观察:所有实验只在Universal Robot单臂/双臂平台上做,论文反复宣称的人形和跨本体迁移并无实验支撑;基线只用200 GPU-hour统一预算训练,而Cortex 2.0得益于超过5亿次交互的海量私有部署数据预训练,这种对比虽在微调数据上公平,但预训练规模的巨大差异使得成功率优势难以单独归因于世界模型规划本身;此外PRO的三个头权重 $\lambda$、$\beta$ 和阈值 $\epsilon(s_t)$ 的选取、以及世界模型隐空间预测的定量保真度指标都未给出,rollout评分质量缺乏独立验证。
独立分析的弱点
第一个弱点是延迟-成功率的严峻权衡:$k=30$ 时每步高达9200ms在30Hz实时控制场景几乎不可用,正式评估退回 $k=2$;改进方向是学习一个自适应预算控制器,仅在高风险决策点(如打包)动态提高 $k$,其余时刻用最小预算。第二个弱点是私有数据不可复现且预训练规模与基线严重不对等,超5亿次交互的部署数据是Sereact独有壁垒;改进方向是公开一个受控的、预训练规模对齐的消融,剥离出世界模型规划的净收益。第三个弱点是PRO作为冻结裁判可能存在分布外脆弱性——它在真实执行轨迹上训练却应用于想象隐向量,若世界模型在新任务上生成不真实的隐向量,PRO评分可能失效;改进方向是加入世界模型预测的置信度/不确定性估计并在PRO打分时校准。第四个弱点是跨本体和人形的宣称缺乏实验证据,仅在UR单双臂验证;改进方向是补做真实人形平台的迁移实验。
未来方向
作者提出的未来方向包括:大幅扩展世界模型训练的计算与数据规模以提升预测rollout保真度,因为策略条件化于选中的未来隐向量,更锐利的预测直接改善生成的动作块;强化在线自适应和PRO中不确定性感知的动态预算分配;收紧视频tokenization与控制的耦合以支持更长时程前瞻;验证从视频演示的in-context learning,即让策略 $\pi_\theta(a|o,\tau_{demo})$ 条件化于一段演示视频从而在测试时对未见任务族免训练执行。基于本文成果可延伸的方向:把力反馈和真空压力等多模态telemetry更深地融入世界模型预测而不仅是PRO监督;探索把闭环部署数据反馈用于持续在线微调世界模型和PRO的自动化pipeline;研究规划预算与任务失败代价的形式化关联以实现代价敏感的最优 $k$ 选择。
复现评估
复现难度极高。核心资产是Sereact私有的部署数据集:超过1000万条被用于训练的交互(从超5亿次运营交互中curated采样)、约4万episode/400小时遥操作数据,这些均不开源,是复现的根本壁垒。训练用等价200 GPU-hour预算,但Cortex 2.0还依赖在互联网级视频上预训练的世界模型,实际算力远超此数。方法描述层面公式较完整(观测编码、世界模型流匹配训练/推理、PRO三头评分、动作头流匹配、Action Mapping),但许多关键超参未给出:$\lambda$、$\beta$、阈值 $\epsilon(s_t)$、世界模型ODE积分步数、VLM骨干具体规模(仅提及2B-VLM动作头)。基线中Diffusion Policy和RDT-2用了公开数据集(Open X-Embodiment、BridgeData V2、DROID、RoboCasa合成)。总体上论文更像工业系统报告,学术复现几乎不可能,但方法思想(世界模型隐空间rollout+过程奖励评分+流匹配动作头)对有自有机器人数据的团队具备参考价值。
论文图表
全文系统总览图,展示Cortex 2.0从感知到规划到控制的整体框架和plan-and-act范式。
作为开篇总览帮助读者建立对四层分层系统的整体认知。
展示从仓储部署以30Hz连续收集的真实生产机器人数据,含RGB、力矩、真空压力、本体感知等多模态观测。
体现本文grounding于真实工业部署数据的核心卖点。
螺丝分拣每螺丝成功率:Cortex 2.0 0.98/180s/0干预,π0.5 0.4/24、Diffusion 0.2/16、RDT-2 0.0/50,性能差距最大。
精密操作任务展示PRO前瞻过滤不稳定接触的直接收益。