Explainable Ensemble Machine Learning for Breast Cancer Diagnosis Based on Ultrasound Image Texture Features

Alireza Rezazadeh, Yasamin Jafarian and Ali Kord
Additional contact information
Alireza Rezazadeh: Department of Electrical and Computer Engineering, University of Minnesota, Minneapolis, MN 55414, USA
Yasamin Jafarian: Department Computer Science and Engineering, University of Minnesota, Minneapolis, MN 55414, USA
Ali Kord: Division of Interventional Radiology, Department of Radiology, University of Cincinnati, Cincinnati, OH 45221, USA

Forecasting, 2022, vol. 4, issue 1, 1-13

Abstract: Image classification is widely used to build predictive models for breast cancer diagnosis. Most existing approaches overwhelmingly rely on deep convolutional networks to build such diagnosis pipelines. These model architectures, although remarkable in performance, are black-box systems that provide minimal insight into the inner logic behind their predictions. This is a major drawback as the explainability of prediction is vital for applications such as cancer diagnosis. In this paper, we address this issue by proposing an explainable machine learning pipeline for breast cancer diagnosis based on ultrasound images. We extract first- and second-order texture features of the ultrasound images and use them to build a probabilistic ensemble of decision tree classifiers. Each decision tree learns to classify the input ultrasound image by learning a set of robust decision thresholds for texture features of the image. The decision path of the model predictions can then be interpreted by decomposing the learned decision trees. Our results show that our proposed framework achieves high predictive performance while being explainable.

Keywords: ultrasound image texture analysis; breast cancer prediction; explainable machine learning; ensemble classification; decision tree classification (search for similar items in EconPapers)
JEL-codes: A1 B4 C0 C1 C2 C3 C4 C5 C8 M0 Q2 Q3 Q4 (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2571-9394/4/1/15/pdf (application/pdf)
https://www.mdpi.com/2571-9394/4/1/15/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jforec:v:4:y:2022:i:1:p:15-274:d:748368

Access Statistics for this article

Forecasting is currently edited by Ms. Joss Chen

More articles in Forecasting from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().