【通用世界模型】把机器人动作变成像素流!NVIDIA 提出 Hydra-0:跨人类、单臂与双臂统一预测和控制

同样是"把布往左折",人类用双手完成,UMI 用手持夹爪完成,Franka 用单臂完成,双臂机器人又有另一套关节与控制器。

动作语义相同,底层控制语言却完全不同:关节数量不同、末端坐标不同、运动学约束不同,甚至数据中可能根本没有机器人状态和相机标定。

这正是通用机器人世界模型面临的一个基本矛盾:互联网与机器人平台已经积累大量交互视频,但只要模型以关节角、末端位姿或某台机器的 action token 为条件,数据就会被 embodiment 切成互不兼容的小岛。

NVIDIA、Brown、Columbia 与 Harvard 团队提出 Hydra-0,试图用一种统一的视觉动作接口打通这些数据:

不告诉视频模型机器人关节怎么转,只告诉它画面中的可见表面点将怎样移动。

作者将这种表示称为 Action Flow。它由一组带可见性的二维像素轨迹组成,可以描述机械臂连杆、夹爪、人手,也可以描述布料、绳索、管线等被操作对象。

由于所有动作都被投影到当前相机的图像平面,人类、UMI、单臂、双臂和仿真机器人第一次可以使用同一种条件格式训练同一个视频世界模型。模型不需要理解每种机器人的原生 action space,只需要学习:这些可见点这样移动时,场景接下来会发生什么。

Hydra-0 不止把 action flow 用作视频控制信号,还构造了一个完整闭环:

  • 训练时: 从机器人元数据或普通交互视频中提取 action flow;

  • 部署时: 在 Isaac Lab 中执行候选机器人命令,通过几何模型与相机标定投影出 flow;

  • 前向模式: 根据机器人 flow 预测动作造成的真实场景变化;

  • 策略评估: 回放不同策略的动作流,用生成结果判断成功与失败;

  • 逆向模式: 输入来自人类演示的目标物体 flow,模型反推出相容的机器人运动,再由 action head 解码为可执行动作。

其训练语料覆盖 7 个来源、178,187 个 episode、2,201.7 小时视频 ,包含人手、手持夹爪、单臂、双臂和人形机器人。过滤后得到约 156.6 万个五秒窗口

最佳配置相较使用原生 6D 末端动作的 Cosmos 2.5 基线,将平均机器人运动误差降低 **90.4%**,物体运动误差降低 **60.2%**。在 RoboLab 的 5 个策略、6 个任务、300 条轨迹上,生成成功率与真实成功率的 Pearson 相关系数达到 r=0.96 ,Spearman 相关系数为 ρ=0.93 ,平均绝对误差只有 5.7 个百分点

更有意思的是,模型出现了一种"反向使用方式":给它一段从人类演示中抽取的柔性管目标运动,不提供机器人 flow,它可以生成能够造成该物体运动的机器人行为,并通过一个学习得到的动作头在真实机器人上执行。

这篇论文真正想回答的不是"视频模型能否画出机器人",而是:

像素运动能否成为机器人世界模型的 USB 接口,让不同本体、不同数据和不同任务共享同一种动作语言?

📷 核心图表

图1|一条 Action Flow,连接人类、UMI、双臂、单臂与三类下游能力

注:不同 embodiment 的可见运动被转化为图像平面点轨迹,统一进入 Hydra-0。前向模式用于世界模拟与策略评估,逆向模式将人类物体运动转成机器人控制。

图2|训练时从视频恢复 flow,部署时从可执行命令因果生成 flow

注:离线训练可利用多形态视频;在线部署则先在 Isaac Lab 中滚动候选控制命令,再结合机器人几何、相机内外参和可见性,将表面点轨迹投影到图像平面,作为真实视频预测条件。

图3|只给目标物体怎么动,模型生成机器人怎么动

注:从 held-out 人类柔性管弯折演示中提取 object flow;Hydra-0 根据物体运动生成相容的机器人运动,action head 读取内部 DiT 特征并输出真实机器人动作。

📑 核心信息提炼

  • 核心接口: 可见性建模的二维 Action Flow

  • 视频骨干: Cosmos 2.5、Wan2.2 TI2V-5B、Wan2.2 I2V-A14B

  • 训练语料: 7 个来源、178,187 episodes、2,201.7 小时

  • 主要能力: 跨 embodiment 世界预测、数据高效适配、加速因果 rollout、开放环策略评估、物体流条件控制

