Copilot4D: Learning Unsupervised World Models for Autonomous Driving via Discrete Diffusion
TLDR
Copilot4D uses VQVAE tokenization and discrete diffusion to learn unsupervised world models for autonomous driving, achieving significant improvements in point cloud prediction.
Reasoning
The paper presents a novel combination of VQVAE and discrete diffusion for world modeling, with strong empirical results on multiple real-world datasets. However, it focuses on point cloud observations and short-term prediction, and lacks explicit interaction or reinforcement learning components.
Read-first score
Read-first score 62.8, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 42.
Field roles
Rank sensitivity
Stability: volatile; rank range: 116.
Keyword Scores
Deep Analysis
Innovations
- Tokenizing sensor observations with VQVAE for world modeling in autonomous driving
- Recasting Masked Generative Image Transformer as discrete diffusion with enhancements for efficient parallel decoding
- Achieving significant reduction in Chamfer distance (over 65% for 1s, over 50% for 3s) compared to prior SOTA on multiple datasets
- Demonstrating that discrete diffusion on tokenized agent experience can unlock GPT-like unsupervised learning for robotics
Methodology
Copilot4D first tokenizes sensor observations (point clouds) using a VQVAE, then models future prediction as a discrete diffusion process. It recasts the Masked Generative Image Transformer as discrete diffusion and introduces simple enhancements to improve efficiency. The model is trained and evaluated on NuScenes, KITTI Odometry, and Argoverse2 datasets, using Chamfer distance as the metric for 1s and 3s prediction horizons.
Key Results
Copilot4D reduces prior state-of-the-art Chamfer distance by more than 65% for 1s prediction and more than 50% for 3s prediction across NuScenes, KITTI Odometry, and Argoverse2 datasets.
Limitations
- Only evaluated on point cloud observations, not other sensor modalities such as cameras or radar.
- Prediction horizons are limited to 1s and 3s; long-term prediction performance is not assessed.