论文标题
共同开采:稀疏注释对象检测的自我监督学习
Co-mining: Self-Supervised Learning for Sparsely Annotated Object Detection
论文作者
论文摘要
对象探测器通常通过对完整实例注释的监督获得有希望的结果。但是,他们的性能远非稀疏的实例注释令人满意。大多数现有的用于稀疏注释的对象检测的方法可以重新体重损失硬性样本或将未标记的实例转换为被忽略的区域,以减少虚假负面因素的干扰。我们认为这些策略是不够的,因为它们最多可以减轻缺失注释造成的负面影响。在本文中,我们提出了一种简单但有效的机制,称为共进,用于稀疏注释的对象检测。在我们的共同开采中,暹罗网络的两个分支预测了彼此的伪标签集。为了增强多视图学习和更好的矿山未标记的实例,原始图像和相应的增强图像分别用作暹罗网络两个分支的输入。共同开采可以用作应用于大多数现代对象探测器的一般培训机制。实验是在MS Coco数据集上使用两个典型框架的三个不同稀疏注释设置的MS进行的:基于锚固的检测器视网膜和无锚定检测器FCO。实验结果表明,与不同的基准相比,我们与视网膜的共同开采提高了1.4%〜2.1%,并且在相同的稀疏注释设置下超过了现有方法。代码可在https://github.com/megvii-research/co-mining上找到。
Object detectors usually achieve promising results with the supervision of complete instance annotations. However, their performance is far from satisfactory with sparse instance annotations. Most existing methods for sparsely annotated object detection either re-weight the loss of hard negative samples or convert the unlabeled instances into ignored regions to reduce the interference of false negatives. We argue that these strategies are insufficient since they can at most alleviate the negative effect caused by missing annotations. In this paper, we propose a simple but effective mechanism, called Co-mining, for sparsely annotated object detection. In our Co-mining, two branches of a Siamese network predict the pseudo-label sets for each other. To enhance multi-view learning and better mine unlabeled instances, the original image and corresponding augmented image are used as the inputs of two branches of the Siamese network, respectively. Co-mining can serve as a general training mechanism applied to most of modern object detectors. Experiments are performed on MS COCO dataset with three different sparsely annotated settings using two typical frameworks: anchor-based detector RetinaNet and anchor-free detector FCOS. Experimental results show that our Co-mining with RetinaNet achieves 1.4%~2.1% improvements compared with different baselines and surpasses existing methods under the same sparsely annotated setting. Code is available at https://github.com/megvii-research/Co-mining.