核心战绩:

  • 平均机器人运动 EPE:Cosmos 2.5 的 34.28 → 最佳模型 3.29,下降 **90.4%**。

  • 平均物体运动 EPE:13.23 → 5.27,下降 **60.2%**。

  • 平均 PSNR:15.62 → 21.84 ;SSIM:0.668 → 0.830

  • 平均 FVD:405.8 → 155.9

  • 4-step 学生模型:61.98 FPS ,相对双向 teacher 为 16.0× generation-only 加速。

  • RoboLab 开放环评估:Pearson r=0.96,Spearman ρ=0.93,MAE 5.7 个百分点。

  • 多形态中训后,IWS 六项未见任务在 0% 目标数据时均获得更强迁移;主要增益通常在 20% 数据前出现。

❓ 通用世界模型的 4 个"动作语言障碍"

  • 关节动作绑定本体结构: 七自由度单臂与双臂、人形机器人的 action vector 维度和语义不同,无法自然合并训练。

  • 末端位姿没有描述整机可见运动: 相同的末端轨迹可能对应不同连杆姿态,视频模型还要额外学习从控制量到像素运动的 embodiment-specific 映射。

  • 大量视频没有机器人元数据: 人类演示、UMI 和互联网交互视频通常没有 URDF、关节状态或精确相机标定,难以转换为标准机器人动作。

  • 视频条件与真实可执行命令脱节: 随手画出的轨迹可以控制画面,却未必满足机器人的关节限制、碰撞约束和控制器动态,因而不能直接用于可信动作评估。

Hydra-0 的目标,是让动作条件同时满足两件事:在视觉空间中统一,又能在部署时追溯到可执行机器人命令。

🔧 核心真相:终极拆解 Hydra-0

1. Action Flow 究竟是什么?

在预测长度 H 内,作者选择 N 个可见点。第 n 个点在时刻 t 的图像位置为:

xₙ,ₜ = (uₙ,ₜ, vₙ,ₜ)

并用二元变量 mₙ,ₜ 表示该点是否可见。Action Flow 写作:

F = {τₙ}ₙ₌₁ᴺ, τₙ = {(xₙ,ₜ, mₙ,ₜ)}ₜ₌₀ᴴ

它不是一张稠密光流图,也不是单个末端轨迹,而是一组覆盖可见机器人表面或物体区域的稀疏时空点轨迹。

相比关节动作,它不暴露 embodiment-specific 控制空间;相比单点末端轨迹,它能描述连杆、夹爪、人手或整个物体的可见运动;相比完整 mask 视频,它更加稀疏,并显式保留每个点跨时间的对应关系。

2. 有机器人几何时:把候选命令投影成像素运动

当已知机器人几何、状态和相机标定时,Hydra-0 在初始帧可见的机器人表面采样三维点。部署时,候选命令先在 Isaac Lab 的控制器和物理仿真中执行,得到各机器人 link 随时间的变换。

每个表面点通过机器人正运动学与相机内外参投影到图像平面:

xₙ,ₜ = π(K · T_CW · T_link(qₜ) · Xₙ)

系统还用深度缓冲进行可见性检查,只保留相机前方、图像范围内且与渲染深度一致的点,从而处理机器人自身遮挡。

这一步是 Hydra-0 与一般轨迹控制视频模型的关键区别:给定的 flow 不是视觉上随意指定,而是由候选可执行命令经过控制器、运动学和物理 rollout 后产生。因此,世界模型负责预测的是同一动作在真实视觉世界中的物体后果,而不是替候选动作解决机器人可达性。

代价也很明确:前向部署需要机器人模型、相机标定和一个可生成本体运动的仿真器。Action Flow 统一了视频模型的接口,但没有取消每台机器人的底层工程。

3. 没有机器人元数据时:直接从视频追踪和分割

大规模人类或机器人视频往往缺少 URDF 与标定。对此,Hydra-0 使用 AllTracker 恢复稠密二维轨迹和可见性,再用 grounded masks 将轨迹分成 acting embodiment 与 manipulated object。

