Estimating Multimodal Aleatoric Uncertainty in Regression Tasks Using Hinge-Wasserstein Loss
The authors propose the hinge-Wasserstein loss, a simple improvement to the Wasserstein loss, to better estimate multimodal aleatoric uncertainty in regression tasks when full ground truth distributions are unavailable.