This repo contains the code and data of the following paper accepted by CVPR 2021
Partially View-aligned Representation Learning with Noise-robust Contrastive Loss
pytorch==1.5.0
numpy>=1.18.2
scikit-learn>=0.22.2
munkres>=1.1.2
logging>=0.5.1.2
The hyper-parameters, the training options (including the ratiao of positive to negative, aligned proportions and switch time) are defined in the args. part in run.py.
The Scene-15 and Reuters-dim10 datasets are placed in "datasets" folder. The NoisyMNIST and Caltech101 datasets could be downloaded from Google cloud or Baidu cloud with password "rqv4".
After setting the configuration and downloading datasets from the cloud desk, one could run the following code to verify our method on NoisyMNIST-30000 dataset for clustering task.
python run.py --data 3If you find our work useful in your research, please consider citing:
@inproceedings{yang2021MvCLN,
title={Partially View-aligned Representation Learning with Noise-robust Contrastive Loss},
author={Mouxing Yang, Yunfan Li, Zhenyu Huang, Zitao Liu, Peng Hu, Xi Peng},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month={June},
year={2021}
}