训练时有四种 flow 采样模式:

  • Embodiment: 只采样机器人、夹爪或人手轨迹,是主要动作条件;

  • Object: 采样被操作物体轨迹,可作为目标运动条件;

  • All: 从 embodiment、object 与未分配场景轨迹的并集中采样,用于语义分割不完整的情况;

  • None: 移除运动条件,作为 conditioning dropout。

默认采样概率分别为 0.40、0.40、0.15 和 0.05;Embodiment/Object 模式采样 1-128 条轨迹,All 模式采样 256-1024 条。

需要注意,训练视频可以利用"未来跟踪结果"构造条件,但部署时不能偷看未来。论文明确要求在线 flow 必须从候选可执行命令因果生成。

4. 轨迹不是直接画到 RGB 上,而是在潜空间搬运首帧特征

Hydra-0 先将初始图像编码成空间 latent。对每条轨迹,从其初始位置双线性采样对应特征,再沿轨迹未来位置传播。每个目标 latent cell 聚合附近轨迹的首帧特征,并配套一个 presence gate,告诉模型哪些位置受运动条件控制。

简化地说:如果初始帧中的夹爪局部特征为 hₙ,模型就按轨迹权重把 hₙ 搬到未来位置:

Mₖ(p) = Σ wₙ,ₖ(p) · hₙ

权重由轨迹可见性和高斯距离决定。最终 motion condition 由传播特征 M 与 presence gate g 共同构成,并在 DiT patch embedding 前与 noisy video latent 拼接。

这种设计比在 RGB 上画彩色线更自然:视频模型收到的不只是"这里有条轨迹",而是"初始画面中这个局部外观应该沿此轨迹移动"。

5. 同一接口可适配不同视频骨干

作者没有把 Action Flow 写死在单一架构里,而是分别接入:

  • Cosmos 2.5 2B;

  • Wan2.2 TI2V-5B;

  • Wan2.2 I2V-A14B;

  • 蒸馏后的 Wan2.2 I2V-A14B 4-step 模型。

Cosmos 2.5 使用额外 17 通道 motion side input;Wan2.2 TI2V-5B 使用 49 通道输入;I2V-A14B 则复用其视觉条件路径。视频骨干主要冻结,只训练 DiT 输入投影与 attention/FFN 上 rank-64 LoRA。

这说明论文贡献更接近一种可移植条件接口,而非依赖某个特定生成器的专属模块。

🗃️ 2,201.7 小时数据:七个来源如何合成一个世界模型?

数据集 Embodiment Episodes 过滤后窗口 小时
DROID 单臂 70,339 223,075 313.7
ABC-130k 双臂 54,961 1,048,681 1,474.7
MolmoAct2 双臂 7,723 126,335 177.7
EgoDex 人手 41,888 89,380 125.7
Deform360 手持 UMI 夹爪 1,714 60,315 84.8
XVLA-Soft-Fold 双臂 1,524 17,772 25.0
H1-Fold-Clothes 人形双臂 38 76 0.1
总计 多形态 178,187 1,565,634 2,201.7

所有视频被重采样为 480p、16fps,并切成不重叠的 81 帧窗口,每个约 5 秒。团队重点保留布、绳、线缆、袋子和纸张等可变形物体交互,因为这些任务很难用显式状态重建和传统物理仿真精确覆盖。

数据清洗包括:过滤运动不足的静态窗口、夹爪几乎不动的窗口,以及语言描述不包含可执行内容的 DROID episode。

最佳 Wan2.2 I2V-A14B 配置使用 32 张 NVIDIA H100、40,000 steps、训练 5 天。这显示 Hydra-0 获得的是强大的数据与表示统一性,不是低成本训练方案。

📊 关键实验一:Action Flow 比原生 6D 动作强多少?

论文在 5 个 held-out 数据集上各使用 100 段验证视频,共 500 段,指标覆盖视觉保真、物体运动、机器人运动、图像/视频分布和 VLM judge。

为了让表格可读,下面列出跨数据集无权平均结果:

