Skip to content

NeurIPS'2020: Part-dependent Label Noise: Towards Instance-dependent Label Noise

Notifications You must be signed in to change notification settings

xiaoboxia/Part-dependent-label-noise

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Part-dependent Label Noise

NeurIPS‘20: Part-dependent Label Noise: Towards Instance-dependent Label Noise (PyTorch implementation).

This is the code for the paper: Part-dependent Label Noise: Towards Instance-dependent Label Noise
Xiaobo Xia, Tongliang Liu, Bo Han, Nannan Wang, Mingming Gong, Haifeng Liu, Gang Niu, Dacheng Tao, Masashi Sugiyama.

Dependencies

We implement our methods by PyTorch on NVIDIA Tesla V100 GPU. The environment is as bellow:

Install requirements.txt

pip install -r requirements.txt

Experiments

We verify the effectiveness of the proposed method on synthetic noisy datasets. In this repository, we provide the used datasets (the images and labels have been processed to .npy format). You should put the datasets in the folder “data” when you have downloaded them.
Here is a training example:

python main.py \
    --dataset mnist \
    --noise_rate 0.2 \
    --gpu 0

If you find this code useful in your research, please cite

@inproceedings{xia2020part,
  title={Part-dependent Label Noise: Towards Instance-dependent Label Noise},
  author={Xia, Xiaobo and Liu, Tongliang and Han, Bo and Wang, Nannan and Gong, Mingming and Liu, Haifeng and Niu, Gang and Tao, Dacheng and Sugiyama, Masashi},
  booktitle={NeurIPS},
  year={2020}
}

About

NeurIPS'2020: Part-dependent Label Noise: Towards Instance-dependent Label Noise

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages