Explainable AI based efficient ensemble model for breast cancer classification using optical coherence tomography

被引:5
作者
Dhiman, Babita [1 ]
Kamboj, Sangeeta [1 ]
Srivastava, Vishal [1 ]
机构
[1] Thapar Inst Engn & Technol, Dept Elect & Instrumentat Engn, Patiala 147004, Punjab, India
关键词
Breast cancer; Optical coherence tomography; Machine learning; Ensemble model; TOPSIS; SHAP values; Optimization; IMAGE-ANALYSIS; PREDICTION; FUTURE; SYSTEM; DEEP;
D O I
10.1016/j.bspc.2024.106007
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
In this paper, we propose a novel framework for breast cancer classification using optical coherence tomography (OCT) images. As the incidence of breast cancer in India continues to rise, early detection is essential for increasing survival rates. For precise classification, an efficient classifier model is required. Due to the variety of datasets and performance metrics, a single classifier may not be sufficient. Thus, we propose an efficient ensemble classifier based on Technique for Order of Preference by Similarity to Ideal Solution (TOPSIS) integrated with the Crow Search Algorithm (CSA). TOPSIS, a technique for Multi-Criteria Decision Making (MCDM), ranks and selects the ensemble's base classifiers. The CSA fine-tunes the weights of the base classifiers. In addition, we employ SHAP (SHapley Additive exPlanations) values to visualize feature attributions at the observation level, thereby facilitating model interpretation and comprehension. The proposed classifier performs admirably on the experimental testing dataset, with values of 0.921, 0.921, 0.923, 0.921, 0.846 and 0.846 for precision, recall, F1-score, Kappa, and MCC, respectively. This model's ultimate goal is to reduce reliance on skilled pathologists, reduce interobserver variability, and speed up the processing of breast tissue assessments.
引用
收藏
页数:12
相关论文
共 37 条
[1]   Going Deep in Medical Image Analysis: Concepts, Methods, Challenges, and Future Directions [J].
Altaf, Fouzia ;
Islam, Syed M. S. ;
Akhtar, Naveed ;
Janjua, Naeem Khalid .
IEEE ACCESS, 2019, 7 :99540-99572
[2]   Towards Breast Cancer Response Prediction using Artificial Intelligence and Radiomics [J].
Amkrane, Yassine ;
El Adoui, Mohammed ;
Benjelloun, Mohammed .
PROCEEDINGS OF 2020 5TH INTERNATIONAL CONFERENCE ON CLOUD COMPUTING AND ARTIFICIAL INTELLIGENCE: TECHNOLOGIES AND APPLICATIONS (CLOUDTECH'20), 2020, :253-257
[3]   A novel metaheuristic method for solving constrained engineering optimization problems: Crow search algorithm [J].
Askarzadeh, Alireza .
COMPUTERS & STRUCTURES, 2016, 169 :1-12
[4]   Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis [J].
Asri, Hiba ;
Mousannif, Hajar ;
Al Moatassime, Hassan ;
Noel, Thomas .
7TH INTERNATIONAL CONFERENCE ON AMBIENT SYSTEMS, NETWORKS AND TECHNOLOGIES (ANT 2016) / THE 6TH INTERNATIONAL CONFERENCE ON SUSTAINABLE ENERGY INFORMATION TECHNOLOGY (SEIT-2016) / AFFILIATED WORKSHOPS, 2016, 83 :1064-1069
[5]   Artificial intelligence and machine learning for medical imaging: A technology review [J].
Barragan-Montero, Ana ;
Javaid, Umair ;
Valdes, Gilmer ;
Nguyen, Dan ;
Desbordes, Paul ;
Macq, Benoit ;
Willems, Siri ;
Vandewinckele, Liesbeth ;
Holmstrom, Mats ;
Lofman, Fredrik ;
Michiels, Steven ;
Souris, Kevin ;
Sterpin, Edmond ;
Lee, John A. .
PHYSICA MEDICA-EUROPEAN JOURNAL OF MEDICAL PHYSICS, 2021, 83 :242-256
[6]   The advantages of the Matthews correlation coefficient (MCC) over F1 score and accuracy in binary classification evaluation [J].
Chicco, Davide ;
Jurman, Giuseppe .
BMC GENOMICS, 2020, 21 (01)
[7]   A COEFFICIENT OF AGREEMENT FOR NOMINAL SCALES [J].
COHEN, J .
EDUCATIONAL AND PSYCHOLOGICAL MEASUREMENT, 1960, 20 (01) :37-46
[8]  
Devi RR, 2018, BIOMED PHARMACOL J, V11, P1649, DOI [10.13005/bpj/1533, DOI 10.13005/bpj/1533]
[9]   Ensemble classifier for improve diagnosis of the breast cancer using optical coherence tomography and machine learning [J].
Dubey, Kavita ;
Singla, Neeru ;
Butola, Ankit ;
Lathe, Astitwa ;
Quaiser, Darakhshan ;
Srivastava, Anurag ;
Mehta, Dalip Singh ;
Srivastava, Vishal .
LASER PHYSICS LETTERS, 2019, 16 (02)
[10]   Prediction of Breast Cancer, Comparative Review of Machine Learning Techniques, and Their Analysis [J].
Fatima, Noreen ;
Liu, Li ;
Hong, Sha ;
Ahmed, Haroon .
IEEE ACCESS, 2020, 8 :150360-150376