← Search

Charlie Frogner

4 accepted papers

2015

Learning with a Wasserstein Loss

NeurIPS 2015poster

Learning to predict multi-label outputs is challenging, but in many problems there is a natural metric on the outputs that can be used to improve predictions. In this paper we develop a loss function for multi-label learning, based on the Wasserstein distance. The Wasserstein distance provides a nat…

Cited by 773SourcePDFScholar