Breast cancer is one of the most prevalent oncological diseases worldwide, and it has a significant impact on public health and the quality of life of those affected. The objective of this study was to identify the stages of breast cancer using Machine Learning algorithms. The proposed methodology consists of 5 phases: obtaining dataset; preprocessing (One-hot Encoding; Label Encoding; Data Scaling; Data Imputation, Data Balancing and Feature Engineering) and model implementation (SVM, RF, DT, KNN, RL and XGBoost); evaluation (Accuracy, Recall, Precision, F1-Score, Specificity, ROC and Confusion Matrix) and validation. Superior results were obtained with the SVM model, with an Accuracy of 82.72%, Specificity of 85.11%, and the following hyperparameters: C (1), gamma (scale), and kernel (linear). The results demonstrate that it is possible to identify breast cancer stages using Machine Learning models.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Robust Breast Cancer Detection Model Based on Machine Learning

  • Rubén Ruiz,
  • Glem Ramos,
  • Wilfredo Ticona

摘要

Breast cancer is one of the most prevalent oncological diseases worldwide, and it has a significant impact on public health and the quality of life of those affected. The objective of this study was to identify the stages of breast cancer using Machine Learning algorithms. The proposed methodology consists of 5 phases: obtaining dataset; preprocessing (One-hot Encoding; Label Encoding; Data Scaling; Data Imputation, Data Balancing and Feature Engineering) and model implementation (SVM, RF, DT, KNN, RL and XGBoost); evaluation (Accuracy, Recall, Precision, F1-Score, Specificity, ROC and Confusion Matrix) and validation. Superior results were obtained with the SVM model, with an Accuracy of 82.72%, Specificity of 85.11%, and the following hyperparameters: C (1), gamma (scale), and kernel (linear). The results demonstrate that it is possible to identify breast cancer stages using Machine Learning models.