English

FusionAD: Multi-modality Fusion for Prediction and Planning Tasks of Autonomous Driving

Computer Vision and Pattern Recognition 2023-08-15 v4 Artificial Intelligence Robotics

Abstract

Building a multi-modality multi-task neural network toward accurate and robust performance is a de-facto standard in perception task of autonomous driving. However, leveraging such data from multiple sensors to jointly optimize the prediction and planning tasks remains largely unexplored. In this paper, we present FusionAD, to the best of our knowledge, the first unified framework that fuse the information from two most critical sensors, camera and LiDAR, goes beyond perception task. Concretely, we first build a transformer based multi-modality fusion network to effectively produce fusion based features. In constrast to camera-based end-to-end method UniAD, we then establish a fusion aided modality-aware prediction and status-aware planning modules, dubbed FMSPnP that take advantages of multi-modality features. We conduct extensive experiments on commonly used benchmark nuScenes dataset, our FusionAD achieves state-of-the-art performance and surpassing baselines on average 15% on perception tasks like detection and tracking, 10% on occupancy prediction accuracy, reducing prediction error from 0.708 to 0.389 in ADE score and reduces the collision rate from 0.31% to only 0.12%.

Keywords

Cite

@article{arxiv.2308.01006,
  title  = {FusionAD: Multi-modality Fusion for Prediction and Planning Tasks of Autonomous Driving},
  author = {Tengju Ye and Wei Jing and Chunyong Hu and Shikun Huang and Lingping Gao and Fangzhen Li and Jingke Wang and Ke Guo and Wencong Xiao and Weibo Mao and Hang Zheng and Kun Li and Junbo Chen and Kaicheng Yu},
  journal= {arXiv preprint arXiv:2308.01006},
  year   = {2023}
}