Source-linked AI summary
ADriver-I: A General World Model for Autonomous Driving
Fan Jia, Weixin Mao, Yingfei Liu, Yucheng Zhao, Yuqing Wen, Chi Zhang, Xiangyu Zhang, Tiancai Wang
TL;DR
Modular driving systems divide perception, prediction, planning, and control, while the paper explores direct control prediction from visual information alongside prediction of future scenes. ADriver-I pairs historical visual features and controls, uses an MLLM to predict current actions and a diffusion model to generate future frames, and reports control and generation results on nuScenes and private data. The authors also identify limitations in video quality, deployment performance, and separate model training.
Problem
The paper asks whether a driving system can directly output control signals from visual information and predict the near-future scene, rather than using a sequential perception-prediction-planning-control pipeline.
Source: §1 · p. 1Method
ADriver-I uses interleaved vision-action pairs with an MLLM for current-frame control prediction and a video diffusion model conditioned on predicted actions and historical pairs for future-frame generation.
Source: §1 · p. 1, §3.1 · p. 3Results
ADriver-I reports speed and steer-angle L1 errors of 0.072 m/s and 0.091 rad, respectively, and FID 5.52 and FVD 97 for generated future four frames.
Source: §1 · p. 2Takeaways & Limitations
With initial vision-action pairs, ADriver-I can repeatedly predict controls and generate subsequent scenes in a world created by the model itself.
Source: §4.5 · p. 8, §5 · p. 8The authors report that rapidly changing controls can coincide with low-quality VDM frames that disturb subsequent control prediction, deployment performance remains unsatisfactory, and separate training prevents end-to-end optimization.
Source: §5 · p. 8
Abstract
from arXiv · showhide
Typically, autonomous driving adopts a modular design, which divides the full stack into perception, prediction, planning and control parts. Though interpretable, such modular design tends to introduce a substantial amount of redundancy. Recently, multimodal large language models (MLLM) and diffusion techniques have demonstrated their superior performance on comprehension and generation ability. In this paper, we first introduce the concept of interleaved vision-action pair, which unifies the format of visual features and control signals. Based on the vision-action pairs, we construct a general world model based on MLLM and diffusion model for autonomous driving, termed ADriver-I. It takes the vision-action pairs as inputs and autoregressively predicts the control signal of the current frame. The generated control signals together with the historical vision-action pairs are further conditioned to predict the future frames. With the predicted next frame, ADriver-I performs further control signal prediction. Such a process can be repeated infinite times, ADriver-I achieves autonomous driving in the world created by itself. Extensive experiments are conducted on nuScenes and our large-scale private datasets. ADriver-I shows impressive performance compared to several constructed baselines. We hope our ADriver-I can provide some new insights for future autonomous driving and embodied intelligence.
1. Introduction
The paper asks whether a driving system can directly produce control signals from visual information and predict near-future scenes, rather than relying on a sequential modular pipeline. ADriver-I combines MLLM-based control prediction with diffusion-based future-scene generation, and reports results on nuScenes and a private dataset.
1. Introduction
- The system targets direct control from visual information and near-future scene prediction, in contrast to the perception-prediction-planning-control sequence of modular driving systems.The modular design is described as interpretable, while human drivers are characterized as acting from visual information and anticipating the near future.Source: §1 · p. 1
- ADriver-I introduces interleaved vision-action pairs to unify visual features and control signals, allowing historical pairs and current visual tokens to inform current-frame control prediction.The control signals include steer angles and ego speed, converted into text-like expressions.Source: §1 · p. 1
- The predicted current action conditions video diffusion to generate future frames, which can then support further control prediction.The abstracted process is described as repeatable in a world generated by ADriver-I itself.Source: §1 · p. 2
- ADriver-I reports current-frame L1 errors of 0.072 m/s for speed and 0.091 rad for steer angle, with FID 5.52 and FVD 97 for generated future four frames.These reported metrics use three historical vision-action pairs for control prediction.Source: §1 · p. 2
2. Related Works
Related work spans multimodal models, end-to-end driving, scene generation, and world models. ADriver-I is positioned as combining control prediction with future-scene generation while avoiding extensive priors for future generation.
2.1. Multimodal Large Language Models
- MLLMs support cross-modal comprehension and have inspired embodied-intelligence work, including systems that predict motor actions or low-level control signals.The related work also identifies DriveGPT4 as a concurrent MLLM-based system that outputs control signals and explanations.Source: §2.1 · p. 2
2.2. End-to-end Autonomous Driving
- End-to-end driving methods range from direct waypoint prediction to approaches using scene representations such as cost maps, segmentation, motion flow, occupancy, or vectorized agents and maps.Transfuser and TCP predict waypoints without constructing scene representations.Source: §2.2 · p. 2
2.3. Generative Models for Autonomous Driving
- Driving-scene generation includes GAN, VAE, and diffusion approaches, with examples spanning action-correlated videos, BEV-conditioned images, and controllable street-view generation.Video latent diffusion is cited as an approach to high-resolution video synthesis.Source: §2.3 · pp. 2–3
2.4. World Models for Autonomous Driving
- ADriver-I unifies control-signal prediction with future-scene generation, whereas GAIA-1 focuses on scenario generation and DriveDreamer relies on HD maps and 3D bounding boxes for future generation.The authors introduce “infinite driving” and state that ADriver-I achieves it in the world it creates.Source: §2.4 · p. 3
3. Method
ADriver-I combines an MLLM for low-level control prediction with a video diffusion model for future-frame generation. Its prompt construction pairs visual and action tokens, while separate training procedures are used for the two models.
3. Method
- The method uses an MLLM and video diffusion model, trained separately and merged during inference.The method section introduces architecture, prompt construction, and training as its three parts.Source: §3 · p. 3, §3.3 · p. 5
3.1. Architecture
- The MLLM predicts current-frame control from the current frame and three historical vision-action pairs, then its action conditions the VDM to predict four future frames.At inference, the predicted next frame becomes the current frame for the next cycle, enabling repeated control and generation.Source: §3.1 · p. 3
3.2. Prompt Construction
- Visual tokens paired with text-tokenized low-level controls form interleaved vision-action pairs, which support variable conversation lengths and unify prediction and generation in the word-embedding space.For VDM prompting, GPT3.5 converts consecutive-frame controls into motion descriptions such as accelerating, decelerating, or turning.Source: §3.2 · pp. 4–5
3.3. Model Training
- MLLM pretraining uses nearly 1.4M private-dataset highway pairs, with nuScenes and private data used for supervised fine-tuning; VDM is pretrained on the private data and fine-tuned on about 23K nuScenes video samples.The models are trained separately before being merged for inference.Source: §3.3 · p. 5
4. Experiments
Experiments evaluate control prediction on nuScenes and a private dataset, future-scene generation, and recurrent joint control and generation.
4.1. Implementation Details
- The MLLM and video diffusion model are trained separately and merged for inference; the MLLM uses three historical vision-action pairs and the current image as input.The VDM is pretrained on private data and fine-tuned on nuScenes.Source: §4.1 · pp. 5–6
4.2. Evaluation Metrics
- Control prediction is evaluated with L1 error and threshold accuracy, while future-frame quality uses frame-wise FID and FVD.Threshold accuracy uses θ values of 0.01, 0.03, 0.05, and 0.07.Source: §4.2 · p. 6
4.3. Control Signal Prediction
- ADriver-I outperforms three baselines on nuScenes; private-dataset speed and steer-angle L1 errors are 0.035 m/s and 0.015 rad.The private dataset focuses on highway scenes and has more supervised-finetuning data than nuScenes (1.4M vs. 23K).Source: §4.3 · pp. 6–7
- Direct absolute-number prediction performs better than English-description and relative-difference encodings.The caption also identifies special-token encoding as an evaluated method, but the text does not state its comparative outcome.Source: Table 3 · p. 6, §4.3 · p. 7
- Two decimal places perform almost as well as three and outperform fewer decimal places.The paper attributes accuracy error to integer or one-decimal-place representations.Source: §4.3 · p. 7, Table 4 · p. 7
- Multi-round conversation outperforms single-round supervision by a large margin on speed prediction.Multi-round training adds supervision on intermediate action predictions.Source: §4.3 · p. 7, Table 5 · p. 7
4.4. Future Scene Generation
- 5.5 FID and 97.0 FVD are reported for four generated future frames conditioned on four historical frames, without 3D bounding boxes or HD maps.The diffusion model uses historical frames and MLLM-predicted control signals to generate future scenes.Source: §4.4 · p. 8
4.5. Joint Control&Generation
- With three historical vision-action pairs, recurrent inference uses MLLM-predicted controls and diffusion-generated frames, with each affecting the next stage.The reported sequence shows predicted controls affecting future scenes and generated scenes pushing the model to take corresponding actions.Source: §4.5 · p. 8
5. Conclusion and Discussion
ADriver-I integrates an MLLM with a video diffusion model for control and future-frame prediction, reporting performance on nuScenes and private data and infinite driving in its generated world. The authors identify limitations in frame quality, deployment readiness, training, and routing, and point to generative scaling as future work.
- ADriver-I integrates an MLLM with a video diffusion model and outputs low-level control without the mainstream sequential pipeline.Source: §5 · p. 8
- ADriver-I reports impressive performance on nuScenes and private data and achieves infinite driving in its generated world from initial vision-action pairs.Source: §5 · p. 8
- ADriver-I predicts near-future frames from historical information.Source: §5 · p. 8
- Quick control changes may lead the video diffusion model to generate low-quality frames that disturb the next control prediction.Source: §5 · p. 8
- The authors consider performance far from satisfactory for deployment and plan to train on more vision-control pairs.Source: §5 · p. 8
- The MLLM and video diffusion model are trained separately, so they do not benefit from end-to-end optimization; the authors call for a unified framework.Source: §5 · p. 8
- The system lacks routing information for long-distance driving, and the authors suggest introducing a navigation map.Source: §5 · p. 8
- The authors hope future work will observe a scaling law from the generative perspective.Source: §5 · p. 9
Related papers
This paper cites
- Learning Transferable Visual Models From Natural Language Supervision2021
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale2020
- Auto-Encoding Variational Bayes2013
- LLaMA: Open and Efficient Foundation Language Models2023
- BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Models2023
- RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Control2023