Doe-1: Closed-Loop Autonomous Driving with Large World Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zheng, Wenzhao, Xia, Zetian, Huang, Yuanhui, Zuo, Sicheng, Zhou, Jie, Lu, Jiwen
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912153841172480
author Zheng, Wenzhao
Xia, Zetian
Huang, Yuanhui
Zuo, Sicheng
Zhou, Jie
Lu, Jiwen
author_facet Zheng, Wenzhao
Xia, Zetian
Huang, Yuanhui
Zuo, Sicheng
Zhou, Jie
Lu, Jiwen
contents End-to-end autonomous driving has received increasing attention due to its potential to learn from large amounts of data. However, most existing methods are still open-loop and suffer from weak scalability, lack of high-order interactions, and inefficient decision-making. In this paper, we explore a closed-loop framework for autonomous driving and propose a large Driving wOrld modEl (Doe-1) for unified perception, prediction, and planning. We formulate autonomous driving as a next-token generation problem and use multi-modal tokens to accomplish different tasks. Specifically, we use free-form texts (i.e., scene descriptions) for perception and generate future predictions directly in the RGB space with image tokens. For planning, we employ a position-aware tokenizer to effectively encode action into discrete tokens. We train a multi-modal transformer to autoregressively generate perception, prediction, and planning tokens in an end-to-end and unified manner. Experiments on the widely used nuScenes dataset demonstrate the effectiveness of Doe-1 in various tasks including visual question-answering, action-conditioned video generation, and motion planning. Code: https://github.com/wzzheng/Doe.
format Preprint
id arxiv_https___arxiv_org_abs_2412_09627
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Doe-1: Closed-Loop Autonomous Driving with Large World Model
Zheng, Wenzhao
Xia, Zetian
Huang, Yuanhui
Zuo, Sicheng
Zhou, Jie
Lu, Jiwen
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
End-to-end autonomous driving has received increasing attention due to its potential to learn from large amounts of data. However, most existing methods are still open-loop and suffer from weak scalability, lack of high-order interactions, and inefficient decision-making. In this paper, we explore a closed-loop framework for autonomous driving and propose a large Driving wOrld modEl (Doe-1) for unified perception, prediction, and planning. We formulate autonomous driving as a next-token generation problem and use multi-modal tokens to accomplish different tasks. Specifically, we use free-form texts (i.e., scene descriptions) for perception and generate future predictions directly in the RGB space with image tokens. For planning, we employ a position-aware tokenizer to effectively encode action into discrete tokens. We train a multi-modal transformer to autoregressively generate perception, prediction, and planning tokens in an end-to-end and unified manner. Experiments on the widely used nuScenes dataset demonstrate the effectiveness of Doe-1 in various tasks including visual question-answering, action-conditioned video generation, and motion planning. Code: https://github.com/wzzheng/Doe.
title Doe-1: Closed-Loop Autonomous Driving with Large World Model
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2412.09627