Skip to content

[ICML2020] Normalized Loss Functions for Deep Learning with Noisy Labels

License

Notifications You must be signed in to change notification settings

HanxunH/Active-Passive-Losses

Repository files navigation

Normalized Loss Functions - Active Passive Losses

Code for ICML2020 Paper "Normalized Loss Functions for Deep Learning with Noisy Labels"

Requirements

Python >= 3.6, PyTorch >= 1.3.1, torchvision >= 0.4.1, mlconfig

How To Run

Configs for the experiment settings

Check '*.yaml' file in the config folder for each experiment.

Arguments
  • noise_rate: noise rate
  • asym: use if it is asymmetric noise, default is symmetric
  • config_path: path to the configs folder
  • version: the config file name
  • exp_name: name of the experiments (as note)
  • seed: random seed

Example for 0.4 Symmetric noise rate with NCE+RCE loss

# CIFAR-10
$  python3  main.py --exp_name      test_exp            \
                    --noise_rate    0.4                 \
                    --version       nce+rce             \
                    --config_path   configs/cifar10/sym \
                    --seed          123

# CIFAR-100
$  python3  main.py --exp_name      test_exp             \
                    --noise_rate    0.4                  \
                    --version       nce+rce              \
                    --config_path   configs/cifar100/sym \
                    --seed          123

Citing this work

If you use this code in your work, please cite the accompanying paper:

@inproceedings{ma2020normalized,
  title={Normalized Loss Functions for Deep Learning with Noisy Labels},
  author={Ma, Xingjun and Huang, Hanxun and Wang, Yisen and Romano, Simone and Erfani, Sarah and Bailey, James},
  booktitle={ICML},
  year={2020}
}