用大模型生成更准目标域标签,提升域适应目标检测效果
Large Self-Supervised Models Bridge the Gap in Domain Adaptive Object Detection
- 用冻结的DINOv2做标签生成器,替代传统Mean Teacher
- 在多个数据集上达到最新最优性能,提升显著
- 适合研究域适应、自监督学习的学者参考
当前最先进的领域自适应目标检测(DAOD)方法采用均值教师自标注机制:教师模型由学生模型的指数移动平均直接生成,在目标域生成标签并反馈优化双模型,形成正向循环。但该机制存在耦合过强、鲁棒性差的问题——仅基于源域训练的学生模型难以生成准确的目标域标签,无法启动有效反馈。我们提出使用大规模预训练视觉基础模型(如DINOv2)作为标签生成器,因其具备更强的泛化能力。本文引入DINO Teacher,包含两部分:首先,仅在源数据上训练一个基于冻结DINOv2骨干的标签生成器,其生成的标签比均值教师更准确;其次,将学生模型的源/目标图像块特征与DINO编码器对齐,使表示趋近于通用的DINO表示。在多个DAOD数据集上实现最新最优性能。代码已开源。
原文摘要 · Abstract (English)
The current state-of-the-art methods in domain adaptive object detection (DAOD) use Mean Teacher self-labelling, where a teacher model, directly derived as an exponential moving average of the student model, is used to generate labels on the target domain which are then used to improve both models in a positive loop. This couples learning and generating labels on the target domain, and other recent works also leverage the generated labels to add additional domain alignment losses. We believe this coupling is brittle and excessively constrained: there is no guarantee that a student trained only on source data can generate accurate target domain labels and initiate the positive feedback loop, and much better target domain labels can likely be generated by using a large pretrained network that has been exposed to much more data. Vision foundational models are exactly such models, and they have shown impressive task generalization capabilities even when frozen. We want to leverage these models for DAOD and introduce DINO Teacher, which consists of two components. First, we train a new labeller on source data only using a large frozen DINOv2 backbone and show it generates more accurate labels than Mean Teacher. Next, we align the student's source and target image patch features with those from a DINO encoder, driving source and target representations closer to the generalizable DINO representation. We obtain state-of-the-art performance on multiple DAOD datasets. Code available at https://github.com/TRAILab/DINO_Teacher
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。