BootstrappingLoss
BootstrappingLoss copied to clipboard
Bootstrapping loss function implementation in pytorch
Bootstrapping loss function implementation
based on "Training Deep Neural Networks on Noisy Labels with Bootstrapping" https://arxiv.org/abs/1412.6596
Experiments on MNIST
Experiments on MNIST:
cd examples/mnist && python main.py run --mode hard_bootstrap --noise_fraction=0.45
cd examples/mnist && python main.py run --mode soft_bootstrap --noise_fraction=0.45
cd examples/mnist && python main.py run --mode xentropy --noise_fraction=0.45
cd examples/mnist && sh run_experiments.sh >> out 2> log
- Experiments on TRAINS
-
WITH_TRAINS=True sh run_experiments.sh >> out 2> log
Requirements:
- pytorch>=1.3
- torchvision>=0.4.1
- pytorch-ignite>=0.4.2
- google fire>=0.3.1
pip install -r requirements.txt