A Statistically Validated and Interpretable Machine Learning Framework for Breast Cancer Classification Using Shape Descriptors

Authors

  • K. A. Sotonwa Department of Computer Science, Lagos State University, Nigeria
  • J. O. Akintayo Department of Computer Science, Lagos State University, Nigeria
  • A. Pearson Department of Medicine, University of Chicago, Chicago, USA
  • A. Sofoluwe Department of Radiology, University of Ibadan College of Medicine, Ibadan, Nigeria
  • S. Arekete Department of Computer Science, Redeemer’s University, Ede, Nigeria
  • S. Sammet Center for Clinical Cancer Genetics and Global Health, University of Chicago; Chicago, USA
  • F. Olopade Department of Medicine, University of Chicago, Chicago, USA & Center for Clinical Cancer Genetics and Global Health, University of Chicago; Chicago, USA
  • B. Aribisala Department of Computer Science, Redeemer’s University, Ede, Nigeria & Department of Medicine, University of Chicago, Chicago, USA

DOI:

https://doi.org/10.63746/njtd.v23i1.4248

Keywords:

Breast Cancer, Feature selection, Machine Learning Algorithm, Shape Descriptor, Chi square

Abstract

Breast cancer (BC) remains a major global health priority, with nearly half a million invasive cases reported in 2024. Early diagnosis is critical, and Machine Learning (ML) offers significant potential for automated detection. This study evaluated ML models using tumour shape descriptors from the Wisconsin Diagnostic Breast Cancer (WDBC) dataset (569 cases: 357 benign, 212 malignant). Ten numerical descriptors, including radius, texture, and concave points, were extracted from digitized fine needle aspirates (FNA). The dataset was split 70:30 into training and test sets, with Chi-square feature selection, mutual information (MI), and ANOVA feature selection applied to reduce dimensionally and prevent data leakage. Three models: Artificial Neural Network (ANN), Classification and Regression Tree (CART), and Logistic Regression (LR) were trained using stratified cross-validation and evaluated on the independent test set. LR achieved the highest AUC (97.31%), significantly outperforming ANN (p = 0.009) and CART (p = 0.001) by DeLong’s test, while McNemar’s test revealed no significant differences in misclassification rates (p > 0.05). CART achieved the best F1-score, reflecting balanced sensitivity and specificity. These findings demonstrated that statistically validated and interpretable ML models provide robust diagnostic performance for BC classification

References

Agarap, A. F. M. (2018). On breast cancer detection: an application of machine learning algorithms on the wisconsin diagnostic dataset. Paper presented at the Proceedings of the 2nd International Conference on Machine Learning and Soft Computing.

Ahsan, M. M., Luna, S. A., & Siddique, Z. (2022). Machine-learning-based disease diagnosis: A comprehensive review. Paper presented at the Healthcare.

Arkoudis, N.-A., & Papadakos, S. P. (2025). Machine learning applications in healthcare clinical practice and research. World Journal of Clinical Cases, 13(1), 99744.

Arnold, M., Morgan, E., Rumgay, H., Mafra, A., Singh, D., Laversanne, M., . . . Siesling, S. (2022). Current and future burden of breast cancer: Global Statistics for 2020 and 2040. The Breast, 66, 15-23.

Arshad, M. A., Shahriar, S., & Anjum, K. (2023). The Power Of Simplicity: Why Simple Linear Models Outperform Complex Machine Learning Techniques--Case of Breast Cancer Diagnosis. arXiv preprint arXiv:2306.02449.

Awan, M. Z., Arif, M. S., Abideen, M. Z. U., & Abodayeh, K. (2024). Comparative analysis of machine learning models for breast cancer prediction and diagnosis: A dual-dataset approach. Indonesian Journal of Electrical Engineering and Computer Science, 34(3), 2032-2044.

Christodoulou, E., Ma, J., Collins, G. S., Steyerberg, E. W., Verbakel, J. Y., & Van Calster, B. (2019). A systematic review shows no performance benefit of machine learning over logistic regression for clinical prediction models. Journal of Clinical Epidemiology, 110, 12-22.

Ekpo, E. U., Alakhras, M., & Brennan, P. (2018). Errors in mammography cannot be solved through technology alone. Asian Pacific Journal of Cancer Prevention: APJCP, 19(2), 291.

Esener, ?. I., Kara, ?., Ergin, S., & Çal???r, C. A hybrid textural and geometrical feature extraction to reveal hidden information from suspicious regions on mammograms. Eski?ehir Technical University Journal of Science and Technology A-Applied Sciences and Engineering, 23(1), 70-86.

