arXiv:2506.18084cs.CV2025-06被引 4

提出轻量级多模态多任务框架,实时提升智能驾驶中的多重感知能力。

TEM^3-Learning: Time-Efficient Multimodal Multi-Task Learning for Advanced Assistive Driving

  • 分两阶段设计:时序空间特征提取+门控多模态融合
  • 在AIDE数据集上四任务全达最优,参数<600万,推理速度142.32帧/秒
  • 适合需要低延迟、高精度的车载系统部署

多任务学习可通过共享表征挖掘任务间关联,推动辅助驾驶发展。然而,现有方法存在两大瓶颈:单一模态限制全面场景理解,低效架构阻碍实时部署。本文提出TEM^3-Learning(时间高效多模态多任务学习)框架,通过两阶段架构联合优化驾驶员情绪识别、行为识别、交通环境识别与车辆行为识别。第一部分MTS-Mamba基于Mamba的多视角时序-空间特征提取子网络,采用前向-后向时序扫描与全局-局部空间注意力,高效提取低成本时序空间特征。第二部分MGMI基于多任务学习的门控多模态特征集成器,利用任务特定多门控模块自适应强化各任务相关模态特征,有效缓解多任务学习中的负迁移问题。在AIDE数据集上的评估显示,所提模型在四项任务上均达到当前最优性能,保持轻量化架构(参数少于600万),实现142.32 FPS的推理速度。严格的消融实验进一步验证了框架有效性及各模块独立贡献。代码已公开于https://github.com/Wenzhuo-Liu/TEM3-Learning。

原文摘要 · Abstract (English)

Multi-task learning (MTL) can advance assistive driving by exploring inter-task correlations through shared representations. However, existing methods face two critical limitations: single-modality constraints limiting comprehensive scene understanding and inefficient architectures impeding real-time deployment. This paper proposes TEM^3-Learning (Time-Efficient Multimodal Multi-task Learning), a novel framework that jointly optimizes driver emotion recognition, driver behavior recognition, traffic context recognition, and vehicle behavior recognition through a two-stage architecture. The first component, the mamba-based multi-view temporal-spatial feature extraction subnetwork (MTS-Mamba), introduces a forward-backward temporal scanning mechanism and global-local spatial attention to efficiently extract low-cost temporal-spatial features from multi-view sequential images. The second component, the MTL-based gated multimodal feature integrator (MGMI), employs task-specific multi-gating modules to adaptively highlight the most relevant modality features for each task, effectively alleviating the negative transfer problem in MTL. Evaluation on the AIDE dataset, our proposed model achieves state-of-the-art accuracy across all four tasks, maintaining a lightweight architecture with fewer than 6 million parameters and delivering an impressive 142.32 FPS inference speed. Rigorous ablation studies further validate the effectiveness of the proposed framework and the independent contributions of each module. The code is available on https://github.com/Wenzhuo-Liu/TEM3-Learning.

多任务学习智能驾驶轻量化多模态

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。