Cost-Sensitive Extreme Gradient Boosting for Imbalanced Classification of Breast Cancer Diagnosis

Abstract

The clinical information can enhance the doctors for predicting and diagnosing the diseases also making the right decisions. Breast cancer is the most dangerous disease, early diagnosis can improve a chance of survival and can support clinical treatment. Detecting breast cancer takes a lot of time and it is hard to classification. However, the problem of the classification occurs when there is an unequal distribution of classes the dataset. This is caused by the low performance in the traditional machine learning models. For this reason, this work proposed the cost-sensitive XGBoost model, which is an improved version of the XGBoost model in conjunction with cost-sensitive learning. The models were applied to classify the four breast cancer datasets that contained the imbalanced data. In the experiment, this work determined the best parameters on each dataset by the hyperparameters optimization technique before configuring the models. The results indicated that the cost-sensitive XGBoost model had been skillful, and could improve classification accuracy in four datasets. In addition, this work evaluated the model performance by accuracy, ROC AUC, and k- Fold cross-validation to ensure that the new models is accurate.