Model-Free Local Recalibration of Neural Networks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Torres, R., Nott, D. J., Sisson, S. A., Rodrigues, T., Reis, J. G., Rodrigues, G. S.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910361174671360
author Torres, R.
Nott, D. J.
Sisson, S. A.
Rodrigues, T.
Reis, J. G.
Rodrigues, G. S.
author_facet Torres, R.
Nott, D. J.
Sisson, S. A.
Rodrigues, T.
Reis, J. G.
Rodrigues, G. S.
contents Artificial neural networks (ANNs) are highly flexible predictive models. However, reliably quantifying uncertainty for their predictions is a continuing challenge. There has been much recent work on "recalibration" of predictive distributions for ANNs, so that forecast probabilities for events of interest are consistent with certain frequency evaluations of them. Uncalibrated probabilistic forecasts are of limited use for many important decision-making tasks. To address this issue, we propose a localized recalibration of ANN predictive distributions using the dimension-reduced representation of the input provided by the ANN hidden layers. Our novel method draws inspiration from recalibration techniques used in the literature on approximate Bayesian computation and likelihood-free inference methods. Most existing calibration methods for ANNs can be thought of as calibrating either on the input layer, which is difficult when the input is high-dimensional, or the output layer, which may not be sufficiently flexible. Through a simulation study, we demonstrate that our method has good performance compared to alternative approaches, and explore the benefits that can be achieved by localizing the calibration based on different layers of the network. Finally, we apply our proposed method to a diamond price prediction problem, demonstrating the potential of our approach to improve prediction and uncertainty quantification in real-world applications.
format Preprint
id arxiv_https___arxiv_org_abs_2403_05756
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Model-Free Local Recalibration of Neural Networks
Torres, R.
Nott, D. J.
Sisson, S. A.
Rodrigues, T.
Reis, J. G.
Rodrigues, G. S.
Methodology
Machine Learning
62G07 (Primary), 68T07, 68T37 (Secondary), 68Q10
G.3; I.5.1; I.6.4
Artificial neural networks (ANNs) are highly flexible predictive models. However, reliably quantifying uncertainty for their predictions is a continuing challenge. There has been much recent work on "recalibration" of predictive distributions for ANNs, so that forecast probabilities for events of interest are consistent with certain frequency evaluations of them. Uncalibrated probabilistic forecasts are of limited use for many important decision-making tasks. To address this issue, we propose a localized recalibration of ANN predictive distributions using the dimension-reduced representation of the input provided by the ANN hidden layers. Our novel method draws inspiration from recalibration techniques used in the literature on approximate Bayesian computation and likelihood-free inference methods. Most existing calibration methods for ANNs can be thought of as calibrating either on the input layer, which is difficult when the input is high-dimensional, or the output layer, which may not be sufficiently flexible. Through a simulation study, we demonstrate that our method has good performance compared to alternative approaches, and explore the benefits that can be achieved by localizing the calibration based on different layers of the network. Finally, we apply our proposed method to a diamond price prediction problem, demonstrating the potential of our approach to improve prediction and uncertainty quantification in real-world applications.
title Model-Free Local Recalibration of Neural Networks
topic Methodology
Machine Learning
62G07 (Primary), 68T07, 68T37 (Secondary), 68Q10
G.3; I.5.1; I.6.4
url https://arxiv.org/abs/2403.05756