Breast cancer is the most common cancer in women. It develops slowly and silently, making early detection crucial for effective treatment. Machine learning provides models for predicting breast cancer based on data from previously diagnosed patients. The collected data are not all equally important, some features contribute more effectively to predict breast cancer better, while others slow down the model and reduce its performance, highlighting the importance of feature selection. Our proposed approach adopts the Weighted Soft Voting classifier, using XGBoost, Random Forest, and Decision Tree classifiers for feature selection. First, hyperparameter optimization is performed to select the best hyperparameters for each classifier that will participate in the voting. Weights are then assigned to the three classifiers based on their individual performance. Next, the Weighted Soft Voting classifier is used to evaluate the importance of the features. Finally, a search for the optimal number of features to keep is performed, and a dataset with the selected features is obtained. Our feature selection approach enhances the performance of the Weighted Soft Voting classifier, as well as the individual performance of XGBoost and Decision Tree.

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

Feature Selection Using a Weighted Soft Voting Classifier for Breast Cancer Prediction

  • Bounzid Khadija,
  • Mohamed Ben Salah

摘要

Breast cancer is the most common cancer in women. It develops slowly and silently, making early detection crucial for effective treatment. Machine learning provides models for predicting breast cancer based on data from previously diagnosed patients. The collected data are not all equally important, some features contribute more effectively to predict breast cancer better, while others slow down the model and reduce its performance, highlighting the importance of feature selection. Our proposed approach adopts the Weighted Soft Voting classifier, using XGBoost, Random Forest, and Decision Tree classifiers for feature selection. First, hyperparameter optimization is performed to select the best hyperparameters for each classifier that will participate in the voting. Weights are then assigned to the three classifiers based on their individual performance. Next, the Weighted Soft Voting classifier is used to evaluate the importance of the features. Finally, a search for the optimal number of features to keep is performed, and a dataset with the selected features is obtained. Our feature selection approach enhances the performance of the Weighted Soft Voting classifier, as well as the individual performance of XGBoost and Decision Tree.