Ctrl-World:为通用策略的 policy-in-the-loop 回放而生的多视角可控世界模型(论文 Figure 1)

配图说明:本文收录 9 张论文插图(/images/ctrlworld/fig1–9)+ 官方 GitHub 演示媒体 9 件(横幅 JPG、2 个长程对比 GIF、2 组三视角合成轨迹 MP4),论文图逐张讲解,GitHub 媒体在第 4 章内嵌播放。论文 Ctrl-World: A Controllable Generative World Model for Robot Manipulation(arXiv 2510.10125,ICLR 2026),斯坦福 + 清华(Yanjiang Guo、Lucy Xiaoyang Shi 共同一作,Jianyu Chen,Chelsea Finn);官网 ctrl-world.github.io;官方实现 github.com/Robert-gyj/Ctrl-World(575★,MIT,同时包含后续 VLAW 论文的世界模型后训管线)。


📄 论文信息

  • 标题:Ctrl-World: A Controllable Generative World Model for Robot Manipulation
  • 机构:Stanford University、Tsinghua University
  • 时间:2025-10 挂 arXiv(2510.10125),ICLR 2026 正式发表
  • 关键词:可控世界模型、policy-in-the-loop、多视角(含腕部)、帧级动作条件、位姿条件记忆、指令跟随评测与改进、DROID、SVD 初始化
  • 一句话总结:给 VLA 配一个「能一起玩」的世界模型——多视角联合预测 + 厘米级动作控制 + 20 秒长程记忆,让策略在想象里被测(排名≈真机)与被改(合成成功轨迹 SFT,成功率 +44.7%)。

🎯 一句话版本:核心数字

指标数值说明
想象排名 ≈ 真机排名指令跟随 y=0.87x−0.04;成功率 y=0.81x−0.11π0/π0-fast/π0.5 × 7 任务,真机 vs 世界模型散点(Fig. 7)
策略改进平均 +44.7%(0.39→0.83)合成成功轨迹 → SFT(Fig. 9)
分任务涨幅Spatial 0.29→0.88;Shape 0.44→0.93;Towel-Dir 0.57→0.80;Novel-Obj 0.25→0.73四类指令理解短板
训练数据DROID 95k 轨迹 / 564 场景全开源可复现
长程生成新场景/新相机位姿下 >20 秒 连贯多视角位姿条件记忆检索
动作控制精度厘米级(帧级条件注入 1 秒动作块)同初始帧 + 不同动作序列 → 不同轨迹
底座Stable Video Diffusion(SVD) 初始化适配为可控、时间一致的世界模型
官方仓库Robert-gyj/Ctrl-World,575★,MIT4 条推理跑法 + 训练/后训(VLAW)

✨ 创新点:论文自己认领的四条 + 为什么难

先看它要打的靶:此前的动作条件世界模型有三宗罪——①单视角导致部分可观测与接触幻觉(碗没碰就「进爪」),也不匹配现代 VLA 的多相机输入;②缺高频动作条件,动作与画面的因果对不上;③长程不一致,误差累积漂移(相关工作 §2 归纳)。

论文的四条创新(按论文声明整理):

  1. 首提「可控多视角世界模型 × policy-in-the-loop」工作流:核心贡献(原文 core contribution)是让 SOTA VLA 官方权重(π0/π0-FAST/π0.5)在世界模型里闭环跑——策略出动作、模型出未来、交替自回归,全程想象空间。附录 B 原话:first world model that enables policy-in-the-loop interactions between state-of-the-art VLA models;
  2. 三机制把「被动视频生成器」改造成「策略兼容的交互模拟器」(原文 transformation 语):
    • 联合多视角预测——含腕部,显著减少接触密集交互的幻觉,同时满足 VLA 输入格式;
    • 帧级动作条件——视觉动态与控制信号逐帧对齐,rollout 反映每个动作的因果效果(厘米级);
    • 位姿条件记忆检索——稀疏历史帧 + 位姿入帧,让模型「 attend to similar past states」,稳住 20 秒+ 长程;
  3. 想象力工作流闭环(评测 + 改进双用途):评测——想象中的指令跟随排名与真机对齐(回归线 0.87/0.81);改进——想象里合成成功轨迹 → SFT → 下游任务成功率 38.7%→83.4%(+44.7pp),全程不采一条真机演示;
  4. 零样本迁移的实用边界:只用开源 DROID 训练 → 新场景、新相机位姿上 20 秒+ 连贯 rollout——作者自建平台与验证集双零样本(策略与模型同时迁移)。

