PAC-Bayes Analysis for Recalibration in Classification

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Fujisawa, Masahiro, Futami, Futoshi
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912474938212352
author Fujisawa, Masahiro
Futami, Futoshi
author_facet Fujisawa, Masahiro
Futami, Futoshi
contents Nonparametric estimation using uniform-width binning is a standard approach for evaluating the calibration performance of machine learning models. However, existing theoretical analyses of the bias induced by binning are limited to binary classification, creating a significant gap with practical applications such as multiclass classification. Additionally, many parametric recalibration algorithms lack theoretical guarantees for their generalization performance. To address these issues, we conduct a generalization analysis of calibration error using the probably approximately correct Bayes framework. This approach enables us to derive the first optimizable upper bound for generalization error in the calibration context. On the basis of our theory, we propose a generalization-aware recalibration algorithm. Numerical experiments show that our algorithm enhances the performance of Gaussian process-based recalibration across various benchmark datasets and models.
format Preprint
id arxiv_https___arxiv_org_abs_2406_06227
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PAC-Bayes Analysis for Recalibration in Classification
Fujisawa, Masahiro
Futami, Futoshi
Machine Learning
Nonparametric estimation using uniform-width binning is a standard approach for evaluating the calibration performance of machine learning models. However, existing theoretical analyses of the bias induced by binning are limited to binary classification, creating a significant gap with practical applications such as multiclass classification. Additionally, many parametric recalibration algorithms lack theoretical guarantees for their generalization performance. To address these issues, we conduct a generalization analysis of calibration error using the probably approximately correct Bayes framework. This approach enables us to derive the first optimizable upper bound for generalization error in the calibration context. On the basis of our theory, we propose a generalization-aware recalibration algorithm. Numerical experiments show that our algorithm enhances the performance of Gaussian process-based recalibration across various benchmark datasets and models.
title PAC-Bayes Analysis for Recalibration in Classification
topic Machine Learning
url https://arxiv.org/abs/2406.06227