UNIVERSE: Unified Video Action Models for Autonomous Driving with Flexible Mask-Modulated Modality Generation
TLDR
UNIVERSE proposes a unified mask-modulated Diffusion Transformer for autonomous driving, co-training future video and trajectory tokens to improve action generalization and enable efficient trajectory-only inference.
Reasoning
The paper presents a well-motivated unified architecture that directly couples video prediction with trajectory generation, addressing a key limitation of cascaded designs. Its Modality-Decoupling Visibility Mask is a clever solution for preventing future-target leakage while enabling efficient deployment. However, the abstract does not provide explicit dataset or benchmark details, making it difficult to assess the breadth of empirical validation.
Read-first score
Read-first score 41.1, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 45.
Field roles
Rank sensitivity
Stability: volatile; rank range: 629.
Keyword Scores
Deep Analysis
Innovations
- Unified single-DiT video-action model co-training future video latents and ego-trajectory tokens with shared generative parameters, unlike cascaded or dual-DiT designs.
- Modality-Decoupling Visibility Mask that shares historical context across modalities while blocking mutual attention between future video and trajectory tokens, preventing future-target leakage.
- Flexible inference modes: the same model supports video-only, trajectory-only, and joint video-action rollouts; trajectory-only inference removes future-video denoising for a 4.3x speedup.
Methodology
UNIVERSE is built on a single mask-modulated Diffusion Transformer that co-trains future video latents and ego-trajectory tokens in shared generative parameters. A Modality-Decoupling Visibility Mask shares historical context while blocking future video-trajectory attention, and the model can perform video-only, trajectory-only, or joint video-action rollouts. Evaluation uses PDMS on NAVSIM, zero-shot transfer to nuScenes and Bench2Drive, and ablations against a Two-DiT variant.
Key Results
UNIVERSE achieves 91.0 PDMS on NAVSIM versus 89.6 for the Two-DiT variant, and shows strong zero-shot transfer to nuScenes and Bench2Drive without fine-tuning. Trajectory-only inference is 4.3x faster than joint video-action rollout while maintaining comparable planning accuracy.
Limitations
- The abstract does not explicitly state limitations of the proposed method.
- Trajectory-only inference is described as maintaining 'comparable' rather than identical planning accuracy, suggesting a possible accuracy trade-off for faster inference.