PostHoc FREE Calibrating on Kolmogorov Arnold Networks

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Liang, Wenhao, Zhang, Wei Emma, Yue, Lin, Xu, Miao, Maennel, Olaf, Chen, Weitong
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909521156243456
author Liang, Wenhao
Zhang, Wei Emma
Yue, Lin
Xu, Miao
Maennel, Olaf
Chen, Weitong
author_facet Liang, Wenhao
Zhang, Wei Emma
Yue, Lin
Xu, Miao
Maennel, Olaf
Chen, Weitong
contents Kolmogorov Arnold Networks (KANs) are neural architectures inspired by the Kolmogorov Arnold representation theorem that leverage B Spline parameterizations for flexible, locally adaptive function approximation. Although KANs can capture complex nonlinearities beyond those modeled by standard MultiLayer Perceptrons (MLPs), they frequently exhibit miscalibrated confidence estimates manifesting as overconfidence in dense data regions and underconfidence in sparse areas. In this work, we systematically examine the impact of four critical hyperparameters including Layer Width, Grid Order, Shortcut Function, and Grid Range on the calibration of KANs. Furthermore, we introduce a novel TemperatureScaled Loss (TSL) that integrates a temperature parameter directly into the training objective, dynamically adjusting the predictive distribution during learning. Both theoretical analysis and extensive empirical evaluations on standard benchmarks demonstrate that TSL significantly reduces calibration errors, thereby improving the reliability of probabilistic predictions. Overall, our study provides actionable insights into the design of spline based neural networks and establishes TSL as a robust loss solution for enhancing calibration.
format Preprint
id arxiv_https___arxiv_org_abs_2503_01195
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PostHoc FREE Calibrating on Kolmogorov Arnold Networks
Liang, Wenhao
Zhang, Wei Emma
Yue, Lin
Xu, Miao
Maennel, Olaf
Chen, Weitong
Machine Learning
Computer Vision and Pattern Recognition
Kolmogorov Arnold Networks (KANs) are neural architectures inspired by the Kolmogorov Arnold representation theorem that leverage B Spline parameterizations for flexible, locally adaptive function approximation. Although KANs can capture complex nonlinearities beyond those modeled by standard MultiLayer Perceptrons (MLPs), they frequently exhibit miscalibrated confidence estimates manifesting as overconfidence in dense data regions and underconfidence in sparse areas. In this work, we systematically examine the impact of four critical hyperparameters including Layer Width, Grid Order, Shortcut Function, and Grid Range on the calibration of KANs. Furthermore, we introduce a novel TemperatureScaled Loss (TSL) that integrates a temperature parameter directly into the training objective, dynamically adjusting the predictive distribution during learning. Both theoretical analysis and extensive empirical evaluations on standard benchmarks demonstrate that TSL significantly reduces calibration errors, thereby improving the reliability of probabilistic predictions. Overall, our study provides actionable insights into the design of spline based neural networks and establishes TSL as a robust loss solution for enhancing calibration.
title PostHoc FREE Calibrating on Kolmogorov Arnold Networks
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.01195