English

Interpretable breast cancer classification using CNNs on mammographic images

Computer Vision and Pattern Recognition 2024-08-26 v1 Machine Learning

Abstract

Deep learning models have achieved promising results in breast cancer classification, yet their 'black-box' nature raises interpretability concerns. This research addresses the crucial need to gain insights into the decision-making process of convolutional neural networks (CNNs) for mammogram classification, specifically focusing on the underlying reasons for the CNN's predictions of breast cancer. For CNNs trained on the Mammographic Image Analysis Society (MIAS) dataset, we compared the post-hoc interpretability techniques LIME, Grad-CAM, and Kernel SHAP in terms of explanatory depth and computational efficiency. The results of this analysis indicate that Grad-CAM, in particular, provides comprehensive insights into the behavior of the CNN, revealing distinctive patterns in normal, benign, and malignant breast tissue. We discuss the implications of the current findings for the use of machine learning models and interpretation techniques in clinical practice.

Keywords

Cite

@article{arxiv.2408.13154,
  title  = {Interpretable breast cancer classification using CNNs on mammographic images},
  author = {Ann-Kristin Balve and Peter Hendrix},
  journal= {arXiv preprint arXiv:2408.13154},
  year   = {2024}
}

Comments

16 pages, 13 figures (9 in the main text, 3 in the appendix). Accepted at PMLR 2024

R2 v1 2026-06-28T18:22:17.761Z