实时多任务自动驾驶感知模型,兼顾精度与速度。
RMT-PPAD: Real-time Multi-task Learning for Panoptic Perception in Autonomous Driving
- 基于Transformer的统一架构,自适应融合共享与任务特有特征
- 在BDD100K上达84.9% mAP50、92.6% mIoU,推理速度32.6 FPS
- 解决车道线标注不一致问题,适合实际道路场景部署
自动驾驶系统依赖于高精度且实时的全景感知能力。本文提出RMT-PPAD,一种基于Transformer的实时多任务模型,联合完成目标检测、可行驶区域分割和车道线分割。引入轻量级门控适配模块,自适应融合共享特征与任务特有特征,有效缓解任务间负迁移。设计自适应分割解码器,在训练阶段自动学习多尺度特征权重,避免为不同分割任务手动设计结构。同时识别并修复车道线分割中训练与测试标签的不一致问题,实现更公平评估。在BDD100K数据集上的实验表明,RMT-PPAD在目标检测上取得84.9% mAP50和95.4%召回率,可行驶区域分割达到92.6% mIoU,车道线分割达到56.8% IoU和84.7%准确率,推理速度达32.6 FPS。此外,通过真实道路场景验证,模型表现稳定。源代码与预训练模型已开源。
原文摘要 · Abstract (English)
Autonomous driving systems rely on panoptic driving perception that requires both precision and real-time performance. In this work, we propose RMT-PPAD, a real-time, transformer-based multi-task model that jointly performs object detection, drivable area segmentation, and lane line segmentation. We introduce a lightweight module, a gate control with an adapter to adaptively fuse shared and task-specific features, effectively alleviating negative transfer between tasks. Additionally, we design an adaptive segmentation decoder to learn the weights over multi-scale features automatically during the training stage. This avoids the manual design of task-specific structures for different segmentation tasks. We also identify and resolve the inconsistency between training and testing labels in lane line segmentation. This allows fairer evaluation. Experiments on the BDD100K dataset demonstrate that RMT-PPAD achieves state-of-the-art results with mAP50 of 84.9% and Recall of 95.4% for object detection, mIoU of 92.6% for drivable area segmentation, and IoU of 56.8% and accuracy of 84.7% for lane line segmentation. The inference speed reaches 32.6 FPS. Moreover, we introduce real-world scenarios to evaluate RMT-PPAD performance in practice. The results show that RMT-PPAD consistently delivers stable performance. The source codes and pre-trained models are released at https://github.com/JiayuanWang-JW/RMT-PPAD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。