Gradient Attention Map Based Verification of Deep Convolutional Neural Networks with Application to X-ray Image Datasets
By: Omid Halimi Milani , Amanda Nikho , Lauren Mills and more
Potential Business Impact:
Checks if AI can read X-rays safely.
Deep learning models have great potential in medical imaging, including orthodontics and skeletal maturity assessment. However, applying a model to data different from its training set can lead to unreliable predictions that may impact patient care. To address this, we propose a comprehensive verification framework that evaluates model suitability through multiple complementary strategies. First, we introduce a Gradient Attention Map (GAM)-based approach that analyzes attention patterns using Grad-CAM and compares them via similarity metrics such as IoU, Dice Similarity, SSIM, Cosine Similarity, Pearson Correlation, KL Divergence, and Wasserstein Distance. Second, we extend verification to early convolutional feature maps, capturing structural mis-alignments missed by attention alone. Finally, we incorporate an additional garbage class into the classification model to explicitly reject out-of-distribution inputs. Experimental results demonstrate that these combined methods effectively identify unsuitable models and inputs, promoting safer and more reliable deployment of deep learning in medical imaging.
Similar Papers
Attention Maps in 3D Shape Classification for Dental Stage Estimation with Class Node Graph Attention Networks
CV and Pattern Recognition
Shows how computers understand tooth growth.
Opening the black box of deep learning: Validating the statistical association between explainable artificial intelligence (XAI) and clinical domain knowledge in fundus image-based glaucoma diagnosis
CV and Pattern Recognition
Shows how computers see eye disease like doctors.
DB-FGA-Net: Dual Backbone Frequency Gated Attention Network for Multi-Class Classification with Grad-CAM Interpretability
Machine Learning (CS)
Finds brain tumors better without extra data.