Semi-Supervised Vision-Centric 3D Occupancy World Model for Autonomous Driving
TLDR
Proposes a semi-supervised 3D occupancy world model using 2D labels and volume rendering for autonomous driving forecasting and planning.
Reasoning
Strengths include a novel semi-supervised paradigm reducing annotation cost and integration of forecasting and planning tasks. Weaknesses are reliance on a single dataset and lack of explicit comparison to fully-supervised baselines in the abstract.
Read-first score
Read-first score 66.1, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 27.
Field roles
Rank sensitivity
Stability: volatile; rank range: 570.
Keyword Scores
Deep Analysis
Innovations
- Semi-supervised learning paradigm for 3D occupancy world model using only 2D labels during pre-training
- Two-stage training: self-supervised pre-training with attribute projection head and volume rendering, followed by fully-supervised fine-tuning
- Attribute projection head that generates RGB, density, and semantic fields from 3D occupancy to enable temporal supervision from 2D labels
- State-conditioned forecasting module for recursive prediction of future occupancy and ego trajectory
Methodology
PreWorld employs a two-stage training paradigm. In the self-supervised pre-training stage, an attribute projection head generates RGB, density, and semantic fields from 3D occupancy, and volume rendering provides temporal supervision from 2D labels. In the fully-supervised fine-tuning stage, 3D occupancy labels are used. A state-conditioned forecasting module recursively predicts future occupancy and ego trajectory. Experiments are conducted on the nuScenes dataset.
Key Results
PreWorld achieves competitive performance on 3D occupancy prediction, 4D occupancy forecasting, and motion planning tasks on the nuScenes dataset, demonstrating effectiveness and scalability.
Limitations
- Still requires 3D occupancy labels for the fine-tuning stage, limiting full label efficiency
- Evaluation is limited to the nuScenes dataset; generalization to other datasets or conditions is not demonstrated