Improving Model Performance and Explainability of Attention-Based CNN Models on Health Image Datasets Using Grad-CAM
摘要
In the realm of medical image analysis, enhancing the interpretability of convolutional neural networks (CNNs) is critical for their adoption in clinical settings. This study presents a novel approach to improve the explainability of a CNN-based medical image classification by integrating Gradient-weighted Class Activation Mapping (GradCAM) with attention mechanisms. This integration not only highlights crucial image regions that influence classification decisions but also ensures the model’s focus on clinically relevant features and enhances the model performance which is crucial in medical domain. Our experiments on a digital pathology images demonstrate that this method enhances both model performance and interpretability, thereby making AI-driven decisions more transparent and trustworthy for healthcare professionals. Attention in CNN is improving the model performance by 3.17% in accuracy as compared to without attention mechanism and Grad-CAM is on attention-based CNN is improving the interpretability of the model thereby making it more trustworthy.