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

基于平衡域分类器的域不变候选区域目标检测

Domain-Invariant Proposals based on a Balanced Domain Classifier for Object Detection

Zhize Wu, Xiaofeng Wang, Tong Xu, Xuebin Yang, Le Zou, Lixiang Xu, Thomas Weise

AI总结:

针对域偏移导致目标检测性能下降的问题,在Faster R-CNN的区域候选网络中引入对抗学习域分类器,结合平衡域分类器与学习率调整策略,有效生成域不变候选区域,提升了跨域检测的鲁棒性。

摘要:

图像中的目标识别旨在自动寻找感兴趣的目标,并返回其类别和位置信息。得益于卷积神经网络(CNNs)和生成对抗网络等深度学习研究,该领域的性能得到了显著提升,尤其是当训练和测试数据来自相似分布时。然而,分布不匹配,即域偏移,会导致性能显著下降。在本文中,我们通过对抗训练学习域分类器来构建域不变检测器。基于先前对齐图像级和实例级特征的工作,我们通过在Faster R-CNN的区域级别引入域自适应组件来进一步缓解域偏移。我们使用对抗学习在区域候选网络(RPN)中嵌入域分类网络。RPN现在可以通过有效地对齐不同域之间的特征,在不同域中生成准确的区域候选。为了缓解对抗学习过程中不稳定的收敛问题,我们引入了平衡域分类器以及网络学习率调整策略。我们使用四个标准数据集进行了全面实验。结果证明了我们的目标检测方法在域偏移场景下的有效性和鲁棒性。

英文摘要:

Object recognition from images means to automatically find object(s) of interest and to return their category and location information. Benefiting from research on deep learning, like convolutional neural networks~(CNNs) and generative adversarial networks, the performance in this field has been improved significantly, especially when training and test data are drawn from similar distributions. However, mismatching distributions, i.e., domain shifts, lead to a significant performance drop. In this paper, we build domain-invariant detectors by learning domain classifiers via adversarial training. Based on the previous works that align image and instance level features, we mitigate the domain shift further by introducing a domain adaptation component at the region level within Faster \mbox{R-CNN}. We embed a domain classification network in the region proposal network~(RPN) using adversarial learning. The RPN can now generate accurate region proposals in different domains by effectively aligning the features between them. To mitigate the unstable convergence during the adversarial learning, we introduce a balanced domain classifier as well as a network learning rate adjustment strategy. We conduct comprehensive experiments using four standard datasets. The results demonstrate the effectiveness and robustness of our object detection approach in domain shift scenarios.

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

↑