Transitive Invariance for Self-Supervised Visual Representation Learning

By X. Wang, Kaiming He, Abhinav Kumar Gupta. 2017

183 citations recorded by the source corpus.

Learning visual representations with self-supervised learning has become popular in computer vision. The idea is to design auxiliary tasks where labels are free to obtain. Most of these tasks end up providing data to learn specific kinds of invariance useful for recognition. In this paper, we propose to exploit different self-supervised approaches to learn representations invariant to (i) inter-instance variations (two objects in the same class should have similar features) and (ii) intra-instance variations (viewpoint, pose, deformations, illumination, etc.). Instead of combining two approaches with multi-task learning, we argue to organize and reason the data with multiple variations. Specifically, we propose to generate a graph with millions of objects mined from hundreds of thousands of videos. The objects are connected by two types of edges which correspond to two types of invariance: “different instances but a similar viewpoint and category” and “different viewpoints of the same instance”. By applying simple transitivity on the graph with these edges, we can obtain pairs of images exhibiting richer visual invariance. We use this data to train a Triplet-Siamese network with VGG16 as the base architecture and apply the learned representations to different recognition tasks. For object detection, we achieve 63.2% mAP on PASCAL VOC 2007 using Fast R-CNN (compare to 67.3% with ImageNet pre-training). For the challenging COCO dataset, our method is surprisingly close (23.5%) to the ImageNet-supervised counterpart (24.4%) using the Faster R-CNN framework. We also show that our network can perform significantly better than the ImageNet network in the surface normal estimation task.

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. Dacheng Tao
  2. Trevor Darrell
  3. Abhinav Kumar Gupta
  4. Luc Van Gool
  5. Andrea Vedaldi
  6. Kaiming He
  7. Ishan Misra
  8. Cordelia Schmid

In Computer Vision and Pattern Recognition

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

In Representation & Self-Supervised Learning

  1. Ishan Misra
  2. Andrea Vedaldi
  3. Armand Joulin
  4. Yingli Tian
  5. Alexei A. Efros
  6. Abhinav Kumar Gupta
  7. Longlong Jing
  8. Andrew Zisserman

Citations in the corpus