Encouraging Discriminative Attention Through Contrastive Explainability Learning for Lung Cancer Diagnosis
V Shravya, M P Sunil, B Natarajan, R Elakkiya
Abstract
Lung cancer diagnosis using CT scans is critical for early detection, but existing deep methods often lack interpretability, especially in highlighting medically relevant regions. Most approaches optimize only for prediction accuracy, leaving model explanations unstructured and inconsistent across samples. We introduce Contrastive Explainability Learning (CEL), a novel training approach that aligns Grad-CAM heatmaps across class-consistent samples while enforcing dissimilarity across different classes. Unlike prior methods, CEL integrates explanation supervision directly into the loss function, enabling interpretable representation learning without sacrificing accuracy. Using only a lightweight, spatially-attended CNN, our model achieves strong performance (99.2% accuracy, 99.5% F1 score) on the IQ-OTH/NCCD dataset and demonstrates robust generalization (93.0% accuracy) on the more complex HF Lung Cancer dataset with multiple cancer subtypes. Statistical analysis across multiple trials confirms that these improvements are significant (p < 0.01). We demonstrate through comprehensive comparisons with alternative XAI methods that CEL produces more consistent, discriminative explanations with minimal computational overhead. Experiments show that our contrastive saliency framework guides the CNN to focus on class-specific anatomical regions, improving both transparency and diagnostic trust while maintaining efficiency suitable for clinical deployment.