Distilling Localization for Self-Supervised Representation Learning

By Nanxuan Zhao, Zhirong Wu, Rynson W. H. Lau, Stephen Lin. 2020

59 citations recorded by the source corpus.

Recent progress in contrastive learning has revolutionized unsupervised representation learning. Concretely, multiple views (augmentations) from the same image are encouraged to map to close embeddings, while views from different images are pulled apart.In this paper, through visualizing and diagnosing classification errors, we observe that current contrastive models are ineffective at localizing the foreground object, limiting their ability to extract discriminative high-level features. This is due to the fact that view generation process considers pixels in an image uniformly.To address this problem, we propose a data-driven approach for learning invariance to backgrounds. It first estimates foreground saliency in images and then creates augmentations by copy-and-pasting the foreground onto a variety of back-grounds. The learning still follows an instance discrimination approach, so that the representation is trained to disregard background content and focus on the foreground. We study a variety of saliency estimation methods, and find that most methods lead to improvements for contrastive learning. With this approach, significant performance is achieved for self-supervised learning on ImageNet classification, and also for object detection on PASCAL VOC and MSCOCO.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Representation & Self-Supervised Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Neil Houlsby
  2. Xiaohua Zhai
  3. Quoc V. Le
  4. Cordelia Schmid
  5. Dacheng Tao
  6. Jianfeng Gao
  7. Yoshua Bengio
  8. Lucas Beyer

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Trevor Darrell
  4. Jitendra Malik
  5. Xiaogang Wang
  6. Ming-Hsuan Yang
  7. Cordelia Schmid
  8. Andrew Zisserman

In Representation & Self-Supervised Learning

  1. Neil Houlsby
  2. Phillip Isola
  3. Xiaohua Zhai
  4. Yonglong Tian
  5. Lucas Beyer
  6. Ishan Misra
  7. Alexander Kolesnikov
  8. Michael Tschannen

Citations in the corpus