方法 PSNR↑ SSIM↑ Obj. EPE↓ Grip. EPE↓ FID↓ FVD↓ VLM↑
ATI(zero-shot) 17.01 0.700 23.19 4.62 36.4 444.2 3.14
Wan-Move(zero-shot) 16.35 0.688 21.53 4.67 34.4 408.3 3.71
Cosmos 2.5 原生 6D action 15.62 0.668 13.23 34.28 39.1 405.8 3.88
Hydra-0 / Cosmos 2.5 2B 18.41 0.725 6.27 13.80 32.4 277.4 3.83
Hydra-0 / Wan2.2 5B 19.64 0.770 6.61 3.88 24.1 248.8 3.90
Hydra-0 / Wan2.2 A14B 20.76 0.805 6.00 3.83 20.7 193.7 3.98
Hydra-0 / A14B 4-step 21.84 0.830 5.27 3.29 18.7 155.9 4.23

最有控制力的对照不是 A14B 与 2B 之间的比较,而是同一 Cosmos 2.5 骨干:把原生相对 6D 末端动作换成 Action Flow 后,五个数据集的 PSNR、SSIM、gripper EPE、FID 和 FVD 全部改善;有物体标注的四个数据集上,object EPE 也全部改善。

这支持论文的核心判断:视频模型更容易理解与画面对齐的运动条件,而不是先从机器人控制量猜测将出现的像素运动。

论文宣传的 90.4% 与 **60.2%**,是用最佳模型平均 EPE 与原生 Cosmos 2.5 action baseline 比较:

  • gripper EPE:34.28 → 3.29,下降 90.4%;

  • object EPE:13.23 → 5.27,下降约 60.2%。

但不能把全部增益都归因于 Action Flow,因为最佳模型还换用了更大的 Wan2.2 A14B 骨干和蒸馏设置。真正隔离动作表示贡献的,是 Cosmos 2.5 内部的受控对照。

VLM 分数也并非每个数据集都由 Hydra-0 稳胜。作者解释,VLM judge 评价物理合理性、时间一致性、物体恒存和运动真实性,并不直接测量是否严格服从给定轨迹;EPE 才是动作条件跟随度更直接的指标。

🧪 关键实验二:多形态预训练能否减少新任务数据?

作者在完全排除于训练语料之外的 IWS 数据集上测试 6 个任务:

  • Bimanual box;

  • Bimanual rope;

  • Bimanual sweep;

  • Push-T;

  • Single chain-in-box;

  • Single grasp。

目标任务数据比例从 0%、1%、2%、4% 一直到 100%。Ours (PT) 直接从原始 Wan2.2 初始化,Ours (MT) 则从多形态 Action Flow 中训 checkpoint 初始化。

在 0% 目标数据的真正 zero-shot 点,MT 在六项任务的 LPIPS、object-flow EPE 和 FVD 上都优于 PT。这说明跨人手、UMI 和多种机器人训练出的交互动态,可以在完全没见过 IWS 任务时迁移。

适配后优势仍然存在。在 100% 数据时,MT 的 6 个任务 LPIPS 与 FVD 全部最低,object EPE 有 4 项最低,另外两项也非常接近最佳。

更重要的是,多数收益在前 20% 数据到达:从 20% 增加到 100%,单任务指标变化最多约为 LPIPS 3.4%、flow EPE 6.7%、FVD 6.8%。这支持"多形态中训降低新任务数据需求",但作者也谨慎指出 FVD 每个点仅估计 40 个 clip,方差较高,20% 之后的非单调变化不能证明严格收敛。

⚙️ 关键实验三:视频世界模型能跑多快?

Hydra-0 将长视频生成从整段双向去噪改成 chunk-wise causal autoregressive rollout。已生成 chunk 进入 KV cache,后续只生成新片段;随后使用 DMD2 将每个 chunk 蒸馏为 4 步采样。

测试设置为单张 80GB H100、bfloat16、batch size 1,生成 81 帧 480×832 视频:

模型阶段 秒/clip↓ FPS↑ 相对加速
Bidirectional teacher 20.92 3.87 1.0×
Autoregressive teacher 12.48 6.49 1.68×
Few-step student 1.31 61.98 16.0×

这个速度很亮眼,但必须连同测试条件一起阅读:

  • 关闭 classifier-free guidance;

  • 不包含 VAE 解码和像素输出时间;

  • 只报告生成部分,不是完整机器人系统端到端延迟;

  • 使用三 chunk 的短 horizon,长 rollout 的 attention cap 未触发;

  • 逆向动作模型实验仍使用 50-step autoregressive teacher,而不是 4-step student。

