Uncertainty Quantification Metrics for Deep Regression

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lind, Simon Kristoffersson, Xiong, Ziliang, Forssén, Per-Erik, Krüger, Volker
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910675849183232
author Lind, Simon Kristoffersson
Xiong, Ziliang
Forssén, Per-Erik
Krüger, Volker
author_facet Lind, Simon Kristoffersson
Xiong, Ziliang
Forssén, Per-Erik
Krüger, Volker
contents When deploying deep neural networks on robots or other physical systems, the learned model should reliably quantify predictive uncertainty. A reliable uncertainty allows downstream modules to reason about the safety of its actions. In this work, we address metrics for evaluating such an uncertainty. Specifically, we focus on regression tasks, and investigate Area Under Sparsification Error (AUSE), Calibration Error, Spearman's Rank Correlation, and Negative Log-Likelihood (NLL). Using synthetic regression datasets, we look into how those metrics behave under four typical types of uncertainty, their stability regarding the size of the test set, and reveal their strengths and weaknesses. Our results indicate that Calibration Error is the most stable and interpretable metric, but AUSE and NLL also have their respective use cases. We discourage the usage of Spearman's Rank Correlation for evaluating uncertainties and recommend replacing it with AUSE.
format Preprint
id arxiv_https___arxiv_org_abs_2405_04278
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Uncertainty Quantification Metrics for Deep Regression
Lind, Simon Kristoffersson
Xiong, Ziliang
Forssén, Per-Erik
Krüger, Volker
Machine Learning
Robotics
When deploying deep neural networks on robots or other physical systems, the learned model should reliably quantify predictive uncertainty. A reliable uncertainty allows downstream modules to reason about the safety of its actions. In this work, we address metrics for evaluating such an uncertainty. Specifically, we focus on regression tasks, and investigate Area Under Sparsification Error (AUSE), Calibration Error, Spearman's Rank Correlation, and Negative Log-Likelihood (NLL). Using synthetic regression datasets, we look into how those metrics behave under four typical types of uncertainty, their stability regarding the size of the test set, and reveal their strengths and weaknesses. Our results indicate that Calibration Error is the most stable and interpretable metric, but AUSE and NLL also have their respective use cases. We discourage the usage of Spearman's Rank Correlation for evaluating uncertainties and recommend replacing it with AUSE.
title Uncertainty Quantification Metrics for Deep Regression
topic Machine Learning
Robotics
url https://arxiv.org/abs/2405.04278