统一多模态多任务学习框架,提升辅助驾驶感知能力
MMTL-UniAD: A Unified Framework for Multimodal and Multi-Task Learning in Assistive Driving Perception
- 设计多轴区域注意力与双分支嵌入结构,缓解任务间干扰
- 在AIDE数据集上四项任务均超越当前最优方法
- 适合自动驾驶感知系统研发人员参考
高级驾驶辅助系统需同时理解驾驶员心理/生理状态与交通环境,但现有研究常忽视任务间的联合学习潜力。本文提出MMTL-UniAD,一个统一的多模态多任务学习框架,可同步识别驾驶员行为(如环顾四周、说话)、情绪(如焦虑、快乐)、车辆行为(如停车、转向)及交通场景(如拥堵、畅通)。核心挑战在于避免任务间负迁移。为此,框架引入两项关键组件:一是多轴区域注意力网络,用于提取全局上下文敏感特征;二是双分支多模态嵌入机制,从任务共享与特定特征中学习嵌入表示。前者通过多注意力机制提取任务相关特征,缓解无关特征引发的负迁移;后者采用双分支结构,自适应调整共享与特定参数,增强跨任务知识迁移并减少任务冲突。在AIDE数据集上进行系列消融实验,结果表明该框架在四项任务中均优于现有最优方法。代码已开源。
原文摘要 · Abstract (English)
Advanced driver assistance systems require a comprehensive understanding of the driver's mental/physical state and traffic context but existing works often neglect the potential benefits of joint learning between these tasks. This paper proposes MMTL-UniAD, a unified multi-modal multi-task learning framework that simultaneously recognizes driver behavior (e.g., looking around, talking), driver emotion (e.g., anxiety, happiness), vehicle behavior (e.g., parking, turning), and traffic context (e.g., traffic jam, traffic smooth). A key challenge is avoiding negative transfer between tasks, which can impair learning performance. To address this, we introduce two key components into the framework: one is the multi-axis region attention network to extract global context-sensitive features, and the other is the dual-branch multimodal embedding to learn multimodal embeddings from both task-shared and task-specific features. The former uses a multi-attention mechanism to extract task-relevant features, mitigating negative transfer caused by task-unrelated features. The latter employs a dual-branch structure to adaptively adjust task-shared and task-specific parameters, enhancing cross-task knowledge transfer while reducing task conflicts. We assess MMTL-UniAD on the AIDE dataset, using a series of ablation studies, and show that it outperforms state-of-the-art methods across all four tasks. The code is available on https://github.com/Wenzhuo-Liu/MMTL-UniAD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。