因此,"62 FPS"证明蒸馏后的 latent rollout 很快,不等于当前 Hydra-0 已能以 62Hz 完成感知、规划、生成、解码与电机控制闭环。

📈 关键实验四:能否在生成世界中评估机器人策略?

RoboLab 实验覆盖 5 个预训练策略:π₀、π₀.₅、GR00T N1.7、Cosmos-3 Edge 和 Cosmos-3 Nano;每个策略在 6 个任务上各运行 10 次,总计 300 个 episode。

Hydra-0 从每条真实 rollout 的首帧开始,接收由完整已记录末端轨迹构造的因果 Action Flow,生成对应未来。盲于策略身份的人类评分者判断生成结果是否成功。

预测成功率与环境真实成功率得到:

  • Pearson r = 0.96

  • Spearman ρ = 0.93

  • mean absolute error = 5.7 个百分点

  • 对任务求平均后,五个策略的整体排名全部复现。

这说明 Hydra-0 对不同质量的动作能保留足够多的结果差异,可作为开放环 neural simulator 评估策略。

但这里的 "policy evaluation" 不是闭环仿真。论文固定回放完整真实动作轨迹,策略不会看到模型生成画面,也不会根据生成误差重新决策。因此,它测试的是"给定已实现动作,生成结果是否保持成功/失败",尚未验证模型误差与策略反馈耦合时的闭环稳定性。

作者还展示了真实布料折叠中成功与失败轨迹的定性回放,但没有给出真机大样本统计。

🔄 最吸睛的逆向能力:从人类物体运动到机器人动作

Hydra-0 的同一 flow 接口允许交换条件实体。

前向模式输入 embodiment flow,预测 object motion;逆向模式只输入 desired object flow,不给机器人 flow,要求世界模型生成与该物体目标运动相容的机器人运动。

随后,一个轻量 action head 读取干净 DiT token 特征,预测目标机器人的 action 和 state。这个动作头使用配对真实机器人 rollout 训练,但这些轨迹不需要是任务专家示范:只要记录了实现某种物体运动时的机器人视觉与动作对应关系,就能学习 embodiment-specific 的潜特征到控制命令映射。

论文在一个柔性管弯折任务上展示流程:

  1. 从 held-out 人类演示抽取管线的 object flow;

  2. 不提供任何机器人 flow,让 Hydra-0 生成机械臂运动;

  3. action head 从内部 latent rollout 解码动作;

  4. 真实机器人执行并成功弯折柔性管。

"无需任务专属专家机器人示范"并不等于"完全零机器人数据"或"无需动作监督"。action head 仍然需要配对的真实机器人轨迹学习控制映射;论文只是没有为这个具体管线任务收集专家机器人 demonstration。

而且当前证据只有一个任务的定性 proof-of-concept,没有多次试验成功率、不同物体和不同人类演示的统计。它展示了一个很有潜力的 emergent inverse mode,但还不是通用模仿学习结果。

🆚 Hydra-0 与 Masked Visual Actions 有什么区别?

两者都把机器人动作搬到视频原生坐标系,并都探索"给机器人运动预测物体"与"给物体运动反推机器人"的统一。

主要区别在表示粒度与部署落地:

维度 Masked Visual Actions Hydra-0 Action Flow
条件形式 实体像素区域随时间显露 稀疏点轨迹 + 可见性 + 首帧特征传播
空间密度 稠密 mask / 实体外观 稀疏、带跨帧对应关系
训练数据构造 分割或 URDF 渲染 几何投影或视频跟踪 + grounded mask
前向部署 提供或渲染实体轨迹 候选命令经 Isaac Lab 与标定投影
逆向输出 生成机器人视频,再由 IDM 提取动作 DiT latent 直接由 action head 读出动作
实验重点 一个 checkpoint 的前向/逆向交换 多形态规模训练、跨骨干、加速和策略评估

这两篇工作不是简单替代关系。MVA 强调完整实体条件的稠密约束,Hydra-0 强调轨迹对应、几何可执行性和大规模异构数据统一。未来很可能出现 mask、surface flow、depth、contact 与 robot state 的混合接口。

💬 你可能最关心的 8 个问题

Q1:Action Flow 就是光流吗?

不完全是。它是选定可见点在多帧中的轨迹及可见性,可以通过跟踪得到,也可以从机器人几何与命令 rollout 投影得到。它比逐像素光流稀疏,并且保留点级跨时间对应。

