Optimized Breast Cancer Classification via SHAP-Based Feature Selection and Weighted Voting Ensemble Learning

Breast cancer (BC) is among the most common cancers affecting women worldwide, highlighting the urgent need for early and accurate diagnosis. Machine learning (ML) has emerged as a powerful tool for BC classification, enhancing diagnostic precision and improving patient outcomes. However, ML-based diagnostic systems often struggle to identify the most relevant features, particularly in moderately dimensional datasets containing redundant or non-informative attributes. To address this, we propose an optimized ensemble learning model that integrates SHapley Additive exPlanations (SHAP) based feature selection with a weighted soft-voting ensemble classifier. SHAP provides a model-agnostic, theoretically grounded approach for identifying the most influential features by quantifying their contribution to model predictions. This not only improves interpretability but also reduces the feature space without compromising performance. By selecting the top 15 most important features, the model achieves higher efficiency and clearer clinical insight, both essential for medical decision-making. The ensemble combines Extra Trees (ET), LightGBM, XGBoost (XGB), and Support Vector Machine (SVM), leveraging the strengths of each. Our model achieves state-of-the-art performance on the Wisconsin Breast Cancer Diagnostic (WBCD) dataset, with 99.42% accuracy, 100% precision, and 98.44% recall. For the University of California, Irvine (UCI) preprocessed dataset, the model achieves 95.32% accuracy, 93.81% precision, and 99.07% recall, demonstrating robustness across different datasets. By uniting explainable AI (via SHAP) with an optimized ensemble strategy, our approach enhances both classification accuracy and model transparency, establishing it as a reliable and interpretable tool for early BC detection.

Authors

Institutions

Publication Details

Journal
IEEE Open Journal of the Computer Society
Published
2025-01-01
DOI
https://doi.org/10.1109/ojcs.2025.3610146
Citations
1
Primary Topic
AI in cancer detection
Type
article
Field-Weighted Citation Impact
2.48
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Optimized Breast Cancer Classification via SHAP-Based Feature Selection and Weighted Voting Ensemble Learning

Jungpil Shin, Abu Saleh Musa Miah, Taro Suzuki, Najmul Hassan et al.
1 citations
IEEE Open Journal of the Computer Society
AI in cancer detection
2.48
article

Optimized Breast Cancer Classification via SHAP-Based Feature Selection and Weighted Voting Ensemble Learning

Jungpil Shin, Abu Saleh Musa Miah, Taro Suzuki, Najmul Hassan, Sultan Al Farhood
article en
1 citations

Abstract

Breast cancer (BC) is among the most common cancers affecting women worldwide, highlighting the urgent need for early and accurate diagnosis. Machine learning (ML) has emerged as a powerful tool for BC classification, enhancing diagnostic precision and improving patient outcomes. However, ML-based diagnostic systems often struggle to identify the most relevant features, particularly in moderately dimensional datasets containing redundant or non-informative attributes. To address this, we propose an optimized ensemble learning model that integrates SHapley Additive exPlanations (SHAP) based feature selection with a weighted soft-voting ensemble classifier. SHAP provides a model-agnostic, theoretically grounded approach for identifying the most influential features by quantifying their contribution to model predictions. This not only improves interpretability but also reduces the feature space without compromising performance. By selecting the top 15 most important features, the model achieves higher efficiency and clearer clinical insight, both essential for medical decision-making. The ensemble combines Extra Trees (ET), LightGBM, XGBoost (XGB), and Support Vector Machine (SVM), leveraging the strengths of each. Our model achieves state-of-the-art performance on the Wisconsin Breast Cancer Diagnostic (WBCD) dataset, with 99.42% accuracy, 100% precision, and 98.44% recall. For the University of California, Irvine (UCI) preprocessed dataset, the model achieves 95.32% accuracy, 93.81% precision, and 99.07% recall, demonstrating robustness across different datasets. By uniting explainable AI (via SHAP) with an optimized ensemble strategy, our approach enhances both classification accuracy and model transparency, establishing it as a reliable and interpretable tool for early BC detection.

IEEE Open Journal of the Computer SocietyVol. 6
University of Aizu (JP), King Saud University (SA)
Openalex Percentile: Top 9%
AI in cancer detection
2.48
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.