/Clustering_friendly_representation_learning

Primary LanguagePythonOtherNOASSERTION

Clustering-friendly Representation Learning via Instance Discrimination and Feature Decorrelation

This repository contains the Pytorch implementation of our paper(IDFD). The implementation can reproduce the main result on CIFAR10.

If you found our work useful in your work, please cite:

@article{tao2021idfd,
title = {Clustering-friendly Representation Learning via Instance Discrimination and Feature Decorrelation},
author = {Yaling Tao, Kentaro Takagi, Kouta Nakata},
year = {2021},
journal = {Proceedings of ICLR 2021},
url = {https://openreview.net/forum?id=e12NDM7wkEY}
}

How to train

Setup environment

Install packages as:

pip install -r requirements.txt

Training

To train IDFD on GPU-0, run main script as:

./main.py --gpus 0