mlsanityData to 14 Oct 2026

Beyond Uncertainty Quantification: Learning Uncertainty for Trust-Informed Neural Network Decisions - A Case Study in COVID-19 Classification

arXiv 2410.02805 ↗First posted 19 Sep 2024eess.IV cs.AI cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Reliable uncertainty quantification is critical in high-stakes applications, such as medical diagnosis, where confidently incorrect predictions can erode trust in automated decision-making systems. Traditional uncertainty quantification methods rely on a predefined confidence threshold to classify predictions as confident or uncertain. However, this approach assumes that predictions exceeding the threshold are trustworthy, while those below it are uncertain, without explicitly assessing the correctness of high-confidence predictions. As a result, confidently incorrect predictions may still occur, leading to misleading uncertainty assessments. To address this limitation, this study proposed an uncertainty-aware stacked neural network, which extends conventional uncertainty quantification by learning when predictions should be trusted. The framework consists of a two-tier model: the base model generates predictions with uncertainty estimates, while the meta-model learns to assign a trust flag, distinguishing confidently correct cases from those requiring expert review. The proposed approach is evaluated against the traditional threshold-based method across multiple confidence thresholds and pre-trained architectures using the COVIDx CXR-4 dataset. Results demonstrate that the proposed framework significantly reduces confidently incorrect predictions, offering a more trustworthy and efficient decision-support system for high-stakes domains.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 36 references that are in the map. First posted on arXiv 19 Sep 2024.

Builds on

15 of 36 references.

References · 15

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
  3. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  4. 2020A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challengescited by 819
  5. 2019Big Transfer (BiT): General Visual Representation Learningcited by 769
  6. 2017Bayesian Recurrent Neural Networkscited by 39
  7. 2020An Uncertainty-aware Transfer Learning-based Framework for Covid-19 Diagnosiscited by 22
  8. 2021MCUa: Multi-level Context and Uncertainty aware Dynamic Deep Ensemble for Breast Cancer Histology Image Classificationcited by 12
  9. 2021Uncertainty-Aware Credit Card Fraud Detection Using Deep Learningcited by 10
  10. 2022ULTRA: Uncertainty-aware Label Distribution Learning for Breast Tumor Cellularity Assessmentcited by 7
  11. 2021An Uncertainty-aware Loss Function for Training Neural Networks with Calibrated Predictionscited by 6
  12. 2023COVIDx CXR-4: An Expanded Multi-Institutional Open-Source Benchmark Dataset for Chest X-ray Image-Based Computer-Aided COVID-19 Diagnosticscited by 6
3 more
  1. 2020Using Bayesian deep learning approaches for uncertainty-aware building energy surrogate modelscited by 5
  2. 2021An Uncertainty-Aware Deep Learning Framework for Defect Detection in Casting Productscited by 4
  3. 2024Improving Deep Learning Model Calibration for Cardiac Applications using Deterministic Uncertainty Networks and Uncertainty-aware Trainingcited by 1

Built on by

Not cited yet.