Organizations: Shanghai Jiao Tong University · Eastern Institute of Technology, Ningbo · Fudan University · Shanghai Innovation Institute · Tsinghua University
Vision-language-action (VLA) models have shown great potential in building generalist robots, but still face a dilemma-misalignment of 2D image forecasting and 3D action prediction. Besides, such a vision-action entangled training manner limits model learning from large-scale, action-free web video data. To address these issues, we propose DeFI, a novel framework that Decouples visual Forward and Inverse dynamics pretraining to exploit respective data sources, wherein video generation and action prediction are disentangled. We introduce the General Forward Dynamics Model (GFDM), pretrained on diverse human and robot videos for future prediction, and the General Inverse Dynamics Model (GIDM), trained via self-supervised learning to infer latent actions from unlabeled video transitions. These models are then integrated into a unified architecture for end-to-end finetuning on downstream tasks. In this manner, GFDM and GIDM first shine separately and then cooperate for mutual benefit. Extensive experiments on CALVIN ABC-D and SimplerEnv demonstrate state-of-the-art performance, with DeFI achieving an average task length of 4.51 for CALVIN, 51.2% success rate on SimplerEnv-Fractal benchmark and 81.3% success rate in real-world deployment, significantly outperforming prior methods.
Figures & tables
Figure 1: We disentangle robot learning via two decoupled components: a visual forward dynamics model pretrained on large-scale mixed videos via video generation, and an inverse dynamics model pretrained on mixed videos through self-supervised action representation. Then they are coupled during fine-tuning in an end-to-end manner to adapt to downstream tasks. This decoupled pretraining paradigm unleashes the potential of massive action-free videos for policy learning, while retaining robot-specific action grounding, leading to improved success rates across diverse benchmarks.
Figure 2: Overall framework of DeFI. Stage I (Decoupled pretraining): (a) A general visual forward dynamics model is pretrained with human and robot videos via a video generation objective, predicting future frames from current observations and instructions. (b) In parallel, a general inverse dynamics model is pretrained in a self-supervised manner to map pairs of observations (ot,ot+n) into latent actions, capturing inverse dynamics knowledge without explicit action labels. Stage II (Coupled finetuning): The forward and inverse models are coupled, and a diffusion-based adapter is used to generate executable robot action sequences. This two-stage framework unleashes the rich priors of human videos while grounding them in robot data for scalable policy learning.
Figure 3: Experiments setup on CALVIN ABC-D, SimplerEnv Google Robot and real-world Franka Robot. We evaluate DeFI across 3 simulation environments.
View
Method
1
2
3
4
5
Avg. Len. ↑
Third View
SuSIE ( Black et al., 2023 )
87.0
69.0
49.0
38.0
26.0
2.69
CLOVER ( Bu et al., 2024b )
96.0
83.5
70.8
57.5
45.4
3.53
UniVLA ( Bu et al., 2025b )
95.5
85.8
75.4
66.9
56.5
3.80
DeFI
92.9
87.2
81.2
75.0
68.4
4.05
Multi-View
GR-1 ( Wu et al., 2024 )
85.4
71.2
59.6
49.7
40.1
3.06
OpenVLA ( Kim et al., 2024 )
91.3
77.8
62.0
52.1
43.5
3.27
Table 1: CALVIN ABC-D results. We present the average success computed over 1000 rollouts for each task and the average number of completed tasks to solve 5 instructions consecutively (Avg. Len.). DeFI shows significant superiority over baselines. The best results are bolded . *We reproduced results of π0.5 , GR00T N1 and OpenVLA-OFT on CALVIN.
Figure 5
SimplerEnv on Google Robot Tasks
Model
Visual Matching
Variant Aggregation
Pick Coke Can
Move Near
Open/Close Drawer
Avg.
Pick Coke Can
Move Near
Open/Close Drawer
Avg.
Octo-Base ( Team et al., 2024 )
17.0%
4.2%
22.7%
16.8%
0.6%
3.1%
1.1%
1.1%
TraceVLA ( Zheng et al., 2024 )
28.0%
53.7%
57.0%
42.0%
60.0%
56.4%
31.0%
45.0%
OpenVLA ( Kim et al., 2024 )
16.3%
46.2%
35.6%
27.7%
54.5%
47.7%
17.7%
39.8%
DeFI
54.2 %
60.7%
38.6%
51.2%
53.9%
58.2%
24.0%
45.4%
Table 2: Evaluation results across different policies on SimplerEnv . We evaluate DeFI on 3 tasks on the Google Robot in SimplerEnv.
Method
Success Rate (%)
Place
Open
Close
Cut
Stack Bowl
Stack Cube
Stack Bottle
Pour Water
Average
Diffusion Policy ( Chi et al., 2023 )
70.0
40.0
70.0
50.0
45.0
35.0
40.0
35.0
48.2
Octo-Base ( Team et al., 2024 )
55.0
35.0
60.0
20.0
30.0
25.0
30.0
20.0
34.4
OpenVLA ( Kim et al., 2024 )
50.0
40.0
65.0
40.0
30.0
35.0
45.0
45.0
43.8
DeFI
90.0
75.0
100.0
80.0
80.0
70.0
80.0
75.0
81.3
Table 3: Real-world evaluation with the Franka Robot across eight tasks.
Figure 6: Real-world robot setup.
Table 9Table 10Table 11
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
Train parameter
Value
GPU
NVIDIA H100
Number of GPUs
8
Pretraining time of GFDM
3 days
Pretraining time of GIDM
1.5 days
Finetuning time on CALVIN
0.5 days
Training memory on CALVIN
64G
Appendix
Table 10: Training and model parameters used in our DeFI.
Category
Training dataset mixture
Proportion
Robot Videos
Fractal ( Brohan et al., 2023 )
30%
Bridge ( Ebert et al., 2021 ; Walke et al., 2023 )
10%
CALVIN-ABC ( Mees et al., 2022b )
30%
Human Videos
Something-Something-v2 Goyal et al. (2017)
15%
Ego4D Grauman et al. (2022)
15%
Appendix
Table 11: The general forward dynamics model of our DeFI is trained on a mixture of data from the Open X-Embodiment ( O’Neill et al., 2023 ) and CALVIN ( Mees et al., 2022b ) robot video datasets, as well as the Ego4D ( Grauman et al., 2022 ) and Something-Something-v2 ( Goyal et al., 2017 ) human video datasets. The proportions are normalized to sum to 100%.
Category
Training dataset mixture
Proportion
Robot Video
Fractal ( Brohan et al., 2023 )
16.3%
Kuka ( Kalashnikov et al., 2018 )
7.4%
Bridge ( Ebert et al., 2021 ; Walke et al., 2023 )
8.0%
Taco Play ( Mees et al., 2022a )
4.1%
Jaco Play ( Dass et al., 2023 )
0.7%
Berkeley Cable Routing ( Luo et al., 2024 )
0.4%
Appendix
Table 12: The general inverse dynamics model of our DeFI is trained on a mixture of data from the Open X-Embodiment ( O’Neill et al., 2023 ) robot video dataset and the Ego4D ( Grauman et al., 2022 ) human video dataset. The proportions are normalized to sum to 100%.
Figure 15
Model part
Inference time
General Forward Dynamics Model
86.1ms
General Inverse Dynamics Model
42.9ms
Action Adapter
24.3ms
Appendix
Table 13: The inference time of our DeFI is measured on an NVIDIA GeForce RTX 4090 GPU, averaged over five runs.
Figure 9: Qualitative results of different denosing step for the forward dynamics model.
Figure 10: Qualitative results of failure cases.
Figure 11: Qualitative results of failure cases.
Figure 12: Qualitative attention heatmap results of the general inverse dynamics model on robot and human videos.
Figure 13: Qualitative results of the CALVIN long-horizon task.
Figure 14: Qualitative results of the CALVIN long-horizon task.
Figure 15: Qualitative results of SimplerEnv evaluation on Google Robot.
Figure 16: Qualitative results of SimplerEnv evaluation on Google Robot.
Figure 17: Qualitative results of real-world experiments.
Vision-language-action (VLA) models and world action models (WAMs) have advanced robotic policy learning by building on pretrained vision-language models (VLMs) and video generative models (VGMs), respectively. Yet VLAs learn without explicit supervision from future world states, while WAMs that condition actions on predicted futures incur extra inference cost and let visual prediction errors propagate into actions. We present Forward Dynamics Models (FDM), a family of robotic action models that couples action and future world-state prediction through an action-to-future paradigm. Future prediction is conditioned on action features, never the reverse: end-to-end training lets ground-truth future states supervise the policy through these features, while prediction errors cannot propagate into actions. At inference, FDM skips future generation entirely, avoiding its computational cost. The paradigm is backbone-agnostic, and we instantiate it with both VLM and VGM backbones. Experiments in simulation and on real robots demonstrate the effectiveness of FDM across both backbone families, achieving a state-of-the-art 99.4% average success rate on LIBERO and absolute gains of up to 13% over VLA and WAM methods on real-world tasks.
Zipei Ma, Junzhe Jiang, Jiahui Zhang +6
School of Data Science, Fudan University · Shanghai Innovation Institute · Imperial College London +1
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/.
The development of Vision-Language-Action (VLA) models has been significantly accelerated by pre-trained Vision-Language Models (VLMs). However, most existing end-to-end VLAs treat the VLM primarily as a multimodal encoder, directly mapping vision-language features to low-level actions. This paradigm underutilizes the VLM's potential in high-level decision making and introduces training instability, frequently degrading its rich semantic representations. To address these limitations, we introduce DIAL, a framework bridging high-level decision making and low-level motor execution through a differentiable latent intent bottleneck. Specifically, a VLM-based System-2 performs latent world modeling by synthesizing latent visual foresight within the VLM's native feature space; this foresight explicitly encodes intent and serves as the structural bottleneck. A lightweight System-1 policy then decodes this predicted intent together with the current observation into precise robot actions via latent inverse dynamics. To ensure optimization stability, we employ a two-stage training paradigm: a decoupled warmup phase where System-2 learns to predict latent futures while System-1 learns motor control under ground-truth future guidance within a unified feature space, followed by seamless end-to-end joint optimization. This enables action-aware gradients to refine the VLM backbone in a controlled manner, preserving pre-trained knowledge. Extensive experiments on the RoboCasa GR1 Tabletop benchmark show that DIAL establishes a new state-of-the-art, achieving superior performance with 10x fewer demonstrations than prior methods. Furthermore, by leveraging heterogeneous human demonstrations, DIAL learns physically grounded manipulation priors and exhibits robust zero-shot generalization to unseen objects and novel configurations during real-world deployment on a humanoid robot.
Yi Chen, Yuying Ge, Hui Zhou +3
The University of Hong Kong · XPENG Robotics · University of North Carolina at Chapel Hill