Search

Breast cancer is one of the leading causes of death among women worldwide. However, early prediction of breast cancer plays a crucial role. Therefore, strong needs exist for automatic accurate early prediction of breast cancer. In this paper, machine learning (ML) classifiers combined with features selection methods are used to build an intelligent tool for breast cancer prediction. The Wisconsin diagnostic breast cancer (WDBC) dataset is used to train and test the model. Classification algorithms, including support vector machine (SVM), light gradient boosting machine (LightGBM), random forest (RF), logistic regression (LR), k-nearest neighbors (k-NN), and naïve Bayes, were employed. Performance measures for each of them were obtained, namely: accuracy, precision, recall, F-score, Kappa, Matthews correlation coefficient (MCC), and time. The results indicate that without feature selection, LightGBM achieves the highest accuracy at 95%. With minimum redundancy maximum relevance (mRMR) feature selection (15 features), LightGBM outperforms other classifiers, achieving an accuracy of 98%. For Pearson correlation coefficient feature selection (15 features), LightGBM also excels with a 95% accuracy rate. Lasso feature selection (5 features) produces varied results across classifiers, with logistic regression achieving the highest accuracy at 96%. These findings underscore the importance of feature selection in refining model performance and in improving detection for breast cancer. [ABSTRACT FROM AUTHOR]

• A novel integrated framework for the diagnosis of breast cancer patients is proposed. • A scheme combining feature selection and hyperparameter optimization is proposed to improve the prediction accuracy of the model. • Compared with the existing models, the proposed model has higher performance and generalization ability. In this paper, a new integrated scheme is proposed to accurately predict breast cancer, help doctors make early diagnosis and treatment plans, and improve the prognosis of patients. We selects five mainstream machine learning models: support vector machine (SVM), artificial neural network (ANN), random forest (RF), extreme gradient boosting (XGBoost) and adaptive boosting (AdaBoost). The Wisconsin Breast Cancer Database (WBCD) and Wisconsin Diagnostic Breast Cancer (WDBC) are used as datasets to investigate the predictive performance of these single and ensemble models. Then, we use multiple linear regression method for feature selection (FS), the experimental results show that the change of feature subset will significantly affect the performance of the model. The recall and f1-score of the five models are improved by 1.19% and 0.84% on average. After that, we apply whale optimization algorithm (WOA) to optimize the hyperparameters of the model to improve their prediction performance. In the best-case scenario, the model demonstrated improvements of 1.02% in accuracy and 1.82% in precision. In addition, we ensemble these models by stacking, investigate the performance changes of the ensemble model when different models are used as meta learners. Finally, the FS-WOA-Stacking model achieves 99.56% accuracy on WBCD and 99.65% accuracy on WDBC. Compared with the existing breast cancer prediction models, the performance of the proposed model is at an excellent level. [ABSTRACT FROM AUTHOR]