Q2:既然已经用 Isaac Lab 模拟动作,为何还需要世界模型?

Isaac Lab 在这里主要提供机器人本体和控制器运动,用于保证 action flow 可执行;Hydra-0 负责预测难以显式模拟的真实视觉后果,尤其是布料、绳索、袋子等可变形物体变化。

Q3:能否直接在图上画一条 object flow 控制机器人?

论文逆向模式支持目标物体 flow,但真实控制还需 action head,并且目前只在一个柔性管任务上定性验证。任意手绘 flow 可能不满足物理可实现性。

Q4:它真的跨 embodiment 吗?

训练数据确实覆盖人手、UMI、单臂、双臂和人形机器人,且多形态中训提升了未见 IWS 任务的 zero-shot 与少样本适配。不过前向部署到新机器人仍需其几何、控制器、仿真和标定。

Q5:90.4% 误差下降是否全部来自 Action Flow?

不是。该宣传数字比较最佳 A14B 4-step 模型与 Cosmos 2.5 原生动作基线,同时包含骨干差异。论文另有相同 Cosmos 2.5 骨干对照,才能更纯粹地证明接口本身的优势。

Q6:为什么蒸馏后的 4-step 模型质量反而最好?

表中 4-step 学生多数指标优于多步 teacher,可能来自因果训练与蒸馏的分布正则化效果。但论文并未把这一现象做完整机制拆解,不能普遍推断"步数越少质量越高"。

Q7:策略评估可以取代真实机器人测试吗?

不能。当前是开放环 replay,只验证成功率聚合相关性。闭环策略会受到生成观测误差反馈,风险远高于固定轨迹回放。

Q8:这项工作最重要的长期价值是什么?

让缺少统一动作标签的人类视频、UMI 数据和多机器人数据共享训练世界模型,并用一个可追溯到执行命令的视觉接口服务预测、评估和控制。

🧠 深度锐评:Hydra-0 强在哪里,又卡在哪里?

1. 它把"通用动作空间"从控制坐标迁移到了观察坐标

构建跨机器人模型时,人们常尝试寻找统一关节或末端动作。Hydra-0 换了一个问题:不同 embodiment 最终都要在相机画面中造成运动,为什么不直接统一可见后果?这是它最有分量的表示选择。

2. 规模化多形态数据比单一 benchmark 提升更重要

2,201.7 小时并非互联网级视频规模,但已经把人类、UMI、单臂、双臂和人形机器人真正放进同一个 action-conditioned 视频训练流程。IWS 的 0% 与少数据结果表明,这种混合不只是数据堆叠,而产生了可迁移的交互先验。

3. "可视统一"并没有消除底层 embodiment adapter

前向部署依赖 Isaac Lab、机器人几何与相机标定;逆向控制依赖目标机器人 action head。Hydra-0 将本体差异隔离到了接口两侧,而不是让差异消失。这其实是合理的工程分层,但"通用控制接口"不应被解读为 plug-and-play 零配置控制。

相关推荐
星云API技术支持1 小时前
企业微信二次开发:文件异步上传、媒体消息与回调处理的组合实践
机器人·企业微信
星云API技术支持2 小时前
企业微信二次开发完整方案:消息收发、外部群机器人、自动回复、Webhook 一套跑通
机器人·yapi·企业微信
yuewell_ai15 小时前
机器人的方向感从哪来-定位传感器一次讲透
机器人
风合星语17 小时前
2026 机器人工程实例(五):让 Allegro Hand 在 MuJoCo 中完成接触抓取——抓取状态机、稳定抬升与受控释放
c++·python·机器人·仿真
视觉小萌新20 小时前
机器人ROS2部署——部署Gazebo和moveit
机器人
菩提树下的凡夫1 天前
激光振镜双目结构光3D相机丢点问题
数码相机·3d
小饕1 天前
Jetson TensorRT vs RK3588 RKNN:两块板子都跑通了 LLM 和 VLM 之后,我悟了
人工智能·机器人·大模型端侧部署
上海蓝色星球1 天前
蓝色星球NG-AIOS新型AI工业操作系统重磅发布——以本体智能为内核,重构“AI+制造“新范式
大数据·数据库·人工智能·机器人