Simple Calibration via Geodesic Kernels

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Dey, Jayanta, Xu, Haoyin, De Silva, Ashwin, Vogelstein, Joshua T.
Format: Preprint
Veröffentlicht: 2022
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908402265882624
author Dey, Jayanta
Xu, Haoyin
De Silva, Ashwin
Vogelstein, Joshua T.
author_facet Dey, Jayanta
Xu, Haoyin
De Silva, Ashwin
Vogelstein, Joshua T.
contents Deep discriminative approaches, such as decision forests and deep neural networks, have recently found applications in many important real-world scenarios. However, deploying these learning algorithms in safety-critical applications raises concerns, particularly when it comes to ensuring calibration for both in-distribution and out-of-distribution regions. Many popular methods for in-distribution (ID) calibration, such as isotonic and Platt's sigmoidal regression, exhibit adequate ID calibration performance. However, these methods are not calibrated for the entire feature space, leading to overconfidence in the out-of-distribution (OOD) region. Existing OOD calibration methods generally exhibit poor ID calibration. In this paper, we jointly address the ID and OOD problems. We leveraged the fact that deep models learn to partition feature space into a union of polytopes, that is, flat-sided geometric objects. We introduce a geodesic distance to measure the distance between these polytopes and further distinguish samples within the same polytope using a Gaussian kernel. Our experiments on both tabular and vision benchmarks show that the proposed approaches, namely Kernel Density Forest (KDF) and Kernel Density Network (KDN), obtain well-calibrated posteriors for both ID and OOD samples, while mostly preserving the classification accuracy and extrapolating beyond the training data to handle OOD inputs appropriately.
format Preprint
id arxiv_https___arxiv_org_abs_2201_13001
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Simple Calibration via Geodesic Kernels
Dey, Jayanta
Xu, Haoyin
De Silva, Ashwin
Vogelstein, Joshua T.
Machine Learning
Artificial Intelligence
Data Structures and Algorithms
Neurons and Cognition
Deep discriminative approaches, such as decision forests and deep neural networks, have recently found applications in many important real-world scenarios. However, deploying these learning algorithms in safety-critical applications raises concerns, particularly when it comes to ensuring calibration for both in-distribution and out-of-distribution regions. Many popular methods for in-distribution (ID) calibration, such as isotonic and Platt's sigmoidal regression, exhibit adequate ID calibration performance. However, these methods are not calibrated for the entire feature space, leading to overconfidence in the out-of-distribution (OOD) region. Existing OOD calibration methods generally exhibit poor ID calibration. In this paper, we jointly address the ID and OOD problems. We leveraged the fact that deep models learn to partition feature space into a union of polytopes, that is, flat-sided geometric objects. We introduce a geodesic distance to measure the distance between these polytopes and further distinguish samples within the same polytope using a Gaussian kernel. Our experiments on both tabular and vision benchmarks show that the proposed approaches, namely Kernel Density Forest (KDF) and Kernel Density Network (KDN), obtain well-calibrated posteriors for both ID and OOD samples, while mostly preserving the classification accuracy and extrapolating beyond the training data to handle OOD inputs appropriately.
title Simple Calibration via Geodesic Kernels
topic Machine Learning
Artificial Intelligence
Data Structures and Algorithms
Neurons and Cognition
url https://arxiv.org/abs/2201.13001