Learning with a Wasserstein Loss

Charlie Frogner, Chiyuan Zhang, Hossein Mobahi +2

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 deve…