arXiv:2603.10448cs.RO2026-03被引 44

用视频生成模型提升机器人动作预测的泛化与效率

DiT4DiT: Jointly Modeling Video Dynamics and Actions for Generalizable Robot Control

  • 将视频与动作扩散模型级联,利用视频生成过程中的中间特征指导动作预测
  • 在LIBERO和RoboCasa上分别达98.6%和50.8%成功率,训练数据减少10倍以上
  • 适合追求高效、强泛化能力的机器人控制研究者

视觉-语言-动作(VLA)模型在机器人学习中展现出前景,但其表征仍主要源自静态图像-文本预训练,物理动态需依赖有限的动作数据学习。生成式视频模型则蕴含丰富的时空结构与隐含物理信息,是机器人操作的有力基础,但尚未被充分探索。为此,我们提出DiT4DiT,一种端到端的视频-动作模型,将视频扩散变换器与动作扩散变换器统一于级联框架中。不同于依赖重建未来帧,DiT4DiT从视频生成过程提取中间去噪特征,并作为动作预测的时序条件。我们进一步提出双流匹配目标,解耦时间步与噪声尺度,实现视频预测、隐藏状态提取与动作推断的协同训练。在仿真与真实世界基准测试中,DiT4DiT表现优异,达到98.6%(LIBERO)与50.8%(RoboCasa GR1)平均成功率,且训练数据显著减少。在Unitree G1机器人上亦展现优越真实性能与强零样本泛化能力。重要的是,该模型将样本效率提升超10倍,收敛速度加快达7倍,表明视频生成可作为机器人策略学习的有效扩展代理。代码与模型已开源:https://dit4dit.github.io/

原文摘要 · Abstract (English)

Vision-Language-Action (VLA) models have emerged as a promising paradigm for robot learning, but their representations are still largely inherited from static image-text pretraining, leaving physical dynamics to be learned from comparatively limited action data. Generative video models, by contrast, encode rich spatiotemporal structure and implicit physics, making them a compelling foundation for robotic manipulation. But their potentials are not fully explored in the literature. To bridge the gap, we introduce DiT4DiT, an end-to-end Video-Action Model that couples a video Diffusion Transformer with an action Diffusion Transformer in a unified cascaded framework. Instead of relying on reconstructed future frames, DiT4DiT extracts intermediate denoising features from the video generation process and uses them as temporally grounded conditions for action prediction. We further propose a dual flow-matching objective with decoupled timesteps and noise scales for video prediction, hidden-state extraction, and action inference, enabling coherent joint training of both modules. Across simulation and real-world benchmarks, DiT4DiT achieves state-of-the-art results, reaching average success rates of 98.6% on LIBERO and 50.8% on RoboCasa GR1 while using substantially less training data. On the Unitree G1 robot, it also delivers superior real-world performance and strong zero-shot generalization. Importantly, DiT4DiT improves sample efficiency by over 10x and speeds up convergence by up to 7x, demonstrating that video generation can serve as an effective scaling proxy for robot policy learning. We release code and models at https://dit4dit.github.io/.

机器人控制扩散模型视频生成动作预测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。