OccLLaMA: An Occupancy-Language-Action Generative World Model for Autonomous Driving

TL;DR

OccLLaMA introduces a unified generative world model for autonomous driving that leverages semantic occupancy, vision, language, and action modalities to improve scene understanding, prediction, and planning.

Contribution

It proposes a novel occupancy-language-action generative model with a scene tokenizer and a unified multi-modal vocabulary, enabling multi-task autonomous driving capabilities.

Findings

Achieves competitive performance in 4D occupancy forecasting

Effective in motion planning tasks

Demonstrates versatility in visual question answering

Abstract

The rise of multi-modal large language models(MLLMs) has spurred their applications in autonomous driving. Recent MLLM-based methods perform action by learning a direct mapping from perception to action, neglecting the dynamics of the world and the relations between action and world dynamics. In contrast, human beings possess world model that enables them to simulate the future states based on 3D internal visual representation and plan actions accordingly. To this end, we propose OccLLaMA, an occupancy-language-action generative world model, which uses semantic occupancy as a general visual representation and unifies vision-language-action(VLA) modalities through an autoregressive model. Specifically, we introduce a novel VQVAE-like scene tokenizer to efficiently discretize and reconstruct semantic occupancy scenes, considering its sparsity and classes imbalance. Then, we build a unified multi-modal vocabulary for vision, language and action. Furthermore, we enhance LLM, specifically LLaMA, to perform the next token/scene prediction on the unified vocabulary to complete multiple tasks in autonomous driving. Extensive experiments demonstrate that OccLLaMA achieves competitive performance across multiple tasks, including 4D occupancy forecasting, motion planning, and visual question answering, showcasing its potential as a foundation model in autonomous driving.