vix.ing · top · new · best · stats · spec

A Machine Learning Framework for Breast Cancer Treatment Classification Using a Novel Dataset

2025/06/23 by Md Nahid Hasan, Hasan, Md Nahid, Md Monzur Murshed +4
Biochemistry, Genetics and Molecular Biology · Computer Science · Medicine · #AI in cancer detection #Applications (stat.AP) #Breast Cancer Treatment Studies #FOS: Computer and information sciences #Machine Learning (cs.LG) #Radiomics and Machine Learning in Medical Imaging

paper · pdf · doi:10.48550/arxiv.2507.06243

openalex publication_date 2025/06/23 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Breast cancer (BC) remains a significant global health challenge, with personalized treatment selection complicated by the disease's molecular and clinical heterogeneity. BC treatment decisions rely on various patient-specific clinical factors, and machine learning (ML) offers a powerful approach to predicting treatment outcomes. This study utilizes The Cancer Genome Atlas (TCGA) breast cancer clinical dataset to develop ML models for predicting the likelihood of undergoing chemotherapy or hormonal therapy. The models are trained using five-fold cross-validation and evaluated through performance metrics, including accuracy, precision, recall, specificity, sensitivity, F1-score, and area under the receiver operating characteristic curve (AUROC). Model uncertainty is assessed using bootstrap techniques, while SHAP values enhance interpretability by identifying key predictors. Among the tested models, the Gradient Boosting Machine (GBM) achieves the highest stable performance (accuracy = 0.7718, AUROC = 0.8252), followed by Extreme Gradient Boosting (XGBoost) (accuracy = 0.7557, AUROC = 0.8044) and Adaptive Boosting (AdaBoost) (accuracy = 0.7552, AUROC = 0.8016). These findings underscore the potential of ML in supporting personalized breast cancer treatment decisions through data-driven insights.

Citations

Related