M3Net统一处理自动驾驶中的检测、分割与占用预测,提升多任务感知效率。
M3Net: Multimodal Multi-task Learning for 3D Detection, Segmentation, and Occupancy Prediction in Autonomous Driving

- 通过模态自适应融合和任务专用查询初始化,实现多模态数据高效共享
- 在nuScenes上三项任务均达最优,检测精度提升2.1%,占用预测提升3.4%
- 支持Transformer与Mamba双解码器架构,适合追求端到端感知的工程师
自动驾驶感知系统需同时完成多项子任务,但现有方法通常分步处理,效率低下。部分多任务学习方法虽尝试统一模型,却未能解决任务间冲突。本文提出M3Net,一种新型多模态多任务网络,可同时处理3D目标检测、分割与占用预测,性能优于单任务模型。M3Net采用多模态输入,通过查询-令牌交互实现多任务协同。为增强多模态特征融合,提出模态自适应特征融合(MAFI)模块,使单模态特征能为各自优势任务生成通道注意力权重。基于融合特征,设计任务特异性查询初始化策略,适配检测/分割与3D占用预测需求。利用合理初始化查询,共享解码器逐层转换查询与鸟瞰图特征,促进多任务学习。进一步提出任务导向通道缩放(TCS)模块,缓解不同任务优化间的冲突。所提多任务查询与TCS模块兼容Transformer与Mamba解码器,展现架构灵活性。M3Net在nuScenes基准上实现领先多任务学习性能。
原文摘要 · Abstract (English)
The perception system for autonomous driving generally requires to handle multiple diverse sub-tasks. However, current algorithms typically tackle individual sub-tasks separately, which leads to low efficiency when aiming at obtaining full-perception results. Some multi-task learning methods try to unify multiple tasks with one model, but do not solve the conflicts in multi-task learning. In this paper, we introduce M3Net, a novel multimodal and multi-task network that simultaneously tackles detection, segmentation, and 3D occupancy prediction for autonomous driving and achieves superior performance than single task model. M3Net takes multimodal data as input and multiple tasks via query-token interactions. To enhance the integration of multi-modal features for multi-task learning, we first propose the Modality-Adaptive Feature Integration (MAFI) module, which enables single-modality features to predict channel-wise attention weights for their high-performing tasks, respectively. Based on integrated features, we then develop task-specific query initialization strategies to accommodate the needs of detection/segmentation and 3D occupancy prediction. Leveraging the properly initialized queries, a shared decoder transforms queries and BEV features layer-wise, facilitating multi-task learning. Furthermore, we propose a Task-oriented Channel Scaling (TCS) module in the decoder to mitigate conflicts between optimizing for different tasks. Additionally, our proposed multi-task querying and TCS module support both Transformer-based decoder and Mamba-based decoder, demonstrating its flexibility to different architectures. M3Net achieves state-of-the-art multi-task learning performance on the nuScenes benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。