Esteva, A., Kuprel, B., Novoa, R. A., Ko, J., Swetter, S. M., Blau, H. M., & Thrun, S. (2017). Dermatologist-level classification of skin cancer with deep neural networks. Nature, 542(7639), 115-118.

Fogliatto, F. S., Anzanello, M. J., Soares, F., & Brust-Renck, P. G. (2019). Decision support for breast cancer detection: classification improvement through feature selection. Cancer Control, 26(1), 1073274819876598.

Hou, Y. (2020). Breast cancer pathological image classification based on deep learning. Journal of X-ray Science and Technology, 28(4), 727-738.

Huang, Y. L., Chen, D. R., Jiang, Y. R., Kuo, S. J., Wu, H. K., & Moon, W. (2008). Computer?aided diagnosis using morphological features for classifying breast lesions on ultrasound. Ultrasound in Obstetrics and Gynecology: The Official Journal of the International Society of Ultrasound in Obstetrics and Gynecology, 32(4), 565-572.

Japkowicz, N., & Shah, M. (2011). Evaluating learning algorithms: a classification perspective: Cambridge University Press.

Joel, L. O., Doorsamy, W., & Paul, B. S. (2024). On the performance of imputation techniques for missing values on healthcare datasets. arXiv preprint arXiv:2403.14687.

Kim, J., Harper, A., McCormack, V., Sung, H., Houssami, N., Morgan, E., Fidler-Benaoudia, M. M. (2025). Global patterns and trends in breast cancer incidence and mortality across 185 countries. Nature Medicine, 1-9.

Litjens, G., Kooi, T., Bejnordi, B. E., Setio, A. A. A., Ciompi, F., Ghafoorian, M., Sánchez, C. I. (2017). A survey on deep learning in medical image analysis. Medical Image Analysis, 42, 60-88.

McKinney, S. M., Sieniek, M., Godbole, V., Godwin, J., Antropova, N., Ashrafian, H., Darzi, A. (2020). International evaluation of an AI system for breast cancer screening. Nature, 577(7788), 89-94.

Park, J., Witowski, J., Xu, Y., Trivedi, H., Gichoya, J., Brown-Mulry, B., Lewin, A. (2025). A Multi-Modal AI System for Screening Mammography: Integrating 2D and 3D Imaging to Improve Breast Cancer Detection in a Prospective Clinical Study. arXiv preprint arXiv:2504.05636.

Patro, S., & Sahu, K. K. (2015). Normalization: A preprocessing stage. arXiv preprint arXiv:1503.06462.

Petushi, S., Garcia, F. U., Haber, M. M., Katsinis, C., & Tozeren, A. (2006). Large-scale computations on histology images reveal grade-differentiating parameters for breast cancer. BMC Medical Imaging, 6, 1-11.

Ramosaj, B., Amro, L., & Pauly, M. (2020). A cautionary tale on using imputation methods for inference in matched-pairs design. Bioinformatics, 36(10), 3099-3106.

Rashmi, R., Prasad, K., & Udupa, C. B. K. (2022). Breast histopathological image analysis using image processing techniques for diagnostic purposes: A methodological review. Journal of Medical Systems, 46(1), 1-7.

Rezazadeh, A., Jafarian, Y., & Kord, A. (2022). Explainable ensemble machine learning for breast cancer diagnosis based on ultrasound image texture features. Forecasting, 4(1), 262-274.

Sarwar, M. A., Kamal, N., Hamid, W., & Shah, M. A. (2018). Prediction of diabetes using machine learning algorithms in healthcare. Paper presented at the 2018 24th international conference on automation and computing (ICAC).

Tamrakar, D., & Ahuja, K. (2017). Density-wise two stage mammogram classification using texture exploiting descriptors. arXiv preprint arXiv:1701.04010.

Wilkinson, L., & Gathani, T. (2022). Understanding breast cancer as a global health concern. The British journal of radiology, 95(1130), 20211033.

Zhang, Y.-D., Wang, S.-H., Liu, G., & Yang, J. (2016). Computer-aided diagnosis of abnormal breasts in mammogram images by weighted-type fractional Fourier transform. Advances in Mechanical Engineering, 8(2), 1687814016634243.

Zhou, S., Hu, C., Wei, S., & Yan, X. (2024). Breast cancer prediction based on multiple machine learning algorithms. Technology in Cancer Research & Treatment, 23, 15330338241234791.

Published

2026-03-31