一句话创新点:不是让世界模型更像视频,而是让它「听动作的话、记得住过去、接得住策略」——三个机制各治一宗罪,换来一个能测能改 VLA 的仿真替身。

🗺️ 仿真 vs 真机:一眼分清

场景类型说明
世界模型内 rollout(回放/键盘/π0.5 进环)生成式「想象」评测不是物理仿真器——视频扩散模型自回归造未来
真机 rollout(π0.5 七任务 × 20 次)真机(对照组)用来验证「想象排名=真机排名」(Fig. 6/7)
合成成功轨迹 → SFT真机策略改进改进幅度在真机上验收(Fig. 9,+44.7%)
训练/后训(DROID 预训练 + VLAW 下游后训)离线数据8×A100/H100 级别;后训见仓库(3)节

一句话定位:世界模型是「生成式评测器」——它既不是 MuJoCo 那种物理仿真,也不是纯视频玩具;有效性的证据就是 Fig. 7 的两条回归线。


🧭 读前 5 分钟:四个概念

  1. Policy-in-the-loop(策略进环):世界模型不止预测「给定动作的未来」,还要能接住真策略的输出再生成下一步——闭环里策略与模型互相为对手/学生;本文是首个让 SOTA VLA(π0/π0-FAST/π0.5 官方 DROID 权重)在世界模型里闭环交互的工作(附录 B 原话)。
  2. 多视角联合预测(含腕部):现代 VLA 吃第三人称 + 腕部相机;腕部视野在一条轨迹里剧烈变化,单视角世界模型要么幻觉(物体无接触就「粘」进夹爪)要么不兼容策略输入。
  3. 帧级动作条件(frame-level action conditioning):把动作转成笛卡尔位姿后逐帧注入生成过程——同初始帧、换动作序列就换轨迹,精度到厘米级。
  4. 位姿条件记忆检索(pose-conditioned memory retrieval):长程自回归会漂;按位姿检索历史帧进上下文,并把位姿嵌入帧做交叉注意——预测第 4 秒时会「回看」第 0 秒的相似位姿帧(Fig. 4 左侧注意力可视化就是证据)。

📐 问题形式化:两条方程的自回归闭环

现代通才策略 π 把 n 路观测 + 语言 映射成 H 步动作块:

$$o_t = [I^1_t, \dots, I^n_t, q_t], \qquad a_{t+1:t+H} \sim \pi(\cdot \mid o_t, l) \tag{1}$$

世界模型 W 的任务是预测执行这串动作后的多视角未来,把结果喂回策略形成闭环:

$$o_{t+1:t+H} \sim W(\cdot \mid o_t, A_t), \qquad A_{t+H} \sim \pi(\cdot \mid o_{t+H}, l) \tag{2}$$

如此策略 ↔ 世界模型交替自回归,长程 rollout 全程发生在想象空间——既不需要物理仿真器,也不需要真机。要让 (2) 成立,W 必须同时做到三件事:多视角一致、动作可控、长程不漂——这正是下面三个改造要解决的。


🛠️ 方法详解:SVD 底座上的三个改造(原文 §4.1 + 附录 A)

