Estimating Regression Predictive Distributions with Sample Networks - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

Estimating Regression Predictive Distributions with Sample Networks

Résumé

Estimating the uncertainty in deep neural network predictions is crucial for many real-world applications. A common approach to model uncertainty is to choose a parametric distribution and fit the data to it using maximum likelihood estimation. The chosen parametric form can be a poor fit to the data-generating distribution, resulting in unreliable uncertainty estimates. In this work, we propose SampleNet, a flexible and scalable architecture for modeling uncertainty that avoids specifying a parametric form on the output distribution. SampleNets do so by defining an empirical distribution using samples that are learned with the Energy Score and regularized with the Sinkhorn Divergence. SampleNets are shown to be able to well-fit a wide range of distributions and to outperform baselines on large-scale real-world regression tasks.

Dates et versions

hal-04255241 , version 1 (23-10-2023)

Identifiants

Citer

Ali Harakeh, Jordan Sir Kwang Hu, Naiqing Guan, Steven Waslander, Liam Paull. Estimating Regression Predictive Distributions with Sample Networks. Proceedings of the AAAI Conference on Artificial Intelligence, Feb 2023, Washington DC, United States. pp.7830-7838, ⟨10.1609/aaai.v37i6.25948⟩. ⟨hal-04255241⟩
37 Consultations
0 Téléchargements

Altmetric

Partager

More