Wasserstein Distributional Normalization : Nonparametric Stochastic Modeling for Handling Noisy Labels

Sung Woo Park, Junseok Kwon

Keywords: Wasserstein distributional normalization, Noisy labels, Classification

Abstract: We propose a novel Wasserstein distributional normalization (WDN) algorithm to handle noisy labels for accurate classification. In this paper, we split our data into uncertain and certain samples based on small loss criteria. We investigate the geometric relationship between these two different types of samples and enhance this relation to exploit useful information, even from uncertain samples. To this end, we impose geometric constraints on the uncertain samples by normalizing them into the Wasserstein ball centered on certain samples. Experimental results demonstrate that our WDN outperforms other state-of-the-art methods on the Clothing1M and CIFAR-10/100 datasets, which have diverse noisy labels. The proposed WDN is highly compatible with existing classification methods, meaning it can be easily plugged into various methods to improve their accuracy significantly.

One-sentence Summary: We propose a novel Wasserstein distributional normalization (WDN) algorithm to handle noisy labels for accurate classification.

Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics

Reviewed Version (pdf): https://openreview.net/references/pdf?id=QpelVZd5IT