Interpretable breast cancer classification using CNNs on mammographic images

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Balve, Ann-Kristin, Hendrix, Peter
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912000103153664
author Balve, Ann-Kristin
Hendrix, Peter
author_facet Balve, Ann-Kristin
Hendrix, Peter
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2408_13154
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Interpretable breast cancer classification using CNNs on mammographic images
Balve, Ann-Kristin
Hendrix, Peter
Computer Vision and Pattern Recognition
Machine Learning
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.
title Interpretable breast cancer classification using CNNs on mammographic images
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2408.13154