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

MTTrans:使用Mean-Teacher Transformer的跨域目标检测

MTTrans: Cross-Domain Object Detection with Mean-Teacher Transformer

Beihang University(北京航空航天大学);Peking University(北京大学);Panasonic Holdings Corporation(松下控股公司);University of California, Berkeley(加州大学伯克利分校)

Jinze Yu, Jiaming Liu, Xiaobao Wei, Haoyi Zhou, Yohei Nakata, Denis Gudovskiy, Tomoyuki Okuno, Jianxin Li, Kurt Keutzer, Shanghang Zhang

摘要:

最近,检测Transformer(DETR),端到端的目标检测管道,取得了良好的性能。然而,它需要大规模的标记数据,并遭受域转移,特别是当没有标记的数据是在目标域。针对这一问题,提出了一种基于均值教师框架MTTrans的端到端跨域检测算法Transformer,该算法能够充分利用目标检测训练中未标记的目标域数据,并通过伪标签在域间传递知识。我们进一步提出了全面的多层次特征对齐,以改善伪标签的平均教师框架,利用跨尺度的自我注意力机制变形DETR。图像和对象特征在局部、全局和实例级别上通过基于域查询的特征对齐(DQFA)、基于双层图的原型对齐(BGPA)和标记图像特征对齐(TIFA)进行对齐。另一方面,未标记的目标域数据被伪标记并且可用于由均值教师框架进行的目标检测训练,可以导致更好的特征提取和对齐。因此,可以基于Transformers的体系结构迭代地和相互地优化平均教师框架和综合的多级特征对齐。大量的实验表明,该方法在三种领域自适应场景下均取得了较好的性能,尤其是在Sim 10 k到Cityscapes场景下,自适应效果从52.6 mAP显著提高到57.9 mAP.代码将被释放。

英文摘要:

Recently, DEtection TRansformer (DETR), an end-to-end object detection pipeline, has achieved promising performance. However, it requires large-scale labeled data and suffers from domain shift, especially when no labeled data is available in the target domain. To solve this problem, we propose an end-to-end cross-domain detection Transformer based on the mean teacher framework, MTTrans, which can fully exploit unlabeled target domain data in object detection training and transfer knowledge between domains via pseudo labels. We further propose the comprehensive multi-level feature alignment to improve the pseudo labels generated by the mean teacher framework taking advantage of the cross-scale self-attention mechanism in Deformable DETR. Image and object features are aligned at the local, global, and instance levels with domain query-based feature alignment (DQFA), bi-level graph-based prototype alignment (BGPA), and token-wise image feature alignment (TIFA). On the other hand, the unlabeled target domain data pseudo-labeled and available for the object detection training by the mean teacher framework can lead to better feature extraction and alignment. Thus, the mean teacher framework and the comprehensive multi-level feature alignment can be optimized iteratively and mutually based on the architecture of Transformers. Extensive experiments demonstrate that our proposed method achieves state-of-the-art performance in three domain adaptation scenarios, especially the result of Sim10k to Cityscapes scenario is remarkably improved from 52.6 mAP to 57.9 mAP. Code will be released.

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

↑