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 条
  • [1] Contrastive Learning for Unsupervised Video Highlight Detection
    Badamdorj, Taivanbat
    Rochan, Mrigank
    Wang, Yang
    Cheng, Li
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, : 14022 - 14032
  • [2] Weakly-Supervised Contrastive Learning for Unsupervised Object Discovery
    Lv, Yunqiu
    Zhang, Jing
    Barnes, Nick
    Dai, Yuchao
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2024, 33 : 2689 - 2702
  • [3] Unsupervised Outlier Detection Using Memory and Contrastive Learning
    Huyan, Ning
    Quan, Dou
    Zhang, Xiangrong
    Liang, Xuefeng
    Chanussot, Jocelyn
    Jiao, Licheng
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 : 6440 - 6454
  • [4] DeepfakeUCL: Deepfake Detection via Unsupervised Contrastive Learning
    Fung, Sheldon
    Lu, Xuequan
    Zhang, Chao
    Li, Chang-Tsun
    [J]. 2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [5] Unsupervised contrastive learning based transformer for lung nodule detection
    Niu, Chuang
    Wang, Ge
    [J]. PHYSICS IN MEDICINE AND BIOLOGY, 2022, 67 (20):
  • [6] Object Discovery via Contrastive Learning for Weakly Supervised Object Detection
    Seo, Jinhwan
    Bae, Wonho
    Sutherland, Danica J.
    Noh, Junhyug
    Kim, Daijin
    [J]. COMPUTER VISION, ECCV 2022, PT XXXI, 2022, 13691 : 312 - 329
  • [7] Contrastive fusion representation learning for foreground object detection
    Wang, Pei
    Wu, Junsheng
    Fang, Aiqing
    Zhu, Zhixiang
    Wang, Chenwu
    Mu, Pengyuan
    [J]. ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 123
  • [8] Improving the Adversarial Robustness of Object Detection with Contrastive Learning
    Zeng, Weiwei
    Gao, Song
    Zhou, Wei
    Dong, Yunyun
    Wang, Ruxin
    [J]. PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT IX, 2024, 14433 : 29 - 40
  • [9] Unsupervised anomaly detection by densely contrastive learning for time series data
    Zhu, Wei
    Li, Weijian
    Dorsey, E. Ray
    Luo, Jiebo
    [J]. NEURAL NETWORKS, 2023, 168 : 450 - 458
  • [10] UBoCo : Unsupervised Boundary Contrastive Learning for Generic Event Boundary Detection
    Kang, Hyolim
    Kim, Jinwoo
    Kim, Taehyun
    Kim, Seon Joo
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 20041 - 20050