Heterogeneous Contrastive Learning: Encoding Spatial Information for Compact Visual Representations
Xinyue Huo,Lingxi Xie,Longhui Wei,Xiaopeng Zhang,Xin Chen,Hao Li,Zijie Yang,Wengang Zhou,Houqiang Li,Qi Tian
DOI: https://doi.org/10.1109/tmm.2021.3115335
IF: 7.3
2022-01-01
IEEE Transactions on Multimedia
Abstract:Unsupervised pretraining is of great significance for visual representation. Especially, contrastive learning has achieved great success recently, but existing approaches mostly ignored spatial information which is often crucial for visual representation. Strong semantic embedding has an inherent advantage for classification, but dense prediction tasks require more spatial and low-level representation. This paper presents heterogeneous contrastive learning (HCL), an effective approach that adds spatial information to the encoding stage to alleviate the learning inconsistency between the contrastive objective and strong data augmentation operations. We demonstrate the effectiveness of HCL by showing that (i) it achieves higher accuracy in instance discrimination, (ii) it surpasses existing pre-training methods in a series of downstream tasks (iii) and it shrinks the pre-training costs by half for almost 800 GPU-hours. More importantly, we show that our approach achieves higher efficiency in visual representations, and thus delivers a key message to inspire the future research of self-supervised visual representation learning.
computer science, information systems,telecommunications, software engineering