Summary (Overview)
- RynnWorld-4D introduces a projective 4D representation (RGB, Depth, Optical Flow – RGB-DF) that makes geometry and motion explicit while staying compatible with large-scale video diffusion priors.
- The method extends a video diffusion transformer into a tri-branch architecture with joint cross-modal attention and 3D RoPE to generate synchronized, physically coherent 4D sequences from a single RGB-D image and language instruction.
- To train the model, the authors curate Rynn4DDataset 1.0, a massive hybrid dataset of 254.4 million frames from egocentric human and robotic manipulation videos, pseudo-annotated with depth and optical flow.
- The RynnWorld-4D-Policy head extracts actions directly from the model’s internal 4D features in a single forward pass, bypassing iterative denoising and enabling closed-loop control at ~9 Hz.
- Experiments show the model achieves state‑of‑the‑art success rates across six real‑world bimanual dexterous manipulation tasks, especially those requiring spatial precision and temporal coordination (e.g., 65.71% on Bowl Stacking and Lid Placement).
Introduction and Theoretical Foundation
Robotic manipulation in open-world environments requires not only visual recognition but also the ability to anticipate how the 3D structure of a scene will move under interaction. Current generative video models operate in 2D pixel space, losing critical spatial relationships and leading to temporal inconsistencies (e.g., fluctuating object scales). The authors argue that predicting synchronized RGB, depth, and optical flow (RGB-DF) provides a physically grounded representation that:
- Lifts each pixel to a 3D location (via depth),
- Tracks 3D motion over time (via depth + optical flow → 3D scene flow),
- Stays in a 2D‑aligned format, inheriting the scalability and rich generative priors of pretrained video diffusion models.
This representation bridges the gap between world prediction and policy learning, as the predicted 4D dynamics align more closely with the robot’s action space than raw 2D pixel changes.
Methodology
1. Rynn4DDataset 1.0 (Data Curation)
The dataset comprises over 254.4 million video frames from human egocentric (Epic-Kitchens, EgoVid) and robotic manipulation (RoboMIND, RDT-1B, Galaxea, RoboCoin, AgiBot) datasets. Each frame is annotated with:
- Video captions via Qwen3-VL (512 tokens, temperature 0.7).
- Optical flow via DPFlow (dense per-frame motion fields).
- Depth via Depth Anything 3 (DA3NESTED-GIANT-LARGE-1.1), with values clipped to [0.0, 5.0] meters and quantized to 8-bit grayscale.
2. 3D Scene Reconstruction from Multi-Modal Videos
The key geometric interpretability is given by:
Geometric Unprojection (Equation 1):
where is a 3D point from pixel at frame , is the depth map, and the camera intrinsic matrix.
Metric Scene Flow (Equation 2):
where is the optical flow.
3. RynnWorld-4D (Generative Model)
The model extends the Wan 2.2-TI2V-5B diffusion transformer into a tri-branch architecture (RGB, depth, flow). Key components:
-
Joint Cross-Modal Attention (JA): Inserted every 3 of 30 layers (10 modules total). Each branch produces queries and shared keys/values . Queries attend to the concatenated keys/values of the complementary modalities (Equation 5):
with , .
-
3D Rotary Positional Embeddings (RoPE) applied frame-wise to maintain cross-modal spatial correspondence.
-
Phased Training: Three stages – (1) independent branch adaptation, (2) frozen backbone + JA training, (3) full joint fine-tuning. A Branch Dropout mechanism randomly drops depth or flow latents to improve robustness.
-
Training Objective (Equation 7):
with flow-matching on frames and a shared noise sample .
| Stage | Fusion mode | Trainable params | LR | Branch Dropout | |
|---|---|---|---|---|---|
| 1 | none | all branches | 0.5 | — | |
| 2 | joint (frozen backbone) | JA + mod. embed. | 1.0 | 0.2 | |
| 3 | joint (full SFT) | all parameters | 1.0 | 0.1 |
4. RynnWorld-4D-Policy (Action Decoder)
Extracts intermediate hidden states from the frozen RynnWorld-4D encoder (block 15, diffusion timestep ), concatenates RGB/depth/flow latents into , and compresses them via a Flow Former (spatial cross-attention + temporal self-attention). Actions are generated with a flow matching head (4-step Euler ODE), predicting chunks of 54-dimensional actions.
Empirical Validation / Results
4D World Modeling Quality (Table 4)
RynnWorld-4D achieves the best performance across depth and flow metrics among all baselines, while maintaining competitive visual generation quality.
| Method | RGB (SSIM ↑) | RGB (PSNR ↑) | Depth (AbsRel ↓) | Depth ( ↑) | Flow (AEPE ↓) |
|---|---|---|---|---|---|
| Wan-2.2-TI2V-5B | 0.593 | 14.54 | N/A | N/A | N/A |
| TesserAct | 0.693 | 16.91 | 0.699 | 0.279 | N/A |
| 4DNeX | 0.649 | 14.47 | 0.423 | 0.327 | N/A |
| RynnWorld-4D | 0.754 | 17.85 | 0.310 | 0.610 | 0.170 |
| w/o 4D Pre-training | 0.651 | 16.25 | 0.797 | 0.263 | 0.729 |
Significant drops in the ablation “w/o 4D Pre‑training” confirm the importance of large‑scale data.
Real‑world Manipulation Tasks (Table 5)
Success rates (%) on six tasks evaluated over 35 trials each.
| Method | Dual Picking | Block Pushing | Hand‑over | Bimanual Lifting | Lid Placement | Bowl Stacking |
|---|---|---|---|---|---|---|
| DP (Chi et al., 2025) | 77.14 | 85.71 | 17.14 | 88.57 | 57.14 | 57.14 |
| 88.57 | 94.29 | 2.86 | 91.43 | 34.29 | 51.43 | |
| RynnWorld-4D-Policy | 94.29 | 97.14 | 28.57 | 97.14 | 65.71 | 65.71 |
| w/o RynnWorld-4D (ResNet-18) | 71.43 | 88.57 | 11.43 | 85.71 | 51.43 | 60.00 |
| RGB only | 77.14 | 91.43 | 14.29 | 91.43 | 57.14 | 60.00 |
| RGB + Depth | 91.43 | 91.43 | 28.57 | 97.14 | 60.00 | 62.86 |
| RGB + Flow | 85.71 | 88.57 | 20.00 | 88.57 | 54.29 | 62.86 |
The full 4D representation (RGB‑DF) consistently outperforms single‑ or two‑modality variants, especially on tasks requiring spatial precision (Hand‑over, Lid Placement, Bowl Stacking).
Inference Latency (Table 1)
| Phase | Latency (ms) | Percentage |
|---|---|---|
| Depth Estimation (DA3) | 85 | 7.7% |
| VAE Encoding & Latent Prep | 18 | 1.6% |
| RynnWorld-4D | 990 | 89.5% |
| Feature Reshape & Concat | 1 | 0.1% |
| Flow Former | 4 | 0.4% |
| Action Flow Matching Head | 8 | 0.7% |
| Total (Forward) | 1,106 | 100% |
With action chunking (), the effective control frequency is ~9 Hz.
Theoretical and Practical Implications
- The RGB-DF projective 4D representation provides a natural bridge between video diffusion priors and robotic action spaces, enabling explicit 3D geometry and motion without abandoning the scalability of 2D generative models.
- By generating synchronized depth and optical flow, the model enforces physical consistency (e.g., optical flow aligns with depth changes), reducing temporal artifacts common in RGB‑only generation.
- The RynnWorld-4D-Policy demonstrates that internal 4D latents can be used as a predictive vision encoder for closed‑loop control, achieving state‑of‑the‑art performance in dexterous bimanual tasks without requiring iterative decoding of full videos during action execution.
- Practical deployment on an RTX 5090 yields a 9 Hz effective control rate, sufficient for human‑scale manipulation, with robustness provided by 3D‑aware planning (wider capture range, motion anticipation).
Conclusion
RynnWorld-4D presents a complete framework for 4D embodied world modeling: from data curation (Rynn4DDataset 1.0) to a tri‑branch diffusion model that co‑generates RGB, depth, and optical flow, to a policy head that leverages these predictive 4D features for real‑time control. Extensive experiments validate that explicit geometry and motion are crucial for precise manipulation. The main limitations are computational overhead (~1.1 s per world model forward pass) and a focus on egocentric perspectives; future work may extend to multi‑view or multi‑robot setups. The work establishes a promising foundation for building general‑purpose embodied intelligence that understands and interacts with the 3D world.
Related papers
- OrbitQuant: Data-Agnostic Quantization for Image and Video Diffusion Transformers
OrbitQuant achieves calibration-free W2A4 quantization of diffusion transformers via a shared RPBH rotation and Lloyd–Max codebook, outperforming prior PTQ.
- Embodied.cpp: A Portable Inference Runtime of Embodied AI Models on Heterogeneous Robots
Embodied.cpp unifies VLA and world-action model deployment on edge devices, achieving 100% success rates and 71.8% memory reduction via modular multi-rate execution.
- Trimming the Long-Tail of Visual World Modeling Evaluation
Visual world models fail on long-tail physical interactions, relying on memorized templates rather than true physical reasoning.