Calibrating Deep Neural Network using Euclidean Distance

Wenhao Liang, Chang Dong, Liangwei Zheng, Wei Zhang, Weitong Chen
10/23/2024
cs.LGcs.CVstat.ML

No confident code implementation found for this paper.

Abstract

Uncertainty is a fundamental aspect of real-world scenarios, where perfect information is rarely available. Humans naturally develop complex internal models to navigate incomplete data and effectively respond to unforeseen or partially observed events. In machine learning, Focal Loss is commonly used to reduce misclassification rates by emphasizing hard-to-classify samples. However, it does not guarantee well-calibrated predicted probabilities and may result in models that are overconfident or underconfident. High calibration error indicates a misalignment between predicted probabilities and actual outcomes, affecting model reliability. This research introduces a novel loss function called Focal Calibration Loss (FCL), designed to improve probability calibration while retaining the advantages of Focal Loss in handling difficult samples. By minimizing the Euclidean norm through a strictly proper loss, FCL penalizes the instance-wise calibration error and constrains bounds. We provide theoretical validation for proposed method and apply it to calibrate CheXNet for potential deployment in web-based health-care systems. Extensive evaluations on various models and datasets demonstrate that our method achieves SOTA performance in both calibration and accuracy metrics.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{liang2024calibrating,
  title  = {Calibrating Deep Neural Network using Euclidean Distance},
  author = {Wenhao Liang and Chang Dong and Liangwei Zheng and Wei Zhang and Weitong Chen},
  year   = {2024},
  eprint = {2410.18321},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.18321v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.18321.svg)](https://codeofpaper.com/paper/2410.18321)

Discussion