arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.09122cs.CV

MaskBEV:面向BEV检测与地图分割的统一框架

MaskBEV: Towards A Unified Framework for BEV Detection and Map Segmentation

Academy for Engineering and Technology, Fudan University(复旦大学工程技术学院)

Xiao Zhao, Xukun Zhang, Dingkang Yang, Mingyang Sun, Mingcheng Li, Shunli Wang, Lihua Zhang

AI总结:

针对自动驾驶多模态多任务感知存在的任务互补不足、联合训练性能下降问题,提出基于掩码注意力的统一框架MaskBEV,通过任务无关解码器与空间调制等策略提升性能,在nuScenes数据集上检测与分割精度均优于现有方法,推理速度略有领先。

摘要:

准确且鲁棒的多模态多任务感知对现代自动驾驶系统至关重要。然而,当前多模态感知研究遵循为特定感知任务设计的独立范式,导致任务间缺乏互补学习,且多任务学习(MTL)的性能会因联合训练而下降。本文提出了MaskBEV,这是一种基于掩码注意力的多任务学习范式,可统一3D目标检测与鸟瞰图(BEV)地图分割任务。MaskBEV引入了一个任务无关的Transformer解码器来处理这些不同的任务,无需额外设计特定任务头,即可在统一解码器中完成多任务学习。为了充分挖掘BEV空间中BEV地图分割与3D目标检测任务之间的互补信息,我们提出了空间调制和场景级上下文聚合策略。这些策略考虑了BEV分割与3D检测之间的内在依赖关系,能够自然地提升多任务学习性能。在nuScenes数据集上开展的大量实验表明,与此前最先进的多任务学习方法相比,MaskBEV在3D目标检测任务上实现了1.3 NDS的提升,在BEV地图分割任务上实现了2.7 mIoU的提升,同时推理速度也略有领先。

英文摘要:

Accurate and robust multimodal multi-task perception is crucial for modern autonomous driving systems. However, current multimodal perception research follows independent paradigms designed for specific perception tasks, leading to a lack of complementary learning among tasks and decreased performance in multi-task learning (MTL) due to joint training. In this paper, we propose MaskBEV, a masked attention-based MTL paradigm that unifies 3D object detection and bird's eye view (BEV) map segmentation. MaskBEV introduces a task-agnostic Transformer decoder to process these diverse tasks, enabling MTL to be completed in a unified decoder without requiring additional design of specific task heads. To fully exploit the complementary information between BEV map segmentation and 3D object detection tasks in BEV space, we propose spatial modulation and scene-level context aggregation strategies. These strategies consider the inherent dependencies between BEV segmentation and 3D detection, naturally boosting MTL performance. Extensive experiments on nuScenes dataset show that compared with previous state-of-the-art MTL methods, MaskBEV achieves 1.3 NDS improvement in 3D object detection and 2.7 mIoU improvement in BEV map segmentation, while also demonstrating slightly leading inference speed.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