Skip to content

tmllab/2024_ICLR_LabelWave

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

Early Stopping Against Label Noise Without Validation Data

ICLR 2024 Poster | [Paper] | [Code]

Suqin Yuan, Lei Feng, Tongliang Liu

TL;DR

The Label Wave method enables early stopping by analyzing training dynamics in learning with noisy labels, eliminating the need for separate validation data.

BibTeX

@inproceedings{
yuan2024early,
title={Early Stopping Against Label Noise Without Validation Data},
author={Suqin Yuan and Lei Feng and Tongliang Liu},
booktitle={The Twelfth International Conference on Learning Representations},
year={2024}
}

Dependencies

We implement our methods by PyTorch on NVIDIA RTX 3090&4090 GPU. The key environments is as bellow:

Experiments

You should put the CIFAR datasets in the folder .\cifar-10 and .\cifar-100 when you have downloaded them.

Here is a training example:

python3 main.py

Contact: Suqin Yuan (suqinyuan.cs@gmail.com).

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages