arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.07780cs.CV

基于多粒度置信度对齐均值教师的跨域目标检测

Cross Domain Object Detection via Multi-Granularity Confidence Alignment based Mean Teacher

National University of Defense Technology (NUDT)(国防科技大学);College of System Engineering, National University of Defense Technology (NUDT)(国防科技大学系统工程学院);College of Electronic Science and Technology, NUDT(国防科技大学电子科学学院);College of Meteorology and Oceanography, NUDT(国防科技大学气象海洋学院)

Jiangming Chen, Li Liu, Wanxia Deng, Zhen Liu, Yu Liu, Yingmei Wei, Yongxiang Liu

AI总结:

针对跨域目标检测中均值教师方法伪标签置信度错位导致性能不佳的问题,提出多粒度置信度对齐框架,从类别、实例、图像三层优化伪监督质量,提升跨域检测效果。

摘要:

跨域目标检测通过从有标注源域迁移知识,为无标注目标域学习一个目标检测器。基于均值教师(Mean Teacher)的方法已取得可观成果,但作为互学习瓶颈的伪标签仍有待进一步探索。本研究发现,预测结果的置信度错位——包括类别层面的过度自信、实例层面的任务置信度不一致以及图像层面的置信度失焦——会导致训练过程中引入带噪声的伪标签,进而造成目标域上的次优性能。为解决这一问题,我们提出一种名为多粒度置信度对齐均值教师(MGCAMT)的新型通用跨域目标检测框架,该框架同时缓解类别、实例和图像三个层面的置信度错位,以获取高质量伪监督,优化师生学习过程。具体而言,为在类别层面实现置信度与准确率的对齐,我们提出分类置信度对齐(CCA)模块,基于证据深度学习(EDL)建模类别不确定性,并通过一种感知不确定性的选择策略过滤类别错误的标签。此外,为缓解实例层面分类与定位之间的错位,我们设计了任务置信度对齐(TCA)模块,增强两个任务分支之间的交互,使每个分类特征能够自适应地找到用于回归的最优特征。最后,我们提出图像聚焦置信度对齐(FCA)模块,采用另一种伪标签学习方式:即直接使用均值教师网络的原始输出进行监督学习,无需标签分配,从而聚焦于目标图像的整体信息。从协同学习的角度来看,这三个模块能够相互促进。

英文摘要:

Cross domain object detection learns an object detector for an unlabeled target domain by transferring knowledge from an annotated source domain. Promising results have been achieved via Mean Teacher, however, pseudo labeling which is the bottleneck of mutual learning remains to be further explored. In this study, we find that confidence misalignment of the predictions, including category-level overconfidence, instance-level task confidence inconsistency, and image-level confidence misfocusing, leading to the injection of noisy pseudo label in the training process, will bring suboptimal performance on the target domain. To tackle this issue, we present a novel general framework termed Multi-Granularity Confidence Alignment Mean Teacher (MGCAMT) for cross domain object detection, which alleviates confidence misalignment across category-, instance-, and image-levels simultaneously to obtain high quality pseudo supervision for better teacher-student learning. Specifically, to align confidence with accuracy at category level, we propose Classification Confidence Alignment (CCA) to model category uncertainty based on Evidential Deep Learning (EDL) and filter out the category incorrect labels via an uncertainty-aware selection strategy. Furthermore, to mitigate the instance-level misalignment between classification and localization, we design Task Confidence Alignment (TCA) to enhance the interaction between the two task branches and allow each classification feature to adaptively locate the optimal feature for the regression. Finally, we develop imagery Focusing Confidence Alignment (FCA) adopting another way of pseudo label learning, i.e., we use the original outputs from the Mean Teacher network for supervised learning without label assignment to concentrate on holistic information in the target image. These three procedures benefit from each other from a cooperative learning perspective.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