/FeatureDecoupling

Self-Supervised Representation Learning by Rotation Feature Decoupling

Primary LanguagePythonMIT LicenseMIT

Self-Supervised Representation Learning by Rotation Feature Decoupling

Introduction

The current code implements on pytorch the following CVPR 2019 paper:
Title: "Self-Supervised Representation Learning by Rotation Feature Decoupling"
Authors: Zeyu Feng, Chang Xu, Dacheng Tao
Institution: UBTECH Sydney AI Centre, School of Computer Science, FEIT, University of Sydney, Australia
Code: https://github.com/philiptheother/FeatureDecoupling
Link: pdf and supp

Abstract:
We introduce a self-supervised learning method that focuses on beneficial properties of representation and their abilities in generalizing to real-world tasks. The method incorporates rotation invariance into the feature learning framework, one of many good and well-studied properties of visual representation, which is rarely appreciated or exploited by previous deep convolutional neural network based self-supervised representation learning methods. Specifically, our model learns a split representation that contains both rotation related and unrelated parts. We train neural networks by jointly predicting image rotations and discriminating individual instances. In particular, our model decouples the rotation discrimination from instance discrimination, which allows us to improve the rotation prediction by mitigating the influence of rotation label noise, as well as discriminate instances without regard to image rotations. The resulting feature has a better generalization ability for more various tasks. Experimental results show that our model outperforms current state-of-the-art methods on standard self-supervised feature learning benchmarks.

Illustration

Citing FeatureDecoupling

If you find the code useful in your research, please consider citing our CVPR 2019 paper:

@InProceedings{Feng_2019_CVPR,
author = {Feng, Zeyu and Xu, Chang and Tao, Dacheng},
title = {Self-Supervised Representation Learning by Rotation Feature Decoupling},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2019}
}

License

Please refer to License files for details.

Notice

Experiments

  • In order to train a FeatureDecoupling model in an unsupervised way with AlexNet-like architecture on the ImageNet training images and then evaluate linear classifiers (for ImageNet and Places205) as well as non-linear classifiers (for ImageNet) on top of the learned features please visit the pytorch_feature_decoupling folder.

  • For PASCAL VOC 2007 detection task please visit the caffe_voc_detection folder.

Download the already trained FeatureDecoupling model

  • In order to download the FeatureDecoupling model (with AlexNet architecture) trained on the ImageNet training set, go to: ImageNet_Decoupling_AlexNet. Note that:

    1. The model is saved in pytorch format.
    2. It expects RGB images that their pixel values are normalized with the following mean RGB values mean_rgb = [0.485, 0.456, 0.406] and std RGB values std_rgb = [0.229, 0.224, 0.225]. Prior to normalization the range of the image values must be [0.0, 1.0].
  • In order to download the FeatureDecoupling model (with AlexNet architecture) trained on the ImageNet training set and convered in caffe format for PASCAL VOC 2007 classification and detection, go to: ImageNet_Decoupling_AlexNet_caffe_cls. Note that:

    1. The model is saved in caffe format.
    2. It expects RGB images that their pixel values are normalized with the following mean RGB values mean_rgb = [0.485, 0.456, 0.406] and std RGB values std_rgb = [0.229, 0.224, 0.225]. Prior to normalization the range of the image values must be [0.0, 1.0].
    3. The weights of the model are rescaled with the approach of Krähenbühl et al, ICLR 2016.

To do

  • Pytorch-Caffe converter
  • PASCAL VOC experiments
  • Pre-trained models