DreamX:将视频扩散模型改造成通用潜在动作学习器
DreamX: Video Diffusion Model is a Universal Latent Action Learner
一篇有意思的论文:把视频扩散模型拿来学潜在动作,每帧压到 4 个 token,人类和机器人的动作能互相迁移,在 LARY-Bench 上拿了最优。
论文提出 DreamX,把预训练视频扩散模型用作通用潜在动作学习器。它从视频模型的时空特征中提取紧凑的潜在动作,将每个视频帧压缩 300 倍(1200 个 token 压到 4 个)。借助视频模型的生成先验,潜在动作与视觉外观解耦,可作为人类与机器人等不同本体之间共享的动作接口。在 LARY-Bench 上 DreamX 取得潜在动作模型中的最优成绩,并支持任意到任意的视频迁移和跨本体的零样本动作迁移。该共享接口还能在紧凑潜在动作空间中训练策略,降低基于视频扩散的策略的计算成本。
DreamX: Video Diffusion Model is a Universal Latent Action Learner
Learning a universal action representation from video requires modeling dynamics that are consistent across different executions of the same action yet invariant to changes in embodiment and appearance. Existing latent action models infer actions from visual transitions, but their representations often entangle the underlying action with irrelevant visual variations, limiting transfer across embodiments. We introduce DreamX, which repurposes a pretrained video diffusion model as a universal latent action learner. Our approach extracts compact latent actions from the video model's rich spatiotemporal features, compressing each latent video frame by $\boldsymbol{300\times}$ ($\boldsymbol{1{,}200 \rightarrow 4}$ tokens). In addition, by leveraging the video model's generative prior, these latent actions capture action-relevant dynamics disentangled from visual appearance. The resulting latent space serves as a shared action interface across embodiments (human and robot), environments, and action parameterizations. On LARY-Bench, DreamX achieves state-of-the-art performance among latent action models. Moreover, actions encoded in one setting remain effective in another: the latent space supports any-to-any video transfer and zero-shot action transfer across diverse embodiments and environments. Furthermore, this shared interface also enables policy learning in a compact latent action space, reducing the computational cost of video-diffusion-based policies.