arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2206.06293cs.CVcs.AI
收藏

学习域自适应概率型目标检测

Learning Domain Adaptive Object Detection with Probabilistic Teacher

Zhejiang University(浙江大学);Hikvision Research Institute(海康威视研究院);National University of Singapore(新加坡国立大学);Chongqing University(重庆大学);Hainan Institute of Zhejiang University(浙江大学海南研究院)

Meilin Chen, Weijie Chen, Shicai Yang, Jie Song, Xinchao Wang, Lei Zhang, Yunfeng Yan, Donglian Qi, Yueting Zhuang, Di Xie, Shiliang Pu

摘要:

无监督域自适应目标检测的自训练是一项具有挑战性的任务,其性能在很大程度上取决于伪盒的质量。尽管取得了令人鼓舞的结果,但之前的工作在很大程度上忽视了自我训练过程中伪框的不确定性。在本文中,我们提出了一个简单而有效的框架,称为Probably Teacher(PT),其目的是从逐渐发展的教师中捕获未标记目标数据的不确定性,并以互利的方式指导学生的学习。具体来说,我们建议利用不确定性指导的一致性训练来促进分类适应和本地化适应,而不是通过精心设计的置信度阈值来过滤伪框。此外,我们进行锚适应与本地化适应并行,因为锚可以被视为一个可学习的参数。与此框架一起,我们还提出了一种新的熵焦点损失(EFL),以进一步促进不确定性引导的自我训练。配备EFL后,PT的性能大大优于所有以前的基准,并达到了新的最先进水平。

英文摘要:

Self-training for unsupervised domain adaptive object detection is a challenging task, of which the performance depends heavily on the quality of pseudo boxes. Despite the promising results, prior works have largely overlooked the uncertainty of pseudo boxes during self-training. In this paper, we present a simple yet effective framework, termed as Probabilistic Teacher (PT), which aims to capture the uncertainty of unlabeled target data from a gradually evolving teacher and guides the learning of a student in a mutually beneficial manner. Specifically, we propose to leverage the uncertainty-guided consistency training to promote classification adaptation and localization adaptation, rather than filtering pseudo boxes via an elaborate confidence threshold. In addition, we conduct anchor adaptation in parallel with localization adaptation, since anchor can be regarded as a learnable parameter. Together with this framework, we also present a novel Entropy Focal Loss (EFL) to further facilitate the uncertainty-guided self-training. Equipped with EFL, PT outperforms all previous baselines by a large margin and achieve new state-of-the-arts.

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

↑