Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Contrastive Learning of Image Representations with Cross-Video Cycle-Consistency

About

Recent works have advanced the performance of self-supervised representation learning by a large margin. The core among these methods is intra-image invariance learning. Two different transformations of one image instance are considered as a positive sample pair, where various tasks are designed to learn invariant representations by comparing the pair. Analogically, for video data, representations of frames from the same video are trained to be closer than frames from other videos, i.e. intra-video invariance. However, cross-video relation has barely been explored for visual representation learning. Unlike intra-video invariance, ground-truth labels of cross-video relation is usually unavailable without human labors. In this paper, we propose a novel contrastive learning method which explores the cross-video relation by using cycle-consistency for general image representation learning. This allows to collect positive sample pairs across different video instances, which we hypothesize will lead to higher-level semantics. We validate our method by transferring our image representation to multiple downstream tasks including visual object tracking, image classification, and action recognition. We show significant improvement over state-of-the-art contrastive learning methods. Project page is available at https://happywu.github.io/cycle_contrast_video.

Haiping Wu, Xiaolong Wang• 2021

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU35.6
2888
Semantic segmentationADE20K
mIoU35.6
1024
Object DetectionCOCO (val)
mAP42.8
633
Action RecognitionUCF101 (test)--
307
Object DetectionLVIS (val)
mAP24.5
141
Object DetectionCOCO
mAP42.8
137
Action RecognitionUCF101
Top-1 Acc82.8
48
Image ClassificationGBCU (test)
Accuracy86.1
18
Semantic segmentationPASCAL (train)
mIoU69.2
11
Global-shape recognitionShape-Biased Stimuli modified ImageNet (val)
Accuracy Difference23.7
10
Showing 10 of 12 rows

Other info

Follow for update