Disentangled Robot Learning via Separate Forward and Inverse Dynamics Pretraining
- 01The Bottleneck Was Never Just the World Model
- 02Decoupling Forward and Inverse Dynamics Unlocks Internet-Scale Video for Policy Learning
- 03Human Videos Are a Measurable, Scalable Performance Lever
- 04State-of-the-Art Results Across All Major Benchmarks
- 05Data Efficiency: 60% of the Data to Beat the Previous SOTA
Why should someone building or funding robots care? This paper offers a concrete architectural answer to one of the hardest unsolved problems in robot learning: how do you train capable robot policies when labeled robot demonstrations are scarce and expensive, but the internet has essentially unlimited unlabeled video showing humans doing useful things? DeFI's answer — train the "what will happen" model and the "what action caused that" model separately, then fuse them — is both principled and practically validated, beating every major VLA benchmark in head-to-head comparison.
1. Key Themes
The Bottleneck Was Never Just the World Model — It Was the Action Inference Model
Every prior approach that used video generation as a policy (SuSIE, VPP, Vidar) invested heavily in predicting what the future looks like, then bolted on a lightweight action decoder. DeFI's central argument is that this imbalance is the real performance ceiling. "Accurate action inference is as important as accurate future prediction, which still needs sufficient data for pretraining to unleash its full ability." (Section 1) The ablation confirms this directly: removing inverse dynamics pretraining while keeping forward dynamics pretraining drops average task length from 4.51 to 3.28 — a 27% collapse in performance. (Table 4, Section 4.5)
Decoupling Forward and Inverse Dynamics Unlocks Internet-Scale Video for Policy Learning
The fundamental innovation is architectural: by separating the forward dynamics model (predicts future frames from current observation + language) from the inverse dynamics model (infers latent actions from frame pairs), each module can be pretrained on the data it's actually suited for — without requiring action labels on any of it. "This decoupled pretraining paradigm unleashes the potential of massive action-free videos for policy learning, while retaining robot-specific action grounding." (Figure 1 caption) The GIDM is trained entirely self-supervised: given two frames, predict the latent action token that would connect them, with future frame reconstruction as the proxy loss.
Human Videos Are a Measurable, Scalable Performance Lever
The paper doesn't just assert human video helps — it quantifies the scaling curve. Removing human videos from both models drops performance from 4.51 to 3.92 average task length. Adding human video incrementally from 0% to 100% shows monotonic improvement with no saturation observed. "No saturation is observed, suggesting the potential for further improvements with even larger datasets." (Appendix A.4.2, Figure 8) The training mix includes Something-Something-v2 and Ego4D — both publicly available, zero cost to license for research.
State-of-the-Art Results Across All Major Benchmarks
DeFI achieves an average task length of 4.51 on CALVIN ABC-D (the hardest generalization split, training on ABC, testing on unseen environment D), beating the prior SOTA VPP at 4.33 (+4.2%), Physical Intelligence's π0 at 3.84, and GR00T N1 at 4.01. (Table 1, Section 4.2) On SimplerEnv-Fractal (Google Robot), DeFI hits 51.2% average success vs. TraceVLA's 42.0%. (Table 2, Section 4.3) In real-world Franka Panda deployment across 8 tasks, DeFI achieves 81.3% average success vs. Diffusion Policy's 48.2% and OpenVLA's 43.8%. (Table 3, Section 4.4)
Data Efficiency: 60% of the Data to Beat the Previous SOTA
This is the deployment-critical finding. "Our method requires only about 60% of the data on CALVIN ABC-D to surpass the previous state-of-the-art baseline." With only 10% of available training data, DeFI still achieves an 18% relative improvement over VPP in average task length. (Figure 5, Section 4.2) For teams deploying in new environments or new task categories, this means significantly fewer costly human-in-the-loop demonstrations to reach competent performance.
2. Contrarian Perspectives
Freezing the Foundation Model During Fine-Tuning Is the Right Move — Joint Training Hurts
Conventional wisdom in deep learning says end-to-end joint training almost always beats frozen pretrained modules. DeFI argues the opposite for the forward dynamics model. The paper demonstrates that fine-tuning everything simultaneously underperforms freezing GFDM and only fine-tuning GIDM + adapter: "Although 'All Train' updates FDM, IDM, and the action adapter jointly, its performance is lower than IDM+Adapter(Ours) because joint optimization introduces representation instability and gradient interference. When GFDM is finetuned, its latent outputs change throughout training, causing the input distribution of IDM to drift." (Section 4.5, Q6) In Table 9, "All Train" achieves 4.40 vs. 4.51 for GIDM+Adapter. The practical implication: for teams building on top of large video generation models (Stable Video Diffusion, Sora-class models), the pretraining is valuable precisely because it should be left alone during robot-specific fine-tuning.
Latent Action Extraction from Human Videos Is Not Enough — You Need a Pretrained Inverse Dynamics Model Too
Several companies and research groups (including the team behind LAPA, which directly inspired some of DeFI's GIDM design) have pursued a strategy of extracting latent action tokens from human videos, then pretraining a large VLA on those pseudo-labels. DeFI directly challenges the sufficiency of this approach: "This route is indirect—the latents must be consumed by sizable policy models that are costly to pretrain and fine-tune, and the learned codes are not guaranteed to align with the action manifold needed for execution across embodiments." (Section 2.2) The comparison against UniVLA (which uses latent action labels to pretrain VLA) is concrete — DeFI's multi-view result (4.51) beats UniVLA's third-view result (3.80), despite UniVLA being trained on larger-scale VLA pretraining. (Table 1)
Single-Step Video Denoising Is Sufficient for Control — Pixel Fidelity Is Wasted Computation
There is an implicit assumption in video-as-policy systems that better visual prediction quality leads to better control. DeFI finds this is false in practice: "A single denoising step already captures sufficient semantic information about future frames, and further steps do not yield improvements in manipulation performance." (Section 4.5, Q3) Five denoising steps takes ~250ms; one step takes ~150ms. (Table 13 and Q3) Critically, the paper argues that "the key signal for control lies in motion dynamics rather than appearance." This has major implications for inference latency budgets in deployed systems — you don't need to run a full diffusion rollout.
3. Companies Identified
Physical Intelligence (π) Robotics AI company building general-purpose robot policies. DeFI benchmarks directly against π0 and π0.5 on CALVIN. DeFI (4.51 avg. len.) outperforms π0 (3.84) and π0.5 (3.97). "We compare our model with the latest state-of-the-art generalist manipulation policies, including π0 (Black et al., 2024)." (Table 1, Section 4.2)
NVIDIA (GR00T N1) Developer of the GR00T N1 open foundation model for humanoid robots. DeFI outperforms GR00T N1 (4.01 avg. len. vs. 4.51). The paper notes: "*We reproduced results of GR00T N1 and OpenVLA-OFT on CALVIN." (Table 1 footnote) This matters because GR00T N1 is positioned as a broad-ecosystem foundation model; DeFI's performance gap suggests the architecture choice matters more than ecosystem scale alone.
Stability AI / Stability-Adjacent (Stable Video Diffusion) Developer of SVD, the video generation backbone used for GFDM. DeFI builds directly on SVD: "We adopt the open-sourced Stable Video Diffusion (SVD) (Blattmann et al., 2023) as the general forward dynamics model." (Appendix A.3) This is a direct commercial technology transfer — SVD's pretraining is repurposed for robot forward dynamics modeling.
Google DeepMind (RT Series, Fractal Dataset) Developer of RT-1, RT-2, and the Fractal demonstration dataset used widely in robot learning. DeFI uses the Fractal dataset as 30% of GFDM training data and 16.3% of GIDM training data. (Tables 11, 12) DeFI also benchmarks on SimplerEnv-Fractal (Google Robot), achieving 51.2% vs. the baseline OpenVLA at 27.7%. The Fractal dataset's open availability is a key enabler of this work.
Hugging Face / Open X-Embodiment Consortium Curators of the Open X-Embodiment dataset used extensively in DeFI pretraining. DeFI uses OXE as the primary robot video source for both GFDM and GIDM pretraining. "We first train the general forward dynamics model on a diverse collection of datasets...including Open X-Embodiment (O'Neill et al., 2023)." (Section 4.1) This underscores OXE's role as infrastructure for the entire field.
Octo Model Team Developers of the Octo generalist robot policy. Used as a baseline across CALVIN, SimplerEnv, and real-world experiments. DeFI significantly outperforms Octo-Base (34.4% vs. 81.3% on real-world Franka tasks). (Table 3)
4. People Identified
Wenyao Zhang Shanghai Jiao Tong University / Eastern Institute of Technology, Ningbo Lead author. Also co-authored DreamVLA (Zhang et al., 2025c), suggesting a sustained research program on video-grounded robot policies. The architecture choices in DeFI (separate pretraining, VQ-VAE discretization, frozen GFDM) appear to reflect hard-won lessons from prior VLA work.
Li Zhang Fudan University / Shanghai Innovation Institute Corresponding author. Oversees a lab working across video understanding and embodied AI. The institutional affiliation with the Shanghai Innovation Institute signals proximity to Chinese national AI infrastructure priorities.
Xin Jin Eastern Institute of Technology, Ningbo Co-corresponding author. The Ningbo grant funding (multiple awards listed in acknowledgments) suggests active government-backed research deployment context.
Zekun Qi Tsinghua University Co-author, also appears on SoFar (language-grounded spatial reasoning for manipulation) and ShapeLLM (3D object understanding for embodied interaction). Represents a cross-institution collaboration on the physical AI stack.
Yang Tian et al. (Seer / ICLR 2024) Referenced as prior work, not an author of this paper Authors of Seer (Predictive Inverse Dynamics Models), the most direct architectural predecessor to DeFI. Seer achieves 4.28 on CALVIN; DeFI improves to 4.51. Understanding Seer's design is essential context for understanding what DeFI changes.
5. Operating Insights
Inference Latency Is Deployable Today — But Only If You Commit to Single-Step Denoising
The full DeFI inference pipeline (GFDM + GIDM + Action Adapter) runs at approximately 153ms total on an RTX 4090: GFDM at 86.1ms, GIDM at 42.9ms, adapter at 24.3ms. (Table 13, Appendix A.5) This is within the latency budget for many manipulation tasks at 5-10Hz control frequency. However, this requires the single-step denoising design choice. Any team considering this architecture should build around that constraint from day one — the 5-step variant at ~250ms tips into latency-constrained territory for reactive tasks. The inference memory footprint is also notable: only 7GB on CALVIN, despite a 64GB training memory requirement. (Table 10)
The Real Data Moat Is Inverse Dynamics Pretraining, Not Forward Dynamics
For teams deciding where to invest data collection effort: the forward dynamics model (GFDM) benefits from commodity human video (Something-Something-v2, Ego4D — both public), and gains from robot data plateau quickly. The inverse dynamics model (GIDM) is where the architecture's data leverage is most sensitive. The ablation shows GIDM without pretraining still achieves 4.16 (reasonable baseline), but GIDM with pretraining on the full mixed dataset reaches 4.51. More importantly, the GIDM pretraining uses action-free video — no teleoperation required. "Although actions and proprioceptive states are available in these robot datasets, we exclude them during pretraining and rely only on episode frames and text instructions." (Appendix A.2) Teams building novel robot platforms can bootstrap GIDM pretraining from existing public video before collecting a single labeled demonstration.
6. Overlooked Insights
62% of Real Failures Come From the World Model, Not Action Inference — Reversing the Paper's Own Framing
The paper's central argument is that inverse dynamics is underappreciated. But the failure case analysis tells a different story. "We examined 200 failure cases on CALVIN. The errors can be grouped into two major categories: (i) Forward-dynamics failures (62%): More challenging scenarios occur in contact-rich or cluttered interactions, where the FDM may generate hallucinated or physically implausible predictions... (ii) Inverse-dynamics failures (38%): Even when the predicted future is accurate, the IDM may still produce incorrect actions." (Appendix A.4.2) This is a materially important finding for deployment teams: even with DeFI's improved inverse dynamics, the majority of real-world failures will trace back to the visual prediction model's inability to handle contact-rich or cluttered scenes. Any deployment in unstructured environments (warehouse picking, kitchen manipulation, construction) will hit this ceiling first. The paper acknowledges "long-horizon consistency and contact modeling remain bottlenecks for world-model-based approaches" — but this finding deserves more weight than it receives in the main text.
The VQ-VAE Codebook Size Is Tiny — And That's Load-Bearing
The inverse dynamics model quantizes the continuous action space into a vocabulary of only 128 tokens via VQ-VAE. (Appendix A.2, Table 10: "Vocabulary size of VQ codebook: 128") This is dramatically smaller than the typical vocabulary sizes used in language-model-style action tokenization (often 256-1024+). The paper's ablation (Table 8) shows this discrete VQ-VAE approach outperforms Gaussian Mixture, Simple Binning, and Continuous Latent Action alternatives — and the explanation is structural: "The VQ-VAE in our method serves not only as a discretization tool but also as an information-bottleneck mechanism, which stabilizes the learning of inverse dynamics. This quantization step helps prevent future-state leakage into the decoder, ensuring the model learns meaningful action representations instead of relying on low-level visual shortcuts." (Section 4.5, Q5) For anyone building action tokenization into their robot stack, this is an underappreciated design principle: tighter quantization may improve generalization by forcing the model to compress, not just encode.