Data Analytics Approaches for Breast Cancer Survivability: Comparison of Data Mining Methods

摘要

In the early stages of breast cancer, surgery, chemotherapy, and radiotherapy are considered effective methods to remove a cancerous tumor that is detected in the breast area and on the lymph nodes. However, undetected cancer cell remnants on the breast tissue and lymph nodes, inefficient treatment methods, as well as the patient's health condition may impact the patient's lifetime expectancy. In this study, given a set of explanatory variables that include the patient's demographics, health condition, and cancer treatment regimen, our objective is to investigate the performance of four different machine learning methods including an artificial neural network (ANN), classification and regression tree (C&RT), logistic regression, and Bayesian belief network (BBN). We utilize these four methods with a ten-fold cross validation in order to predict the ten-year survivability of a breast cancer patient after initial diagnosis. The results of each method are compared with respect to accuracy, sensitivity, specificity, and area under the curve (AUC) metrics. We observe that the logistic regression method shows better performance compared to the others with respect to the AUC metric. In all prediction models, the stage of the cancer is the most important predictor of breast cancer survivability.