UniAction: Unified Action Conditioning
for World Models

Comparison of Conditioning Mechanisms

Overlays indicate the input action.

Benefits of Unified Training

Temporal Generalization

Both models are trained with forces applied only at t = 0.