汽车工程 ›› 2023, Vol. 45 ›› Issue (11): 2082-2091.doi: 10.19562/j.chinasae.qcgc.2023.11.009

所属专题: 智能网联汽车技术专题-感知&HMI&测评2023年

• • 上一篇    下一篇

基于特征和标签联合分布匹配的智能驾驶跨域自适应目标检测

刘正发1,吴亚1,刘佩根1,顾荣琦2,陈广1()   

  1. 1.同济大学汽车学院,上海 201804
    2.上海西井科技股份有限公司,上海 200052
  • 收稿日期:2023-09-07 修回日期:2023-10-22 出版日期:2023-11-25 发布日期:2023-11-27
  • 通讯作者: 陈广 E-mail:guangchen@tongji.edu.cn
  • 基金资助:
    国家重点研发计划(2021YFB2501104)

Cross-Domain Object Detection for Intelligent Driving Based on Joint Distribution Matching of Features and Labels

Zhengfa Liu1,Ya Wu1,Peigen Liu1,Rongqi Gu2,Guang Chen1()   

  1. 1.School of Automotive Studies,Tongji University,Shanghai 201804
    2.Shanghai Westwell Technology Company Limited,Shanghai 200052
  • Received:2023-09-07 Revised:2023-10-22 Online:2023-11-25 Published:2023-11-27
  • Contact: Guang Chen E-mail:guangchen@tongji.edu.cn

摘要:

现有的跨域自适应目标检测器主要通过特征分布匹配学习域不变特征来减小域偏移。然而,它们忽略了现实世界中因目标检测对象组合不同和类不平衡等引起的标签分布偏移问题,导致这些方法泛化性较差。为解决这一问题,本文提出一种基于特征和标签联合分布匹配的域自适应目标检测算法,以同时在特征和标签级别上显式对齐域分布。首先,提出一个图像级分类嵌入模块,通过对比学习增强全局特征可迁移性和可判别性。然后,提出类级分布对齐模块,通过多级特征对齐实现域间多模态结构对齐。最后,提出增强一致性正则化模块,通过区域一致性正则化实现跨域标签分布对齐。在多个数据集上的实验表明,所提出的域对齐算法能有效地促进跨域数据迁移前后语义一致性,为智能汽车跨视觉域应用提供了一种有效的解决方案。

关键词: 智能驾驶, 智能汽车, 目标检测, 域自适应

Abstract:

Current cross-domain adaptive object detection methods primarily focus on reducing domain shift by learning domain-invariant features through feature distribution alignment. However, they often overlook label distribution shift issues caused by variations in object combinations and class imbalances in real-world scenarios, resulting in poor generalization performance. To address this, this paper proposes a novel domain adaptive object detection algorithm that simultaneously aligns domain distributions at both feature and label levels. Firstly, an image-level classification embedding module is introduced to enhance the transferability and discriminability of global features through contrastive learning. Next, a class-level distribution alignment module is presented to achieve inter-domain multimodal structure alignment through multi-level feature alignment. Finally, an enhanced consistency regularization module is proposed to achieve cross-domain label distribution alignment through region-based consistency regularization. Experimental results across multiple datasets demonstrate that the proposed domain alignment algorithm effectively improves semantic consistency both before and after cross-domain data adaptation. This provides a valuable solution for the effective deployment of autonomous vehicles in cross-domain scenarios.

Key words: autonomous vehicles, autonomous driving, object detection, domain adaptation