Home

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:

Running Max-Matching on benchmark datasets from Amazon

Here is an example:


python main.py

Contact: Qizhou Wang (qizhouwang.nanjing@gmail.com).