4.1 底座与训练目标

  • 底座:预训练 1.5B Stable Video Diffusion(SVD)——时空 Transformer(spatial-temporal transformers)视频扩散骨干,继承其全部视频先验;
  • 唯一新参数:一个 3 层 MLP,把 7 维笛卡尔空间动作投影到 1024 维潜变量——除它之外初始化时参数全不动;
  • 编码:图像过 8×8 空间下采样 VAE;3 路相机各 192×320 → 潜特征 24×40;
  • 扩散训练(x0 预测式,Eq. 3):目标帧 $x_0=o_{t+1:t+H}$ 加噪 $x_{t'}=\sqrt{\alpha_{t'}}x_0+\sqrt{1-\alpha_{t'}}\epsilon$,模型输入 = [历史帧 tokens ‖ 加噪未来 tokens],条件 c = [历史位姿 + 未来动作位姿 + 历史帧],损失:
$$L = \mathbb{E}_{x_0,\epsilon,t'} \| \hat{x}_0(x_{t'}, t', c) - x_0 \|_2^2$$
  • 训练配方(附录 A):95k DROID 轨迹全量;每步随机采轨迹→随机采当前帧→向过去采记忆帧→预测其后未来帧;2×8 H100、batch 64、lr 1e-5、100k 步、约 2–3 天。

4.2 改造一:多视角联合预测

N 路输入图像(每帧 H×W 个潜 token)沿 token 维拼接,一次联合去噪所有视角的未来 $o_{t:t+H}$。联合预测不只是「顺便多出两个视角」——论文观察它本身提升一致性并大幅减少幻觉:腕部「看」到的接触事件可以由第三人称视角的几何信息背书(反之亦然)。数量关系:每帧 token 数 P = N × H_latent × W_latent。

4.3 改造二:位姿条件的记忆检索

  • 稀疏历史:向过去按步长 m 采 k 帧(实装 k=7,历史帧间隔 1–2 秒)——上下文被硬性限长;
  • 位姿入帧:把机械臂位姿 $[q_{t-km},\dots,q_t]$ 通过帧级交叉注意力(frame-wise cross-attention)注入空间 Transformer 的对应帧——每个视觉 token 组「知道」自己帧的本体位姿;
  • 检索语义:模型用「位姿相似度」从历史里抓相关帧,把未来预测锚回曾经见过的相似状态(re-anchoring)——预测 t=4s 帧时注意力强指回 t=0s 的同位姿帧(Fig. 4 左图的注意力热图),这就是「检索」的可视化证据。

4.4 改造三:帧级动作条件

  • 预训练 SVD 只认文本+图像,控制精度为零——Ctrl-World 追加动作条件:
  • 动作块先转成笛卡尔位姿序列 $[a'_{t+1:t+H}]$,与历史位姿 $[q_{t-km},\dots,q_t]$ 拼成一条位姿序列;
  • 同一套帧级交叉注意力分工:历史帧的视觉 token 注意自己帧的 q,未来帧的视觉 token 注意自己帧的 a’——「每帧一个位姿 token($B\times1, C_a$)」,把高频动作钉到对应帧上(消融里去掉它,腕部 PSNR 从 19.18 掉到 15.69——三改造里对控制精度贡献最大)。
  • 15 步进、5 步算:DROID 动作块 15 步(1 秒),为省显存先转笛卡尔再时间下采样到 5 步喂给模型——控制窗口不变、序列变稀。

4.5 Token 账本(架构一图流)

总输入形状:$B \times (7\ \text{历史帧} + 5\ \text{未来位姿帧}) \times (3\ \text{视角} \times 24 \times 40)$——12 个「帧位置」每个装 2880 个视觉 token,再叠 CLIP 语义 token(文本)、历史/未来位姿嵌入($B\times1\times C_a$);进入 SVD 的 Spatial Transformer(B×P,T,C)→ Temporal Transformer(B×T,P,C) 交替栈(Fig. 2 右侧结构图即此账本)。历史帧的噪声是独立采样的(对鲁棒性的又一个小设计)。

4.6 策略桥:从关节速度到世界模型的位姿接口(附录 B)

官方 openpi 策略吃关节角+双视角、吐关节速度;世界模型要笛卡尔位姿——桥接三步:

  1. Adapter(2 层 MLP,在 DROID 上训练):$q^{joint}_{t+1:t+H} = \text{Adapter}(q^{joint}_t, a^{jv}_{t+1:t+H})$——把「当前关节+速度指令」积成未来关节构型;
  2. 正运动学:$q^{cartesian} = \text{FK}(q^{joint})$(Franka Panda FK);
  3. 喂模型:$o_{t+1:t+H} = \text{WM}(o_t,\ q^{cartesian}_{t+1:t+H},\ q^{cartesian}_{history})$。

这条链让 π0/π0-FAST/π0.5 的官方权重零改动进环——「policy-in-the-loop」的工程关键就在这个 2 层 MLP 上。


🗂️ 数据集:DROID 怎么用(平台 / 规模 / 拆分 / 为什么带失败)

维度内容
平台DROID(Khazatsky et al. 2024):Franka Panda + Robotiq 夹爪;1 腕部相机 + 2 个随机摆位第三人称相机(作者评测用同款布局自建平台)
规模95,599 条轨迹 / 564 个场景——含 约 76k 成功 + 约 19k 失败轨迹
失败数据是资产论文明说:多样动作与失败数据对训练可控世界模型至关重要——模型要能模拟各种未来(包括失败走向),策略进环才不会一遇「没见过的烂摊子」就崩
训练采样每步随机采一条轨迹 → 随机采当前帧 → 向过去采记忆帧 → 目标 = 其后未来帧(附录 A)
验证划分留出 2% 轨迹作验证集;质量评测从中随机 256 段 × 10 秒
改进阶段用数不用演示!每任务自采 400 条想象 rollout → 人工偏好挑 25–50 条成功者 → SFT 2k 步(数据是「自己想出来的」)
预训练配置2×8 H100、batch 64、lr 1e-5、100k 步、2–3 天;三相机 192×320、历史 7 帧(间隔 1–2s)、动作窗 1s=15 步(进模型前降到 5 步)

一句话:DROID 提供「密集动作 + 失败样本 + 多相机」的原料,Ctrl-World 只花 2–3 天把它炖成可交互的替身;改进阶段连真机都不再碰——数据完全来自想象。

📸 逐图讲解(论文 9 张)

Figure 1:总览——策略进环的多视角世界模型(头图)

(即文首头图。)左侧通用 VLA 策略与右侧世界模型构成闭环:初始帧 + 动作块进模型 → 联合多视角未来(含腕部)吐回策略;标注两大用途——(1) 想象中评测指令跟随 (2) 对未见指令做定向改进。读图抓手:「多视角 + 帧级控制 + 位姿记忆」三件套是闭环能转起来的前提,缺任何一个都会在多步交互里散架。

Figure 2:从 SVD 到可控世界模型

初始化管线(论文 Figure 2)

SVD 预训练权重 → 适配为可控、时间一致的世界模型:动作、多视角、记忆等条件通路被接入原视频扩散骨干。工程含义:站在 8GB 级 SVD 肩膀上——生成质量靠底座,可控性靠适配,这是小团队能训起世界模型的现实路径(与 Sonic/SONIC 系「站在大规模预训练上」的哲学一致)。

Figure 3:长程 rollout 定性对比(验证集)

长程对比(论文 Figure 3)

Ctrl-World 与单视角前作在验证集长程回放上的多视角帧对比:前作单视角预测易出幻觉/漂移,本文多视角联合在 20 秒级轨迹上保持场景与物体恒常。配 Table 1(交互式长程生成的量化指标)阅读——定性图给「肉眼连续性」,表格给分数。

Figure 4:可控性与消融——厘米级换动作换轨迹

可控性(论文 Figure 4)

同一初始帧 + 不同动作序列 → 不同轨迹的多组对照,叠加消融可视化(去帧级条件/去多视角的退化)。这是「Controllable」的字面证据:控制粒度到厘米级,而不是「大方向差不多」。

Figure 5:腕部一致性

腕部一致性(论文 Figure 5)

单条轨迹内腕部相机视野剧变下的多视角一致性对照:腕部预测不是附属品——它既是现代 VLA 的输入刚需,也是「近距接触是否发生」的裁判;多视角联合显著降低腕部引发的不一致(配伍 Table 2 消融)。

Figure 6:π0.5 真机 vs 世界模型(20 次交互)

π0.5 真机 vs 想象(论文 Figure 6)

每条轨迹含 π0.5 与环境的 20 次交互,上真机、下世界模型逐段对照——真机成功/失败的行为模式(何时抓偏、指令走到哪一步)在想象里被复刻。这是 Fig. 7 定量相关的定性注脚:不是像素级复刻,是「策略行为级」复刻。

Figure 7:真机-世界模型定量相关

相关性(论文 Fig. 7)

两张散点:指令跟随 y=0.87x−0.04、成功率 y=0.81x−0.11(虚线=oracle);图例 = π0/π0-fast/π0.5 × 7 任务(Pick-Place、Fold-Towel、Drawer、Wipe-table、Close-laptop、Pull-tissue、Stack)。结论一句话:不用真机 rollout 也能给策略正确排名——世界模型评测有效性的核心证据。

Figure 8:后训任务与合成轨迹

后训与合成(论文 Fig. 8)

上排:后训任务(新物体/新指令等短板场景);下排:世界模型内生成的合成轨迹。这是「改」的原材料——在想象里把失败任务的成功解法造出来。仓库 README 的两个 GIF(真机 vs 世界模型 20 秒并排)就是这条线的动态版,见第 4 章。

Figure 9:策略改进 +44.7%

改进柱图(论文 Fig. 9)

Base vs Finetuned 五组柱:Spatial 0.29→0.88、Shape 0.44→0.93、Towel-Dir 0.57→0.80、Novel-Obj 0.25→0.73、Average 0.39→0.83(+44.7%)。四个维度恰好对应「空间理解、形状理解、方向指令、新物体」四类指令跟随短板——合成数据专打语言接地的软肋(配 Table 4–7 分任务细表)。


🧪 实验详读:质量、可控性、评测、改进四条证据链

先给目录再进细节——论文一共 6 组实验,回答 §5 开头的三问(生成质量?排名可信?能否改进):

#实验协议一句话指标结论一句话
E1世界模型质量(Table 1)验证集 256 段×10s,每轮 15 步动作、自回归 10 轮 vs 真值;对照 WPE/IRASim/自家单视角版PSNR/SSIM/LPIPS/FID/FVD全文版全指标第一(PSNR 23.56、FVD 97.4)
E2组件消融(Table 2)去记忆 / 去帧级条件 / 去联合预测,分第三人称与腕部两组视角同上五指标腕部最敏感;帧级条件单点影响最大
E3可控性(Fig. 4)同初始帧换厘米级动作序列(±2–6cm、±15° 旋转)多组对照 + 注意力可视化轨迹分叉度 / 注意力分布厘米级可控;注意力指回同位姿历史帧
E4长程一致性(Fig. 5)腕部视角 10 秒 rollout,绿/红框标「跨视角推断 vs 记忆检索」来源定性两通路共同维持腕部连贯
E5想象中评测(Fig. 6/7 + 附录表)自建新 DROID 场景(新相机位姿)× 3 策略(π0/π0-FAST/π0.5 官方权重)× 7 任务 × 每条 20 次交互;真机与想象同初始观测;人工偏好标注成败指令跟随率、成功率 + 相关回归IF 相关 0.87、SR 相关 0.81;系统性低估执行 SR(三条差距见 5.3)
E6想象中改进(Fig. 8/9)LLM 改写指令 + 随机重置位姿扩多样;每任务 400 条想象 rollout 人工挑 25–50 成功 → π0.5 SFT 2k 步4 组任务成功率38.7% → 83.4%(+44.7pp),仅对新指令/新物体成立

逐任务样例(附录表节选,真机 vs 想象):Pick-Place 上 π0.5 指令跟随 0.90/0.80、成功率 0.85/0.70;π0 为 0.75/0.60、0.70/0.55——排序一致、数值被低估正是回归线 0.87/0.81 的微观形态;Fold-Towel 上 π0.5 0.80/0.75(IF)与 Drawer 全员低分(π0 仅 0.05/0.00)说明难任务上想象里同样难,不是放水。

5.1 世界模型质量(Table 1/2,10 秒长程)

协议:留出 2% 轨迹作验证集,随机 256 段 × 10 秒;每轮给 15 步(1 秒)动作块、自回归 10 轮生成 10 秒,与真值比 PSNR/SSIM/LPIPS/FID/FVD。

方法PSNR↑SSIM↑LPIPS↓FVD↓
WPE-Single-View20.330.7720.131156.4
IRASim-Single-View21.360.7740.117138.1
Ctrl-World 单视角版21.270.7930.110127.5
Ctrl-World(全文)23.560.8280.09197.4

单视角版已胜出,多视角联合再抬一截——「联合预测本身有益」的定量版。

消融(Table 2):第三人称:去记忆 23.56→23.06,去帧级条件→21.20;腕部:全文 19.18 / 去记忆 18.84 / 去联合预测 15.94 / 去帧级条件 15.69——腕部是三改造的主战场,帧级动作条件是最大单点功臣。

5.2 可控性与一致性(Fig. 4/5 的定量脚注)

  • 厘米级:只差 2–6cm 的不同动作序列生成不同轨迹(归因:DROID 密集动作覆盖 + 多视角 + 帧级条件);去记忆→画面变糊(蓝),去位姿条件→控制精度掉(紫);
  • 注意力证据:预测 t=4s 帧时强注意 t=0s 同位姿帧——记忆检索不是玄学,是可画出来的注意力边;
  • 腕部一致性(Fig. 5):绿框 = 由其他视角推断补全,红框 = 从记忆检索复用——两条通路共同维持腕部视野剧变下的连贯。

5.3 想象中评测(Fig. 6/7 协议细节)

  • 零样本新场景:作者自建 DROID 平台、新相机位姿——预训练只用开源 DROID,策略与世界模型双双零样本迁移;
  • 三家策略:官方 openpi 的 π0-DROID / π0-FAST-DROID / π0.5-DROID;7 任务(Pick-Place、Fold-Towel、Drawer、Wipe-Table、Close-Laptop、Pull-tissue、Stack),每条 20 次交互;
  • 判定:真机与想象同一初始观测起跑;成功率/指令跟随由人工偏好标注(VLM 当奖励模型留作 future work——诚实声明);
  • 结果:指令跟随 y=0.87x−0.04、成功率 y=0.81x−0.11;系统性低估执行成功率——三条已知差距:①复杂物理(碰撞/滑移/旋转,Fig. 6 合盖不精确)②真机「失败后反复重试」行为想象里没复现 ③数据分布外的失败模式缺失——作者预期补策略自身的 rollout 数据能收窄差距。

5.4 想象中改进(Algorithm 1 逐行读)

  1. 多样性是搜索的前提:策略在同初始条件下高度确定(总抓同一个物体)→ 两种结构化扰动:(i) LLM 改写指令(“place glove in box” → “pick up the cloth and put it inside the box”)、(ii) 随机重置初始位姿(线性插值规划器挪过去);
  2. 循环:对每任务 i,跑 N 轮 rollout(策略带扰动采样 ε_a)→ 拼历史上下文 h → W 预测 → 轨迹入 τ → 人工偏好判成败 → 成功者进合成集 D_s;
  3. 微调:$L_\theta = \mathbb{E}_{o_t,a \sim D_s}\|\pi_\theta(o_t, l) - a_{t:t+H}\|^2$——每任务 400 条 rollout,只留 25–50 条人工判定成功,再训 2k 步;
  4. 成绩(Fig. 9):π0.5 在陌生指令/物体上 38.7% → 83.4%(+44.7pp)——分维:Spatial 0.29→0.88、Shape 0.44→0.91、Towel-Dir 0.57→0.80、Novel-Obj 0.25→0.75、平均 0.39→0.83。改进只对「新指令」成立——作者自陈模型还不够准去提升已见指令的低层成功率(见结论边界)。

🧰 官方 GitHub 详解:4 条跑法 + 训练/后训

仓库:Robert-gyj/Ctrl-World(575★,MIT,ICLR 2026 官方 PyTorch 实现;同时收录 VLAW(arXiv 2602.12063)的世界模型后训管线)。

资产清单(README「CheckPoint and Dataset」表)

资产来源大小
CLIP text/image encoderHF openai/clip-vit-base-patch32~600M
SVD 视频扩散底座HF stabilityai/stable-video-diffusion-img2vid~8G
Ctrl-World ckpt(DROID 训练)HF yjguo/Ctrl-World~8G
DROID 数据集HF cadene/droid_1.0.1(95k 轨迹/564 场景)~370G

环境:conda Python 3.11 + requirements.txt;玩 π0.5 进环需另按 openpi 装 JAX 依赖(GIT_LFS_SKIP_SMUDGE=1 uv sync)。

跑法 ① DROID 轨迹回放(合成长视频)

从记录轨迹里采一帧作初始,反复喂 1 秒动作块自回归出完整 rollout:

CUDA_VISIBLE_DEVICES=0 python scripts/rollout_replay_traj.py \
  --dataset_root_path dataset_example --dataset_meta_info_path dataset_meta_info \
  --dataset_names droid_subset --svd_model_path ${svd} --clip_model_path ${clip} \
  --ckpt_path ${ctrlworld_ckpt}
  • 配置集中在 config.py 的 __post_init__;
  • 速度提示(README 原文):A100 约 10 秒/交互步、H100 约 5 秒——一步=1 秒动作块;
  • 本章贴的两组三视角 mp4(dataset_example/droid_new_setup 与 ..._full/drawer 的 val 样本)就是这条管线的产出物形态(第三人称 + 腕部并排)。

跑法 ② 键盘交互(人在环造轨迹)

键盘命令逐条转动作块、自回归顺序执行,可一次输入多键:

... python scripts/rollout_key_board.py ... --task_type keyboard --keyboard lllrrr

合法键:l/r/f/b(左右前后)、u/d(升降)、o/c(开合夹爪)——官网 Interactive Demo 的网页版同款逻辑(见官网,另附 keyboard_first_frame*.png 初始帧素材)。

跑法 ③ π0.5 策略进环(policy-in-the-loop 的字面实现)

从新开 DROID 场景快照出发,让官方 π0.5-DROID 与世界模型闭环互动(两者都零样本迁移);任务类型:pickplace / towel_fold / wipe_table / tissue / close_laptop / stack:

CUDA_VISIBLE_DEVICES=0 XLA_PYTHON_CLIENT_MEM_FRACTION=0.4 \
python scripts/rollout_interact_pi.py ... --task_type pickplace

工程细节:π0.5 是 JAX 实现,XLA_PYTHON_CLIENT_MEM_FRACTION=0.4 防止 JAX 预占满显存——这是把两家生态(PyTorch 世界模型 + JAX 策略)焊在一起的关键垫片。作者声明:只在开源 DROID 上训模型、新场景全零样本;可评指令跟随,但物理交互可能不准。

跑法 ③-New 论文评测协议(20 次/类)

论文口径:每类任务 20 次(5 或 10 个初始条件 × 重复 2 或 4 次),初始条件在 dataset_example/droid_new_setup_full;对应脚本:

... python scripts/rollout_interact_pi_eval.py ... --task_type fold_tower

跑法 ④ 训练:预训练(DROID)与后训(VLAW)

  1. 潜空间预抽(视频扩散在图像编码器潜空间训练,先抽潜提速):accelerate launch dataset_example/extract_latent.py ...;
  2. 元信息:create_meta_info.py 生成 json、算状态/动作归一化;
  3. 训练:accelerate launch scripts/train_wm.py ...(仓库自带小子集 droid_subset 可先冒烟);论文规模 = 1–2 节点 × 8×A100/H100;
  4. 后训(VLAW,2026-02 新增):接触密集/可变形任务预训练不够准时,按下游任务后训——后训模型支持长程策略进环,README 给出「同一初始条件下真机 20 秒 vs 世界模型 20 秒」的上下行对比 GIF(本章下方内嵌);更多视频见 VLAW 项目页(arXiv 2602.12063)。

仓库结构速记:scripts/(rollout_replay / rollout_key_board / rollout_interact_pi(_eval) / train_wm)+ dataset_example/(数据样例与潜空间工具)+ synthetic_traj/gallery/(README 演示媒体)+ config.py(全局配置)。致谢链:SVD(生成底座)+ openpi(π0.5 策略)。


🎬 演示媒体(GitHub 原样收录)

效果横幅(README 首图)

Ctrl-World 生成轨迹画廊(GitHub README)

同一世界模型生成的多场景轨迹拼画廊——README 开篇即挂它,对应论文多视角合成能力。

效果 1:真机 vs 世界模型 · 长程 20 秒并排(后训模型,上下行)

上=真机、下=世界模型,同一初始条件各 rollout 20 秒(VLAW 后训效果,README 官方素材):

真机上/世界模型下 对照 GIF 1(GitHub)

真机上/世界模型下 对照 GIF 2(GitHub)

读法:盯住物体被移动的关键帧上下对齐程度——行为级一致(不是像素级克隆),这正是 Fig. 6 定量实验的动态版。

效果 2:DROID 新场景三视角合成轨迹(回放管线产出)

dataset_example/droid_new_setup val 样本(第三人称主视角 / 腕部视角 / 第三视角,各约 2–4 秒):

效果 3:抽屉场景三视角(droid_new_setup_full/drawer)

(仓库 dataset_example/ 内还有 13+ 组同类三视角样本,均为跑法①的产物;官网另有键盘交互与对比视频合集——见 ctrl-world.github.io 的 Interactive demos 与 comparison 区。)


💡 评析:三个亮点,三条边界

三个亮点

  1. 把「评测」做成了回归线:Fig. 7 的 0.87/0.81 相关是世界模型从「演示玩具」变「评测基础设施」的通行证——排名一致比像素一致更便宜也更有用。
  2. 改进闭环短且可复现:合成成功轨迹 → SFT → +44.7%,全程不碰专家标注;四类短板(空间/形状/方向/新物体)全部受益——打的是 VLA 的「语言接地软肋」而非重新练操作。
  3. 开源彻底:官方实现 + ckpt(HF)+ 数据管线 + π0.5 进环脚本 + VLAW 后训——四条跑法把论文实验全部落成命令行。

三条边界

  1. 物理可信度:README 自陈「物理交互可能不准」——粘吸、穿透类幻觉在多视角下缓解但未根除;测「会不会做」可靠,测「力学是否对」仍需真机;
  2. 算力门槛不低:全量训练 8×A100/H100 起步,推理一步 5–10 秒——适合评几个关键策略,不适合海量在线评测;
  3. 世界-策略共同漂移风险:用同一个模型既出题又改卷(合成数据 SFT 后再回模型里验),过拟合到模型偏好的解法需要靠真机终验把关(论文 Fig. 9 有真机验收,应用时别省这步)。

❓ 自测 Q&A

Q1:Ctrl-World 和 MuJoCo/Isaac 这类仿真器差在哪?
仿真器解物理方程(准但难长程好看、要资产),Ctrl-World 是视频扩散生成(好看、能零样本新场景、但物理可能不准)——它是评测/改进的快速回路,不是物理金标准;两者互补(真机终验仍是金标准)。

Q2:为什么腕部视角这么关键?
现代 VLA 输入就含腕部;且腕部视野剧变是长程一致性的最大压力源——腕部预测稳了,「夹爪里到底有没有东西」才有裁判(单视角幻觉的病根)。

Q3:+44.7% 是在想象里测的吗?
不是——改进幅度在真机上验收(Fig. 9/Table 4–7);想象负责「造数据」,真机负责「盖章」。

Q4:想最快上手仓库?
下载 HF yjguo/Ctrl-World ckpt + 仓库自带 droid_subset → 跑法①回放看 mp4 → 跑法②键盘玩交互 → 有 π0.5 环境再上跑法③。

Q6:动作块明明 15 步,为什么模型只喂 5 步?
15 步转笛卡尔位姿后再时间下采样到 5 步(附录 A)——控制窗口仍是完整 1 秒,但序列变稀省显存;细节靠帧级交叉注意把稀疏位姿钉回对应帧,而不是靠密集动作采样。

Q7:记忆检索的「按位姿找」具体怎么实现?
历史帧与未来帧各带一个位姿 token,在空间 Transformer 里做帧级交叉注意力——预测第 4 秒时对第 0 秒同位姿帧的注意力显著拉高(Fig. 4 左侧热图);效果是把预测「锚回」曾见过的相似状态,去记忆后 PSNR/清晰度同步下降(Table 2)。

Q8:适配器只有 2 层 MLP,为什么说是工程关键?
因为官方 π 系策略输出关节速度、世界模型输入笛卡尔位姿——语义不同、坐标系不同;2 层 MLP + Franka FK 把两者焊死,三家官方权重才零改动进环(附录 B 公式链)。没有这条桥,policy-in-the-loop 只能停留在论文示意图。

Q5:和 VLAW 什么关系?
同一作者线:Ctrl-World 是「可控世界模型」本体,VLAW(2602.12063)是「策略与世界模型互相迭代改进」的扩展——官方仓库(3)节的后训就是 VLAW 管线,README 的双行 GIF 即其效果。


📊 图表总索引(9 张论文图 + 9 件 GitHub 媒体)

图/媒体内容本站文件
Figure 1(头图)策略进环总览本文插图
Figure 2SVD → 可控世界模型本文插图
Figure 3长程多视角定性对比本文插图
Figure 4厘米级可控性 + 消融本文插图
Figure 5腕部一致性本文插图
Figure 6π0.5 真机 vs 想象 20 次交互本文插图
Figure 7真机-世界模型相关回归(0.87/0.81)本文插图
Figure 8后训任务 + 合成轨迹本文插图
Figure 9策略改进 +44.7% 柱图本文插图
GitHub 横幅README 生成轨迹画廊本文插图
GIF ×2真机 vs 世界模型 20 秒并排本文插图
MP4 ×6两组三视角合成轨迹(视频内嵌)第 4 章 <video> 播放器

(Table 1–7 与 Algorithm 1 的关键数字已以文字表格复现。)


📚 资源与延伸