DreamX:视频扩散模型即通用潜在动作学习器 DreamX: Video Diffusion Model is a Universal Latent Action Learner
将预训练视频扩散模型改造为通用潜在动作学习器,无需配对数据即可实现人到机器人的动作迁移
前置知识
潜在动作模型(Latent Action Model, LAM)
从无动作标注的视频中学习动作的压缩表示。典型做法(LAPO、Genie、LAPA)训练一个编码器把相邻帧之间的视觉变化压缩为潜在动作 token,再用解码器仅凭当前帧和潜在动作重建未来帧,通过这种逆向-正向动力学瓶颈迫使编码器只保留动作相关信息。学到的表示可作为策略学习的预测目标,让无标注人类视频也能监督机器人控制。
DreamX 本质上就是一个潜在动作模型,其全部对比实验(LARYBench 上的 DiLA、UniVLA、LAPA、villa-X 等)都在与这一范式的方法比较,理解该范式的瓶颈(动作与外观纠缠、表示与训练本体绑定)才能理解 DreamX 的改进点。
视频扩散模型与时空注意力
视频扩散 Transformer(如本文使用的 Cosmos-Predict2.5)先用因果视频自编码器把视频压缩为若干潜在帧,再 patch 化为空间 token,通过时空联合注意力建模帧间关系。经大规模视频预训练后,其内部表征自发涌现出对应点匹配、光流、几何与语义结构,能刻画“哪个物体在哪里如何运动”,无需任务特定监督。
DreamX 的核心假设正是这些内部表征本身就是动作相关动态的天然来源:它直接从骨干网络内部“读取”潜在动作,而不是像传统 LAM 那样从像素从头训练一个编码器。
流匹配(Flow Matching)
一种扩散生成训练目标:把干净数据 $x$ 与噪声 $\epsilon$ 按噪声水平 $\tau$ 插值为 $x_\tau = (1-\tau)x + \tau\epsilon$,训练网络 $v_\theta$ 预测速度场,损失为 $\mathbb{E}\,\|v_\theta(x_\tau,\tau;c) - (\epsilon-x)\|_2^2$。相比标准 DDPM 扩散,其训练路径更直、采样更高效,是新一代视频与图像生成模型的主流目标函数。
DreamX 的视频重建损失、机器人动作解码损失以及下游策略训练全部采用流匹配目标,论文中的公式 (1)(4)(7) 均基于它,不熟悉该目标函数就无法理解其训练方式。
LoRA 低秩适配
在冻结预训练权重的同时,为每个目标层注入低秩增量 $W + BA$(秩 $r \ll \min(d,k)$),只训练 $A, B$ 两个小矩阵。它以极小的参数与显存开销微调大模型,并可为不同子模块配置相互独立的适配器,让同一骨干在不同角色下保持各自的修改。
DreamX 用秩 16 的独立 LoRA 分别适配编码器与解码器,策略生成器用秩 128 LoRA 微调,理解 LoRA 才能看懂其“一个预训练骨干、多种角色复用”的实现方式与训练开销。
跨本体迁移
不同机器人乃至人手拥有不同的关节结构、自由度与控制空间:同一个“拿起方块”行为在不同本体上对应完全不同的底层动作参数化方式。跨本体迁移研究如何把一种本体的演示运动用到另一种本体上,常用思路是学习一个本体无关的中间表示(语义动作或潜在动作),再由目标本体的执行器解码为可执行指令。
论文的核心主张是其潜在动作空间天然跨本体:人类视频中提取的潜在动作可以直接解码为双臂 YAM 机器人的 14 维关节指令,且全程不需要任何配对的人-机器人训练数据,这是理解其价值的关键。
VLA 模型与世界动作模型(WAM)
视觉-语言-动作模型(如 OpenVLA、$\pi$0.5)直接把视觉观测与语言指令映射为机器人动作;世界动作模型进一步以视频生成的方式联合建模环境演化与动作。两者的性能都高度依赖大规模机器人演示数据,而这类数据的采集成本远高于人类自我中心视频,这构成了数据瓶颈。
DreamX 的最终目标是让人类视频为这类策略提供可扩展的动作监督:其真实机器人策略实验(关抽屉任务)及与 Mimic-Video 的对比,正是在检验“视频先验驱动策略”这条路线的可行性与效率。
研究动机
通用机器人策略(VLA 模型与世界动作模型)的性能高度依赖多样化的机器人演示,但机器人数据采集昂贵。人类自我中心视频虽然蕴含海量操作行为,却存在一个根本错位:动作监督与产生它的本体绑定。人类视频没有机器人动作标签,不同机器人的控制空间也互不兼容——同样是“拿起方块”,在双臂 YAM 上是 14 维关节绝对指令,在其他本体上则是另一套参数化方式,动作标签无法直接共享。潜在动作模型(LAM)看似提供了出路:把帧间视觉变化压缩成潜在动作即可绕开动作标签,但已有方法有两个具体缺陷。其一,仅靠“重建未来帧”这一目标并不保证表示只编码动作——只要视角、光照、场景外观等信息有助于降低重建误差,编码器就会把它们与动作纠缠在一起(Nikulin 等 2025 年已证明存在干扰物时需要额外监督)。其二,主流 LAM(LAPA、UniVLA、villa-X 等)用独立训练的编码器从单帧对或极短片段学习动作,看不到运动如何跨多步延续,学到的表示与训练本体强绑定、难以迁移。量化对比可见一斑:在 LARYBench 回归基准上,帧对方法的平均归一化 MSE 高达 0.60–0.97,细粒度运动信息几乎丢失。
本文的目标是本文的目标是构建一个通用、紧凑、跨本体的潜在动作空间,作为人类视频与机器人数据之间共享的“动作接口”。具体拆成四个可检验的子目标:第一,表示要动作相关——把每个潜在视频帧的 1,200 个 token 压缩到 4 个潜在动作 token(300 倍压缩)后,仍能恢复末端执行器轨迹、区分语义不同的运动;第二,表示要与外观解耦——给定目标场景的初始帧时,同一潜在动作应能在新场景、新本体上复现源运动;第三,零样本跨本体——从人类视频提取的潜在动作可直接解码为目标机器人可执行的低层动作,全程不需要配对的人-机器人训练数据;第四,可作为下游策略的预测目标——策略在推理时无需生成未来视频即可利用该表示,从而规避视频生成式策略的高推理成本(对比方法 Mimic-Video 单次推理约 28 秒)。衡量标准是在 LARYBench 上达到潜在动作模型的最优水平,并在 HumanAndRobot 基准和真实机器人上验证迁移能力。
与已有工作不同的是,本文的独特切入点是:不是“用视频扩散模型生成像素、再外挂一个从头训练的动作编码器”,而是让潜在动作直接从预训练视频扩散骨干的内部时空表征中被读取出来。已有工作虽然也引入了视频生成先验,但角色分工完全不同——VPP 用微调后视频模型的内部预测表征辅助策略学习但仍依赖动作标签;AdaWorld 与 VideoWorld2 让视频模型充当生成先验提供像素或 rollout,而动作表示由外部模块在帧级转移上从零训练,动作编码器从不接触视频骨干的内部表征。DreamX 则把可学习查询 token 直接拼进骨干的时空注意力流,让查询在与视频 token 逐层交互中聚合动作相关信息;监督信号也不是像素级重建,而是利用骨干自身的生成先验——初始帧负责外观、生成先验负责合理演化,重建目标只逼查询编码“解释未来所必需的运动”。更关键的是,跨本体迁移并非显式训练目标,而是从这种架构中涌现的能力,这与所有显式设计解耦模块的方法(CoMo、DiLA)形成鲜明对照。
核心方法
直觉上,DreamX 把一个预训练视频扩散模型(Cosmos-Predict2.5 2B)“一分为二”地复用:它见过海量视频中“哪些物体在哪里如何运动”,其内部时空表征天然适合作为动作信息的来源;同时它具备生成先验,能在给定初始帧时合成合理的未来,因此让它以潜在动作为条件重建未来帧,就能反向约束“潜在动作里必须装什么”。技术路线分四步:第一步,引入可学习查询 token,与视频 token 一起通过骨干的时空注意力,查询从表征中“读出”紧凑潜在动作 $Q = E_\phi(q; V, \ell)$;第二步,用同一预训练骨干初始化的解码器 $D_\theta$ 以初始帧 $V_0$ 与潜在动作为条件(替代原本的文本条件)重建未来帧,有动作标签的机器人数据额外解码动作块作为辅助任务,整体用流匹配损失 $\mathcal{L} = \mathcal{L}_{video} + \lambda\mathcal{L}_{action}$ 端到端训练;第三步,冻结编码器后即可做两类迁移——视频迁移把源潜在动作接到目标初始帧上生成“目标外观+源运动”的视频,动作迁移把源潜在动作接到目标机器人初始本体状态上解码出可执行低层动作;第四步,训练变分瓶颈把每个 2,048 维潜在动作 token 压到 32 维紧凑码 $Z$,作为下游策略的预测目标,策略同时输出紧凑码与低层动作。
核心创新是“查询读出 + 生成式监督”的组合及其带来的外观-运动解耦机制,与已有方法的本质区别有三层。第一,动作信息的来源不同:LAPA、UniVLA、villa-X 等用独立的小编码器在帧对上从头学动作,而 DreamX 的查询直接参与预训练骨干的时空注意力,等于把网络在海量视频上习得的运动、对应与几何知识“蒸馏”进 4 个 token,每潜在帧 1,200 → 4 tokens 实现 300 倍压缩。第二,监督信号的施加方式不同:传统 LAM 用像素重建,外观信息只要有助于降低误差就会被吸收进动作表示;DreamX 让解码器在“初始帧给外观、生成先验给演化”的条件下重建未来,重建误差对查询的压力只剩“告诉我场景该如何演化”,从而把外观从潜在动作中挤出去。这一点由精心设计的注意力掩码从信息流上强制保证:编码器中查询对视频 token 的注意力是帧因果的(第 $g$ 组查询只见 $V_{0:g}$),查询之间自由注意;解码器中潜在动作只作条件、自身不注意任何 token,视频流与动作流都读取查询但互不注意。第三,跨本体能力是涌现的而非显式优化的:训练目标中没有任何配对人-机器人数据,共享潜在空间却能支撑零样本人类→机器人迁移。
方法步骤详情
训练与使用全流程如下。(1)预处理:视频经 Cosmos 的因果视频自编码器压缩为 $N$ 个潜在帧 $V=(V_0,\dots,V_{N-1})$,patch 化后每个潜在帧约对应 1,200 个 token;训练片段为 5–29 帧(对应 2–8 个潜在帧),分辨率 480$\times$640。(2)编码:为每个潜在帧分配 $K=4$ 个可学习查询 token,与视频 token 拼接后通过骨干,语言指令 $\ell$ 经骨干原有的交叉注意力注入;第 $g$ 组查询只 attend $V_{0:g}$、查询间无限制,一次前向即得全部组 $Q = E_\phi(q; V, \ell)$。(3)解码:以同一预训练骨干初始化的解码器 $D_\theta$ 接收 $Q_{0:N-1}$(沿时间维与视频 token 拼接)、$V_0$ 与本体状态 $p_0$,输出 $\hat V, \hat A = D_\theta(Q_{0:N-1}, V_0, p_0)$;掩码保证视频帧 $V_t$ 只读查询前缀 $Q_{0:t}$、动作块 $A_m$ 只读 $Q_{0:m}$。(4)优化:$\mathcal{L}=\mathcal{L}_{video}+\lambda\mathcal{L}_{action}$(无动作标签的数据 $\lambda=0$),编码器与解码器各用独立的秩 16 LoRA、骨干权重冻结,共训练 78,000 步,LoRA 与查询学习率 $10^{-4}$、其余可训练参数 $5\times10^{-5}$。(5)迁移使用:给定源视频,用冻结编码器提取 $Q^{src}=E_\phi(q; V^{src}, \ell^{src})$;视频迁移为 $\hat V^{tgt}=D_\theta(Q^{src}, V_0^{tgt})$,动作迁移为 $\hat A^{tgt}=D_\theta(Q^{src}, p_0^{tgt})$,后者开环执行。(6)策略学习:先在冻结表示上训练变分瓶颈(重参数化采样 $Z=\mu_\psi+\sigma_\psi\odot\epsilon$,KL 权重 $\beta=0.01$,5 万步),把每个 2,048 维 token 压成 32 维码 $Z=\mu_\psi(Q)$;再微调骨干(秩 128 LoRA,10 万步)从当前帧与指令预测 $Z$,独立 DiT 动作头以潜在动作特征为条件预测低层动作,损失 $\mathcal{L}_{policy}=\mathcal{L}_{latent}+\lambda\mathcal{L}_{action}$。
技术新颖性
技术新颖性可从四方面评估。其一,“骨干内读取”范式:这是让潜在动作直接从视频扩散 Transformer 内部时空表征中读出的框架,此前 VPP、AdaWorld、VideoWorld2 中视频模型只充当生成先验或像素供应商,动作编码器仍是外部模块;该设计让动作表示直接继承预训练时空知识,也免去为每个本体单独训练编码器。其二,生成式监督替代像素监督:把“初始帧给外观、潜在动作给演化”的职责划分显式编码进注意力掩码——查询在解码器中纯条件化、视频流与动作流互不接触——从信息流层面而非损失权重层面强制解耦,比 CoMo、DiLA 等显式解耦模块更简洁。其三,一套权重服务三种用途:同一骨干经 LoRA 适配后既是编码器(读出潜在动作)又是解码器(重建视频与动作),下游还能转生成为潜在动作生成器(策略),最大化预训练投资。其四,紧凑码作为策略预测目标:每个潜在动作 token 保持骨干全宽 2,048 维会使扩散动作头维度爆炸,作者用 FAE 式变分瓶颈压到 32 维并保持 token 布局,使策略推理无需生成视频(较 Mimic-Video 快 19.1 倍)。消融显示仅 RGB 监督即可达到接近完整监督的表示质量(回归 MSE 0.24 对 0.23),进一步印证解耦来自架构本身而非动作标签。
实验结果
实验从表示质量、视频迁移、动作迁移、策略学习四个层面展开。(1)表示质量(Table 1,LARYBench):DreamX 在全部子基准上第一——动作回归平均归一化 MSE 0.17,较最强基线 DiLA 的 0.38 降低 55%;动作分类平均准确率 69.67%,较 DiLA 的 62.25% 提升 7.4 个百分点。分数据集看,回归 MSE 为 CALVIN 0.15、VLABench 0.05、RoboCOIN 0.26、AgiBotWorld-Beta 0.22;分类准确率为原子机器人 68.03、复合人类 72.72、复合机器人 68.26。压缩到 32 通道的 FAE-32 码(回归 0.22、分类 67.15%)仍全面超过所有基线。(2)零样本人→机器人视频迁移(Table 2,HumanAndRobot 的 1,008 段、21 个任务):无任何配对训练数据,DreamX 取得 FVD 57.44、LPIPS 0.179、PSNR 26.52、SSIM 0.917,四项全面优于 AdaWorld(170.19/0.344/22.95/0.846)与 VideoWorld2(241.92/0.299/22.53/0.881);定性上基线出现视觉退化或“人手泄漏”,DreamX 保持目标机器人外观。(3)动作迁移(Figure 6):人类视频的潜在动作开环解码为双臂 YAM 的 14 维关节指令,可复现字母 N 的笔画结构、单臂推“先前后再左转”的方向与时序、以及双臂推的左右臂分工。(4)真实世界策略(Table 3,关抽屉):DreamX 平均子目标成功率 96.7%(上/下/双抽屉为 90/100/100),Mimic-Video 仅 50.0%;单次推理延迟 1,474 ms 对 28,226 ms,快 19.1 倍;仅用人类视频训练的配置在单抽屉任务达到 100% SR(Table 6)。(5)消融与分析:CALVIN 上潜在动作条件化把平均完成序列长度从 2.378 提到 2.643(Table 7);仅 RGB 监督与 RGB+动作监督质量相当(Table 4);对潜在动作加噪 $\alpha\in\{0.1,0.3,1.0\}$ 时解码运动单调改变而静态区域外观变化远小于全图,置零查询使平均运动比例降至 0.456(Table 8);2,362 段 RoboCOIN 片段的 t-SNE 显示同一运动方向的潜在动作质心在不同机器人本体间聚类一致(Figure 12)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LARYBench 动作回归(4 数据集平均) | 归一化动作 MSE(↓) | 0.17(CALVIN 0.15 / VLABench 0.05 / RoboCOIN 0.26 / AgiBotWorld-Beta 0.22) | DiLA 0.38(次优);帧对方法 LAPA-DINOv3 0.60、LAPA 0.97 | 较 DiLA 降低 55% |
| LARYBench 动作分类(3 类别平均) | 准确率(↑) | 69.67%(原子机器人 68.03 / 复合人类 72.72 / 复合机器人 68.26) | DiLA 62.25% | +7.42 个百分点 |
| LARYBench(32 通道紧凑码 FAE-32) | 回归 MSE(↓)/ 分类准确率(↑) | 0.22 / 67.15% | DiLA 0.38 / 62.25% | 300 倍压缩后再压至 32 通道仍全面超过所有基线 |
| 零样本人→机器人视频迁移(HumanAndRobot,1,008 段 21 任务) | FVD(↓) | 57.44 | AdaWorld 170.19;VideoWorld2 241.92 | 较 AdaWorld 降低约 66%,较 VideoWorld2 降低约 76% |
| 零样本人→机器人视频迁移(同上) | LPIPS(↓)/ PSNR(↑)/ SSIM(↑) | 0.179 / 26.52 / 0.917 | AdaWorld 0.344 / 22.95 / 0.846;VideoWorld2 0.299 / 22.53 / 0.881 | 三项感知/像素指标全面领先 |
| 真实世界关抽屉策略(双臂 YAM,上/下/双抽屉 3 配置) | 平均子目标成功率(↑) | 96.7%(90 / 100 / 100) | Mimic-Video 50.0%(三种配置均 50.0) | +46.7 个百分点 |
| 真实世界策略推理 | 单次调用延迟(↓) | 1,474 ms | Mimic-Video 28,226 ms | 快 19.1 倍 |
| CALVIN 策略消融(有无潜在动作条件化) | 平均完成序列长度(↑) | 2.643(≥5 连续任务成功率 25.7%) | 无潜在动作条件化 2.378(≥5 为 20.4%) | +0.265;各序列长度成功率提升 2.3–5.3 个百分点 |
局限与改进
作者在伦理与结论部分承认:跨本体迁移与机器人策略在评估设定之外可能失败,部署前需要安全验证;视频迁移输出为合成内容需明确标识。从正文看,局限还包括:动作迁移是开环执行、只提供初始本体状态、无视觉反馈纠错,任何误差都会沿序列累积;真实策略评估只覆盖单一关抽屉任务、45 条训练轨迹、每配置 10 次试验,样本量小且任务简单,96.7% 的成功率外推性有限;辅助动作监督收益边际(回归 MSE 0.24→0.23,分类反而 65.3%→64.8%),说明当前架构对本体状态信息的利用很弱。我自己的观察还有五点:其一,推理延迟 1.47 秒/次虽比视频生成快 19.1 倍,但对 10–30 Hz 的高频闭环控制仍太慢,策略只能以低频动作块方式执行;其二,所有定量结果依赖 Cosmos-Predict2.5 的生成能力上限,其预训练偏见会直接传导到潜在动作空间;其三,语言指令 $\ell$ 经交叉注意力注入,但论文未消融指令对潜在动作的影响,“语言如何调制动作空间”仍是黑箱;其四,LARYBench 采用冻结表示加公共探针的协议,公平但只能测表示的线性可分性,不能完全代表下游控制性能;其五,迁移质量依赖目标初始帧准确给出外观,若目标与源在视角、遮挡上差异过大,外观-运动共享假设可能失效。
独立分析的弱点
以下弱点均附改进方向。(1)开环动作迁移无纠错机制:潜在动作一次性解码整段动作后开环执行,长序列中模型误差与机器人动力学偏差会持续累积;改进方向是滑动窗口重提取——每执行 $k$ 步就用机器人当前观测重新编码剩余运动,形成视觉伺服式闭环。(2)策略评估单薄:真实世界只有关抽屉一个任务、45 条轨迹,无法说明方法在接触丰富、高灵巧任务上的表现;应在多任务真机套件上系统评估并报告长时程成功率。(3)潜在动作长度固定:每潜在帧固定 4 个 token,静止片段浪费容量、高速运动可能欠表达;可引入按运动幅度自适应的 token 数或门控机制。(4)压缩瓶颈事后训练:FAE 瓶颈在冻结表示上单独训练 5 万步,压缩中丢失的信息无法反馈回编码器;端到端联合训练或在骨干内直接做低维投影可能更优。(5)语言通道未验证:指令对 $Q$ 的影响没有消融,未来应量化 $\ell$ 扰动对潜在动作的改变,并测试无指令视频下的鲁棒性。(6)评估偏桌面操纵:LARYBench 与抽屉任务均为桌面场景,对导航、全身移动等领域的适用性未知,论文的 DAVIS 通用视频示例也只是定性展示。(7)可解释性有限:注意力指向手/夹爪、方向质心可分只是间接证据,4 个 token 各自承载什么语义尚不清楚,可加线性探针或与语言嵌入对齐来解释每个通道。
未来方向
作者方向上强调把预训练视频模型作为可复用动作表示的基础,具体可延伸:其一,规模化——用更大的视频骨干(如 14B 级 Cosmos)与更多样的数据(含第三人称、非操纵视频)预训练,检验表示质量随规模与数据构成的缩放规律;其二,与世界模型规划结合——DreamX 解码器本身可在“目标初始帧+潜在动作”下想象未来,可将其作为潜空间模拟器做 MPC 或树搜索式规划,实现零样本试错;其三,闭环迁移——把开环动作迁移升级为滚动时域控制,用视觉重编码校正累积误差;其四,语言接地——指令经交叉注意力注入,可研究用文本对潜在动作空间做分区检索,实现“以语言查询视频中的动作片段”;其五,作为 VLA 预训练目标——把紧凑码 $Z$ 作为大规模人类视频上的预训练目标、再用少量机器人数据微调动作头,可能优于现有 VLA 的隐式预训练;其六,扩展本体与模态——从双臂操纵延伸到人形全身、四足等本体,并纳入触觉或音频条件;其七,理论刻画——为什么生成式监督能把外观信息从动作表示中挤出,可从信息瓶颈视角给出定量分析。
复现评估
复现条件总体透明但算力门槛较高。有利因素:骨干 Cosmos-Predict2.5 2B 开源;全部训练与评测数据集公开(人类:EgoDex 350.5 小时、EgoVerse 59.0 小时;机器人:AgiBot 818.2 小时、MolmoAct2 502.3 小时、DROID 410.1 小时、Bridge 47.9 小时,合计 2.19k 小时);评测基准 LARYBench、HumanAndRobot、CALVIN、RoboCOIN 均公开,且附录给出探针结构(768 维投影+注意力池化)与特征提取协议;超参文档完备(LoRA 秩、学习率、步数、$\beta=0.01$、分辨率 480$\times$640、片段 5–29 帧等),有可复现性声明与补充视频;项目页 cvlab-kaist.github.io/DreamX 汇总资源。不利因素:预训练 78,000 步 + 瓶颈 50,000 步 + 策略 100,000 步(2 块 H200、全局批次 60),意味着需要 H100/H200 级多卡集群与 2k 小时级视频的存储预处理流水线;真机部分需要双臂 YAM 机器人与 RealSense D405 的特定顶视配置,普通实验室难以复刻,但可按文中协议在其他平台复现视频迁移部分;论文未明确承诺代码开源,仅凭论文细节复现的工程难度为中高。建议路径:先在论文消融所用子集(MolmoAct2+EgoVerse+EgoDex)上小规模复现表示学习并用 LARYBench 探针验证,再决定是否全量训练。
论文图表
概览图:人类自我中心视频与机器人视频经过同一个视频扩散模型编码到紧凑且通用的潜在动作空间;该空间同时支撑三种用途——把源视频运动迁移到目标外观的视频迁移、把人类运动解码为可执行机器人动作的动作迁移、以及联合预测潜在动作与低层动作的轻量策略(DreamX Policy)。
一图总览论文的问题设定、方法骨架与三大应用,是快速建立全局图景的入口。