Uncertainty Quantification Metrics for Deep Regression
Simon Kristofferson Lind, Ziliang Xiong, Per-Erik Forssén, Volker KrügerPattern Recognition Letters
Volume 186, Number 1, Pages 91-97
October 2024
Abstract
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 uncertainty quantification. Specifically, we focus on regression tasks, and investigate Area Under Sparsification Error (AUSE), Calibration Error (CE), Spearman's Rank Correlation, and Negative Log-Likelihood (NLL). Using multiple 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.
Full Paper
Manuscript on ArXiv published by Elsevier [DOI].
Bibtex entry
@Article{lind24, author = {Simon Kristofferson Lind and Ziliang Xiong and Per-Erik Forss\'en and Volker Kr\"uger}, title = {Uncertainty Quantification Metrics for Deep Regression}, journal = {Pattern Recognition Letters}, year = {2024}, volume = {186}, number = {1}, pages = {91-97}, month = {October}, publisher = {{Elsevier}}, url = {https://doi.org/10.1016/j.patrec.2024.09.011} }
Per-Erik Forssén
Contact:
Computer Vision Laboratory
Department of Electrical Engineering
Building B
Room 2D:521
SE-581 83 Linköping, Sweden
+46(0)13 285654
< >
My pages:
Informationsansvarig: Per-Erik Forssén
Senast uppdaterad: 2024-09-28