GitHub - ryl0427/Code-for-OT-Filter: Code for OT-Filter: An Optimal Transport Filter for Learning with Noisy Labels

ryl0427 / Code-for-OT-Filter Public

Notifications You must be signed in to change notification settings
Fork 1
Star 6

Code for OT-Filter: An Optimal Transport Filter for Learning with Noisy Labels

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
CIFAR_Dataset.py		CIFAR_Dataset.py
Readme.txt		Readme.txt
Train_CIFAR.py		Train_CIFAR.py
data_augment.py		data_augment.py
model.py		model.py
opt.py		opt.py

Repository files navigation

PyTorch Code for the following paper at CVPR2023:
Title:OT-Filter: An Optimal Transport Filter for Learning with Noisy Labels
Authors:Chuanwen Feng, Yilong Ren, Xike Xie
Institute: University of Science and Technology of China(USTC)
https://openaccess.thecvf.com/content/CVPR2023/papers/Feng_OT-Filter_An_Optimal_Transport_Filter_for_Learning_With_Noisy_Labels_CVPR_2023_paper.pdf

Requirements:
python 3.9
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
pip install scikit-learn
pip install pot

Experiments:
python Train_CIFAR.py

Cite OT-Filter
If you find the code useful in your research, please consider citing our paper:
@InProceedings{Feng_2023_CVPR,
    author    = {Feng, Chuanwen and Ren, Yilong and Xie, Xike},
    title     = {OT-Filter: An Optimal Transport Filter for Learning With Noisy Labels},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2023},
    pages     = {16164-16174}