mlsanityData to 7 Oct 2026

Enhancing Multi-Label Thoracic Disease Diagnosis with Deep Ensemble-Based Uncertainty Quantification

arXiv 2511.18839 ↗First posted 24 Nov 2025cs.CV cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

The utility of deep learning models, such as CheXNet, in high stakes clinical settings is fundamentally constrained by their purely deterministic nature, failing to provide reliable measures of predictive confidence. This project addresses this critical gap by integrating robust Uncertainty Quantification (UQ) into a high performance diagnostic platform for 14 common thoracic diseases on the NIH ChestX-ray14 dataset. Initial architectural development failed to stabilize performance and calibration using Monte Carlo Dropout (MCD), yielding an unacceptable Expected Calibration Error (ECE) of 0.7588. This technical failure necessitated a rigorous architectural pivot to a high diversity, 9-member Deep Ensemble (DE). This resulting DE successfully stabilized performance and delivered superior reliability, achieving a State-of-the-Art (SOTA) average Area Under the Receiver Operating Characteristic Curve (AUROC) of 0.8559 and an average F1 Score of 0.3857. Crucially, the DE demonstrated superior calibration (Mean ECE of 0.0728 and Negative Log-Likelihood (NLL) of 0.1916) and enabled the reliable decomposition of total uncertainty into its Aleatoric (irreducible data noise) and Epistemic (reducible model knowledge) components, with a mean Epistemic Uncertainty (EU) of 0.0240. These results establish the Deep Ensemble as a trustworthy and explainable platform, transforming the model from a probabilistic tool into a reliable clinical decision support system.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 17 references that are in the map. First posted on arXiv 24 Nov 2025.

Builds on

10 of 17 references.

References · 10

  1. 2017On Calibration of Modern Neural Networkscited by 4,860
  2. 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
  3. 2017CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learningcited by 459
  4. 2024Estimating Epistemic and Aleatoric Uncertainty with a Single Modelcited by 52
  5. 2019Challenges in Markov chain Monte Carlo for Bayesian neural networkscited by 43
  6. 2022ZLPR: A Novel Loss for Multi-label Classificationcited by 16
  7. 2024Variational Inference for Uncertainty Quantification: an Analysis of Trade-offscited by 11
  8. 2022Controlled Dropout for Uncertainty Estimationcited by 4
  9. 2025Benchmarking Uncertainty and its Disentanglement in multi-label Chest X-Ray Classificationcited by 4
  10. 2025Reproducing and Improving CheXNet: Deep Learning for Chest X-ray Disease Classificationcited by 2

Built on by

Not cited yet.