跳到主要导航 跳到搜索 跳到主要内容

RepCo: Replenish sample views with better consistency for contrastive learning

  • Xi'an Jiaotong University

科研成果: 期刊稿件文章同行评审

2 引用 (Scopus)

摘要

Contrastive learning methods aim to learn shared representations by minimizing distances between positive pairs, and maximizing distances between negative pairs in the embedding space. To achieve better performance of contrastive learning, one of the key problems is to design appropriate sample pairs. In most previous works, random cropping on the input image is utilized to obtain two views as positive pairs. However, such strategies lead to suboptimal performance since the sampled crops may have inconsistent semantic information, which consequently degrades the quality of contrastive views. To address this limitation, we explore to replenish sample views with better consistency of the image and propose a novel self-supervised learning (SSL) framework RepCo. Instead of searching for semantically consistent patches between two different views, we select patches on the same image as the replenishment of positive/negative pairs, encourage patches that are similar but come from different positions as positive pairs, and force patches that are dissimilar but come from adjacent positions to have different representations, i.e. construct negative pairs to enrich the learned representations. Our method effectively generates high-quality contrastive views, explores the untapped semantic consistency on images, and provides more informative representations for downstream tasks. Experiments on adequate downstream tasks have shown that, our approach achieves +2.1 AP50 (COCO pre-trained) and +1.6 AP50 (ImageNet pre-trained) gains on Pascal VOC object detection, +2.3 mIoU gains on Cityscapes semantic segmentation, respectively.

源语言英语
页(从-至)171-179
页数9
期刊Neural Networks
168
DOI
出版状态已出版 - 11月 2023

学术指纹

探究 'RepCo: Replenish sample views with better consistency for contrastive learning' 的科研主题。它们共同构成独一无二的指纹。

引用此