Interpretable Deep Learning Framework for Improved Disease Classification in Medical Imaging
Not on the map. Why?
Abstract (arXiv)
Deep learning models have gained increasing adoption in medical image analysis. However, these models often produce overconfident predictions, which can compromise clinical accuracy and reliability. Bridging the gap between high-performance and awareness of uncertainty remains a crucial challenge in biomedical imaging applications. This study focuses on developing a unified deep learning framework for enhancing feature integration, interpretability, and reliability in prediction. We introduced a cross-guided channel spatial attention architecture that fuses feature representations extracted from EfficientNetB4 and ResNet34. Bidirectional attention approach enables the exchange of information across networks with differing receptive fields, enhancing discriminative and contextual feature learning. For quantitative predictive uncertainty assessment, Monte Carlo (MC)-Dropout is integrated with conformal prediction. This provides statistically valid prediction sets with entropy-based uncertainty visualization. The framework is evaluated on four medical imaging benchmark datasets: chest X-rays of COVID-19, Tuberculosis, Pneumonia, and retinal Optical Coherence Tomography (OCT) images. The proposed framework achieved strong classification performance with an AUC of 99.75% for COVID-19, 100% for Tuberculosis, 99.3% for Pneumonia chest X-rays, and 98.69% for retinal OCT images. Uncertainty-aware inference yields calibrated prediction sets with interpretable examples of uncertainty, showing transparency. The results demonstrate that bidirectional cross-attention with uncertainty quantification can improve performance and transparency in medical image classification.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 42 references that are in the map. First posted on arXiv 14 Mar 2025.
Builds on
23 of 42 references.
References · 23
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2017Squeeze-and-Excitation Networkscited by 4,789
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2018CBAM: Convolutional Block Attention Modulecited by 2,477
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
- 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 690
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2019ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networkscited by 575
- 2017CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learningcited by 461
- 2019A Survey on Explainable Artificial Intelligence (XAI): Towards Medical XAIcited by 404
11 more
- 2020Classification with Valid and Adaptive Coveragecited by 396
- 2021Medical Transformer: Gated Axial-Attention for Medical Image Segmentationcited by 320
- 2018Attention Gated Networks: Learning to Leverage Salient Regions in Medical Imagescited by 272
- 2020Can AI help in screening Viral and COVID-19 pneumonia?cited by 200
- 2018Exploring Uncertainty Measures in Deep Networks for Multiple Sclerosis Lesion Detection and Segmentationcited by 146
- 2020Reliable Tuberculosis Detection using Chest X-ray with Deep Learning, Segmentation and Visualizationcited by 50
- 2023A review of uncertainty quantification in medical image analysis: probabilistic and non-probabilistic methodscited by 39
- 2018Cautious Deep Learningcited by 35
- 2021GasHis-Transformer: A Multi-scale Visual Transformer Approach for Gastric Histopathological Image Detectioncited by 26
- 2022TransBTSV2: Towards Better and More Efficient Volumetric Segmentation of Medical Imagescited by 7
- 2024Predictive uncertainty estimation in deep learning for lung carcinoma classification in digital pathology under real dataset shiftscited by 1
Built on by
Not cited yet.