DetCo: Unsupervised Contrastive Learning for Object Detection

被引:168
|
作者
Xie, Enze [1 ]
Ding, Jian [3 ]
Wang, Wenhai [4 ]
Zhan, Xiaohang [5 ]
Xu, Hang [2 ]
Sun, Peize [1 ]
Li, Zhenguo [2 ]
Luo, Ping [1 ]
机构
[1] Univ Hong Kong, Hong Kong, Peoples R China
[2] Huawei Noahs Ark Lab, Hong Kong, Peoples R China
[3] Wuhan Univ, Wuhan, Peoples R China
[4] Nanjing Univ, Nanjing, Peoples R China
[5] Chinese Univ Hong Kong, Hong Kong, Peoples R China
关键词
D O I
10.1109/ICCV48922.2021.00828
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We present DetCo, a simple yet effective self-supervised approach for object detection. Unsupervised pre-training methods have been recently designed for object detection, but they are usually deficient in image classification, or the opposite. Unlike them, DetCo transfers well on downstream instance-level dense prediction tasks, while maintaining competitive image-level classification accuracy. The advantages are derived from (1) multi-level supervision to intermediate representations, (2) contrastive learning between global image and local patches. These two designs facilitate discriminative and consistent global and local representation at each level of feature pyramid, improving detection and classification, simultaneously. Extensive experiments on VOC, COCO, Cityscapes, and ImageNet demonstrate that DetCo not only outperforms recent methods on a series of 2D and 3D instance-level detection tasks, but also competitive on image classification. For example, on ImageNet classification, DetCo is 6.9% and 5.0% top-1 accuracy better than InsLoc and DenseCL, which are two contemporary works designed for object detection. Moreover, on COCO detection, DetCo is 6.9 AP better than SwAV with Mask R-CNN C4. Notably, DetCo largely boosts up Sparse R-CNN, a recent strong detector, from 45.0 AP to 46.5 AP (+1.5 AP), establishing a new SOTA on COCO.
引用
收藏
页码:8372 / 8381
页数:10
相关论文
共 50 条
  • [41] A Theoretical Analysis of Contrastive Unsupervised Representation Learning
    Arora, Sanjeev
    Khandeparkar, Hrishikesh
    Khodak, Mikhail
    Plevrakis, Orestis
    Saunshi, Nikunj
    [J]. INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 97, 2019, 97
  • [42] Contrastive Learning for Unsupervised Radar Place Recognition
    Gadd, Matthew
    De Martini, Daniele
    Newman, Paul
    [J]. 2021 20TH INTERNATIONAL CONFERENCE ON ADVANCED ROBOTICS (ICAR), 2021, : 344 - 349
  • [43] Contrastive Intrinsic Control for Unsupervised Reinforcement Learning
    Laskin, Michael
    Liu, Hao
    Peng, Xue Bin
    Yarats, Denis
    Rajeswaran, Aravind
    Abbeel, Pieter
    [J]. ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35 (NEURIPS 2022), 2022,
  • [44] Relative Contrastive Loss for Unsupervised Representation Learning
    Tang, Shixiang
    Zhu, Feng
    Bai, Lei
    Zhao, Rui
    Ouyang, Wanli
    [J]. COMPUTER VISION - ECCV 2022, PT XXVII, 2022, 13687 : 1 - 18
  • [45] Extending Contrastive Learning to Unsupervised Redundancy Identification
    Ju, Jeongwoo
    Jung, Heechul
    Kim, Junmo
    [J]. APPLIED SCIENCES-BASEL, 2022, 12 (04):
  • [46] Unsupervised Anomaly Detection on Attributed Networks With Graph Contrastive Learning for Consumer Electronics Security
    Xu, Bo
    Wang, Jinpeng
    Zhao, Zhehuan
    Lin, Hongfei
    Xia, Feng
    [J]. IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2024, 70 (01) : 4062 - 4072
  • [47] Learning Informative Health Indicators Through Unsupervised Contrastive Learning
    Rombach, Katharina
    Michau, Gabriel
    Burzle, Wilfried
    Koller, Stefan
    Fink, Olga
    [J]. IEEE TRANSACTIONS ON RELIABILITY, 2024,
  • [48] Few-Shot Object Detection via Transfer Learning and Contrastive Reweighting
    Wu, Zhen
    Li, Haowei
    Zhang, Dongyu
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2023, PT VII, 2023, 14260 : 78 - 87
  • [49] Achieving domain generalization for underwater object detection by domain mixup and contrastive learning
    Chen, Yang
    Song, Pinhao
    Liu, Hong
    Dai, Linhui
    Zhang, Xiaochuan
    Ding, Runwei
    Li, Shengquan
    [J]. NEUROCOMPUTING, 2023, 528 : 20 - 34
  • [50] Contrastive Learning-based Robust Object Detection under Smoky Conditions
    Wu, Wei
    Chang, Hao
    Zheng, Yonghua
    Li, Zhu
    Chen, Zhiwen
    Zhang, Ziheng
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2022, 2022, : 4294 - 4301