Quantifying Uncertainty in Neural Networks through Residuals

Abstract

Regression models are of fundamental importance in explicitly explaining the response variable in terms of covariates. However, point predictions of these models limit them from many real world applications. Heteroscedasticity is common in most real-world scenarios and is hard to model due to its randomness. The Gaussian process generally captures epistemic (model) uncertainty but fails to capture heteroscedastic aleatoric uncertainty. The framework of HetGP inherently captures both epistemic and aleatoric by placing independent GP's priors on both mean function and error term. We propose the posthoc HetGP on the residuals of the trained deterministic neural network to obtain both epistemic and aleatoric uncertainty. The advantage of posthoc HetGP on residuals is that it can be extended to any type of model, since the model is assumed to be black-box that gives point predictions. We demonstrate our approach through simulation studies and UCI regression datasets.

Methodology and Results

Training and Prediction phases of the proposed method

Figure 1: Training and Prediction phases of the proposed method

Residuals plot with mean and noise functions approximated by HetGP and histogram of residuals

Figure 2: Residuals plot with mean and noise functions approximated by HetGP and histogram of residuals

Comparison of Aleatoric uncertainty in HetGP on residuals, NN and Evidential model

Figure 3: Comparison of Aleatoric uncertainty in HetGP on residuals, NN and Evidential model

Authors

Dalavai Udbhav Mallanna, IISER Bhopal, India (udbhav@students.iisertirupati.ac.in)

Rini Smita Thakur, IISER Bhopal, India (rinithakur@iiserb.ac.in)

Rajeev Ranjan Dwivedi, IISER Bhopal, India (rajeev22@iiserb.ac.in)

Vinod K Kurmi, IISER Bhopal, India (vinodkk@iiserb.ac.in)

Useful Links

Citation (BibTeX)

@inproceedings{Mallanna2024HetGP,
  author = {Dalavai Udbhav Mallanna and Rini Smita Thakur and Rajeev Ranjan Dwivedi and Vinod K Kurmi},
  title = {Quantifying Uncertainty in Neural Networks through Residuals},
  booktitle = {Proceedings of the 33rd ACM International Conference on Information and Knowledge Management},
  year = {2024},
  location = {Boise, ID, USA},
  doi = {10.1145/3627673.3679983},
  publisher = {ACM}
}