Adaptive Conformal Prediction for Reliable and Explainable Medical Image Classification
Not on the map. Why?
Abstract (arXiv)
Deep learning models for medical imaging often exhibit overconfidence, creating safety risks in ambiguous diagnostic scenarios. While Conformal Prediction (CP) provides distribution-free statistical guarantees, standard methods such as Regularized Adaptive Prediction Sets (RAPS) optimize for average efficiency and can mask severe failures on difficult inputs. We propose an Adaptive Lambda Criterion for RAPS that minimizes the worst-case coverage violation across prediction set size strata. On OrganAMNIST (58,850 abdominal CT images, 11 classes), standard size-optimized RAPS converges to near-deterministic behavior with stratified undercoverage on uncertain samples, while our method achieves 95.72 percent global coverage with average set size 1.09 and at least 90 percent coverage across all strata. Cross-domain validation on PathMNIST (107,180 pathology images, 9 classes) confirms generalizability. Quantitative Grad-CAM analysis (rho = -0.30, p < 1e-22) shows that multi-label predictions correspond to focused attention on anatomically ambiguous regions. These results demonstrate that the proposed method improves reliability while maintaining efficiency, making it suitable for safety-critical medical AI applications.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 26 references that are in the map. First posted on arXiv 13 May 2026.
Builds on
17 of 26 references.
References · 17
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 777
- 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 682
- 2016Distribution-Free Predictive Inference For Regressioncited by 657
- 2020Classification with Valid and Adaptive Coveragecited by 390
- 2020Uncertainty Sets for Image Classifiers using Conformal Predictioncited by 365
- 2022On the Utility of Prediction Sets in Human-AI Teamscited by 34
- 2022A review of predictive uncertainty estimation with machine learningcited by 29
5 more
- 2024Conformal Prediction: A Data Perspectivecited by 25
- 2024Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibrationcited by 18
- 2023Improving Adaptive Conformal Prediction Using Self-Supervised Learningcited by 17
- 2025Conformal Prediction for Image Segmentation Using Morphological Prediction Setscited by 13
- 2025Adaptive Coverage Policies in Conformal Predictioncited by 4
Built on by
Not cited yet.