Awesome
instance-dependent-label-noise
AAAA'21: Tackling Instance-Dependent Label Noise via a Universal Probabilistic Model (Pytorch implementation).
========
This is the code for the paper:
Tackling Instance-Dependent Label Noise via a Universal Probabilistic Model
Qizhou Wang, Bo Han, Tongliang Liu, Gang Niu, Jian Yang, Chen Gong.
To be presented at AAAI 2021.
If you find this code useful in your research then please cite
@inproceedings{wang2021tackling_instance,
title={Tackling Instance-Dependent Label Noise via a Universal Probabilistic Model},
author={Qizhou Wang and Bo Han and Tongliang Liu and Gang Niu and Jian Yang and
Chen Gong},
booktitle={AAAI},
pages={10183--10191},
year={2021}
}
Setups
All code was developed and tested on a single machine equiped with a NVIDIA GTX3090 GPU. The environment is as bellow:
-
Window 10
-
CUDA 10.2.89
-
Python 3.7.6 (Anaconda 4.9.2 64 bit)
-
PyTorch 1.5.0
-
numpy 1.18.1
Running Max-Matching on benchmark datasets from Amazon
Here is an example:
python main.py
Contact: Qizhou Wang (qizhouwang.nanjing@gmail.com).