Related Experiment Videos
Explainable machine learning for breast cancer prediction in resource-constrained settings: A multi-algorithmic
Oluwaseun Adebayo Bamodu1,2,3, Sumaiya Nezam3, Chen-Chih Chung4,5,6
1Building Research and Implementation to Drive Growth and Equity in Africa (BRIDGEAfrica), Dar Es Salaam, Tanzania.
Abstract:
Breast cancer remains the most commonly diagnosed malignancy among women globally, with disproportionately higher mortality rates in low- and middle-income countries (LMICs) where diagnostic delays and limited specialist pathology capacity are widespread. While machine learning (ML) approaches achieve strong predictive performance for cancer classification, algorithmic opacity and absence of interpretability frameworks tailored to resource-constrained environments have impeded clinical adoption. This study bridges the translational gap between predictive accuracy and clinical utility by developing an explainable artificial intelligence (XAI) framework specifically designed for breast cancer diagnosis in underserved healthcare settings. Using the Wisconsin Breast Cancer Diagnostic Dataset (569 fine-needle aspirate cytological specimens with 30 nuclear morphometric features), we systematically benchmarked eight supervised classification algorithms: Logistic Regression, Random Forest, XGBoost, LightGBM, Support Vector Machine (SVM), Gradient Boosting, Decision Tree, and K-Nearest Neighbors, using stratified 10-fold cross-validation and an independent hold-out test set (80:20 split). Performance was evaluated across discriminative and probabilistic metrics, including AUC-ROC, F1-score, Matthews Correlation Coefficient (MCC), and Brier score, and interpretability was operationalized through SHapley Additive exPlanations (SHAP) analysis with global feature importance, cross-model consensus ranking, and individual-level dependence characterization. All ensemble and regularized models achieved test-set AUCs above 0.98, with XGBoost and SVM attaining the highest AUC of 0.996, and Logistic Regression the highest accuracy (98.25%) and MCC (0.962). SHAP analysis consistently identified worst perimeter, worst concave points, and worst area as the dominant predictors, with strong concordance across gradient-boosted models (pairwise Spearman rho: XGBoost-LightGBM 0.86, XGBoost-Random Forest 0.82, Random Forest-LightGBM 0.67). Logistic Regression also demonstrated superior probability calibration, a critical requirement for clinical risk stratification. Collectively, these findings deliver a reproducible, transparent framework whose SHAP-derived signatures align with established cytopathological principles, supporting responsible integration of interpretable ML into resource-limited diagnostic workflows and providing a template for equitable AI deployment in global oncology.