Exploring Localization for Self-supervised Fine-grained Contrastive Learning

30 Jun 2021  ·  Di wu, Siyuan Li, Zelin Zang, Stan Z. Li ·

Self-supervised contrastive learning has demonstrated great potential in learning visual representations. Despite their success in various downstream tasks such as image classification and object detection, self-supervised pre-training for fine-grained scenarios is not fully explored. We point out that current contrastive methods are prone to memorizing background/foreground texture and therefore have a limitation in localizing the foreground object. Analysis suggests that learning to extract discriminative texture information and localization are equally crucial for fine-grained self-supervised pre-training. Based on our findings, we introduce cross-view saliency alignment (CVSA), a contrastive learning framework that first crops and swaps saliency regions of images as a novel view generation and then guides the model to localize on foreground objects via a cross-view alignment loss. Extensive experiments on both small- and large-scale fine-grained classification benchmarks show that CVSA significantly improves the learned representation.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Fine-Grained Image Classification CUB-200-2011 BYOL+CVSA (ResNet-50) Accuracy 77.1 # 24
Fine-Grained Image Classification FGVC Aircraft BYOL+CVSA (ResNet-50) Accuracy 87.27 # 48
Fine-Grained Image Classification NABirds BYOL+CVSA (ResNet-50) Accuracy 79.64% # 21
Fine-Grained Image Classification Stanford Cars BYOL+CVSA (ResNet-50) Accuracy 89.76% # 69

Methods