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):

Pt=Dt(u,v)K1ptP_t = D_t(u, v) \cdot K^{-1} p_t

where PtP_t is a 3D point from pixel pt=[u,v,1]p_t = [u, v, 1]^\top at frame tt, DtD_t is the depth map, and KK the camera intrinsic matrix.

Metric Scene Flow (Equation 2):

Pt+1=Dt+1(u+Δu,v+Δv)K1(pt+[Δu,Δv,0])P_{t+1} = D_{t+1}(u+\Delta u, v+\Delta v) \cdot K^{-1} (p_t + [\Delta u, \Delta v, 0]^\top) f3D=Pt+1Ptf_{\text{3D}} = P_{t+1} - P_t

where [Δu,Δv][\Delta u, \Delta v]^\top 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 mm produces queries QlmQ^m_l and shared keys/values Klm,VlmK^m_l, V^m_l. Queries attend to the concatenated keys/values of the complementary modalities (Equation 5):

    Alm=AttnRoPE(Qlm,Kcrossl,Vcrossl)A^m_l = \text{Attn}_{\text{RoPE}}(Q^m_l, K_{\text{cross}}^l, V_{\text{cross}}^l)

    with Kcrossl=concat({Klj}jm)K_{\text{cross}}^l = \text{concat}(\{K^j_l\}_{j \neq m}), Vcrossl=concat({Vlj}jm)V_{\text{cross}}^l = \text{concat}(\{V^j_l\}_{j \neq m}).

  • 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):

    Ltotal=mMλmE[vθm(ztm,t,c)[1:](ϵmz0m)[1:]22]\mathcal{L}_{\text{total}} = \sum_{m \in \mathcal{M}} \lambda_m \, \mathbb{E}\left[\left\| v_\theta^m(z_t^m, t, c)_{[1:]} - (\epsilon^m - z_0^m)_{[1:]} \right\|_2^2\right]

    with flow-matching on frames 1,,T11,\dots,T-1 and a shared noise sample ϵrgb=ϵdepth=ϵflow\epsilon^\text{rgb}=\epsilon^\text{depth}=\epsilon^\text{flow}.

StageFusion modeTrainable paramsLRλflow\lambda_{\text{flow}}Branch Dropout
1noneall branches2×1052\times10^{-5}0.5
2joint (frozen backbone)JA + mod. embed.5×1055\times10^{-5}1.00.2
3joint (full SFT)all parameters1×1051\times10^{-5}1.00.1

4. RynnWorld-4D-Policy (Action Decoder)

Extracts intermediate hidden states from the frozen RynnWorld-4D encoder (block 15, diffusion timestep t=500t=500), concatenates RGB/depth/flow latents into FpRB×T×3C×H×W\mathbf{F}_p \in \mathbb{R}^{B \times T \times 3C \times H \times W}, 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 K=10K=10 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.

MethodRGB (SSIM ↑)RGB (PSNR ↑)Depth (AbsRel ↓)Depth (δ1\delta_1 ↑)Flow (AEPE ↓)
Wan-2.2-TI2V-5B0.59314.54N/AN/AN/A
TesserAct0.69316.910.6990.279N/A
4DNeX0.64914.470.4230.327N/A
RynnWorld-4D0.75417.850.3100.6100.170
w/o 4D Pre-training0.65116.250.7970.2630.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.

MethodDual PickingBlock PushingHand‑overBimanual LiftingLid PlacementBowl Stacking
DP (Chi et al., 2025)77.1485.7117.1488.5757.1457.14
π0\pi_088.5794.292.8691.4334.2951.43
RynnWorld-4D-Policy94.2997.1428.5797.1465.7165.71
w/o RynnWorld-4D (ResNet-18)71.4388.5711.4385.7151.4360.00
RGB only77.1491.4314.2991.4357.1460.00
RGB + Depth91.4391.4328.5797.1460.0062.86
RGB + Flow85.7188.5720.0088.5754.2962.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)

PhaseLatency (ms)Percentage
Depth Estimation (DA3)857.7%
VAE Encoding & Latent Prep181.6%
RynnWorld-4D99089.5%
Feature Reshape & Concat10.1%
Flow Former40.4%
Action Flow Matching Head80.7%
Total (Forward)1,106100%

With action chunking (K=10K=10), 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