自动驾驶环境感知多任务去耦-融合算法

作     者：廖存燚 郑毅 刘玮瑾 于欢 刘守印 LIAO Cunyi;ZHENG Yi;LIU Weijin;YU Huan;LIU Shouyin

作者机构：华中师范大学物理科学与技术学院武汉430079 武汉大学测绘学院武汉430079 

出 版 物：《计算机应用》 (journal of Computer Applications)

年 卷 期：2024年第44卷第2期

页      面：424-431页

学科分类：12[管理学] 1201[管理学-管理科学与工程(可授管理学、工学学位)] 081104[工学-模式识别与智能系统] 08[工学] 080202[工学-机械电子工程] 0835[工学-软件工程] 0802[工学-机械工程] 0811[工学-控制科学与工程] 0812[工学-计算机科学与技术（可授工学、理学学位）] 

基　　金：国家自然科学基金资助项目(62277027) 

主　　题：自动驾驶 环境感知 目标检测 实例分割 目标跟踪 多任务学习 

摘      要：自动驾驶车辆在行驶过程中,需要对行人和车辆同时完成目标检测、实例分割和目标跟踪三个任务。提出一种基于深度学习的环境感知模型同时对三个任务进行多任务学习。首先,通过卷积神经网络对连续帧图像提取时空特征;然后,通过注意力机制对时空特征进行去耦再融合,充分利用任务间的相关性,实现不同任务对时空特征的差异化选择;最后,为平衡不同任务间的学习速率,使用动态加权平均的方式对模型进行训练。在KITTI数据集上的实验结果表明,所提模型在目标检测方面,比CenterTrack模型F1得分提高了0.6个百分点;在目标跟踪方面,比TraDeS(Track to Detect and Segment)模型多目标跟踪精度(MOTA)提高了0.7个百分点;在实例分割方面,比SOLOv2(Segmenting Objects by LOcations version 2)模型AP_(50)和AP_(75)分别提高了7.4和3.9个百分点。