Calibrating Deep Neural Network using Euclidean Distance

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liang, Wenhao, Dong, Chang, Zheng, Liangwei, Zhang, Wei, Chen, Weitong
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913978127482880
author Liang, Wenhao
Dong, Chang
Zheng, Liangwei
Zhang, Wei
Chen, Weitong
author_facet Liang, Wenhao
Dong, Chang
Zheng, Liangwei
Zhang, Wei
Chen, Weitong
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2410_18321
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Calibrating Deep Neural Network using Euclidean Distance
Liang, Wenhao
Dong, Chang
Zheng, Liangwei
Zhang, Wei
Chen, Weitong
Machine Learning
Computer Vision and Pattern Recognition
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.
title Calibrating Deep Neural Network using Euclidean Distance
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.18321