English

Logit-based Uncertainty Measure in Classification

Machine Learning 2021-07-08 v1

Abstract

We introduce a new, reliable, and agnostic uncertainty measure for classification tasks called logit uncertainty. It is based on logit outputs of neural networks. We in particular show that this new uncertainty measure yields a superior performance compared to existing uncertainty measures on different tasks, including out of sample detection and finding erroneous predictions. We analyze theoretical foundations of the measure and explore a relationship with high density regions. We also demonstrate how to test uncertainty using intermediate outputs in training of generative adversarial networks. We propose two potential ways to utilize logit-based uncertainty in real world applications, and show that the uncertainty measure outperforms.

Keywords

Cite

@article{arxiv.2107.02845,
  title  = {Logit-based Uncertainty Measure in Classification},
  author = {Huiyu Wu and Diego Klabjan},
  journal= {arXiv preprint arXiv:2107.02845},
  year   = {2021}
}
R2 v1 2026-06-24T03:56:45.704Z