This is the PyTorch implementation of our IJCAI 2023 paper ProMix. A previous version at the LMNL_challenge
branch won the 1st Learning and Mining with Noisy Labels Challenge in IJCAI-ECAI 2022.
Title: ProMix: Combating Label Noise via Maximizing Clean Sample Utility
Authors: Ruixuan Xiao, Dong Yiwen, Haobo Wang, Lei Feng, Runze Wu, Gang Chen, Junbo Zhao
Affliations: Zhejiang University, Nanyang Technological University, NetEase Fuxi AI Lab
@inproceedings{ijcai2023p494,
title = {ProMix: Combating Label Noise via Maximizing Clean Sample Utility},
author = {Xiao, Ruixuan and Dong, Yiwen and Wang, Haobo and Feng, Lei and Wu, Runze and Chen, Gang and Zhao, Junbo},
booktitle = {Proceedings of the Thirty-Second International Joint Conference on
Artificial Intelligence, {IJCAI-23}},
publisher = {International Joint Conferences on Artificial Intelligence Organization},
editor = {Edith Elkind},
pages = {4442--4450},
year = {2023},
month = {8},
note = {Main Track},
doi = {10.24963/ijcai.2023/494},
url = {https://doi.org/10.24963/ijcai.2023/494},
}
After creating a virtual environment, run
pip install -r requirements.txt
We provide the shell codes for model training in the run.sh
file. Please download the source data of CIFAR-10/100 and the noise file of CIFAR-N following Learning with Noisy Labels Revisited: A Study Using Real-World Human Annotations and put them under the data
folder.
Environment settings.
conda create -n promix python=3.9
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
Then delete packages of torch, torchaudio, torchfile... in the requirements.txt. Finally installing other packages with conda or pip.
This paper is supported by Netease Youling Crowdsourcing Platform. As the importance of data continues rising, Netease Youling Crowdsourcing Platform is dedicated to utilizing various advanced algorithms to provide high-quality, low-noise labeled samples. Feel free to contact us for more information.