Applications of Machine Learning Techniques to Predict Diagnostic Breast Cancer

被引:0
作者
Chaurasia V. [1 ]
Pal S. [1 ]
机构
[1] Department of Computer Applications, VBS Purvanchal University, Jaunpur
关键词
Classification; Ensemble; k-Nearest neighbors; Linear regression; Machine learning; Multilayer perceptron; Stack; Support vector machine;
D O I
10.1007/s42979-020-00296-8
中图分类号
学科分类号
摘要
This article compares six machine learning (ML) algorithms: Classification and Regression Tree (CART), Support Vector Machine (SVM), Naïve Bayes (NB), K-Nearest Neighbors (KNN), Linear Regression (LR) and Multilayer Perceptron (MLP) on the Wisconsin Diagnostic Breast Cancer (WDBC) dataset by estimating their classification test accuracy, standardized data accuracy and runtime analysis. The main objective of this study is to improve the accuracy of prediction using a new statistical method of feature selection. The data set has 32 features, which are reduced using statistical techniques (mode), and the same measurements as above are applied for comparative studies. In the reduced attribute data subset (12 features), we applied 6 integrated models AdaBoost (AB), Gradient Boosting Classifier (GBC), Random Forest (RF), Extra Tree (ET) Bagging and Extra Gradient Boost (XGB), to minimize the probability of misclassification based on any single induced model. We also apply the stacking classifier (Voting Classifier) ​​to basic learners: Logistic Regression (LR), Decision Tree (DT), Support-vector clustering (SVC), K-Nearest Neighbors (KNN), Random Forest (RF) and Naïve Bays (NB) to find out the accuracy obtained by voting classifier (Meta level). To implement the ML algorithm, the data set is divided in the following manner: 80% is used in the training phase and 20% is used in the test phase. To adjust the classifier, manually assigned hyper-parameters are used. At different stages of classification, all ML algorithms perform best, with test accuracy exceeding 90% especially when it is applied to a data subset. © 2020, Springer Nature Singapore Pte Ltd.
引用
收藏
相关论文
共 30 条
[1]  
(2019)
[2]  
Luca M., Kleinberg J., Mullainathan S., Algorithms need managers, too, (2016)
[3]  
Coiera E., Guide to medical informatics, the Internet and telemedicine, (1997)
[4]  
Elsayad A.M., Predicting the severity of breast masses with ensemble of Bayesian classifiers, J Comput Sci, 6, 5, pp. 576-584, (2010)
[5]  
Huang M., Hung Y., Chen W., Neural network classifier with entropy based feature selection on breast cancer diagnosis, J Med Syst, 34, pp. 865-873, (2010)
[6]  
Lavanya D., Rani D.K., Analysis of feature selection with classification: Breast cancer datasets, Indian J Comput Sci Eng (IJCSE), 2, 5, pp. 756-763, (2011)
[7]  
Bekaddour F., A neuro-fuzzy inference model for breast cancer recognition, Int J Comput Sci Inf Technol., 4, 5, pp. 163-173, (2012)
[8]  
Al-Bahrani R., Agrawal A., Choudhary A., Colon cancer survival prediction using ensemble mining on SEER data, Proceeding of IEEE International Conference on Big Data, pp. 9-16, (2013)
[9]  
Zheng B., Yoon S.W., Lam S.S., Breast cancer diagnosis based on feature extraction using a hybrid of K-means and support vector machine algorithms, Expert Syst Appl, 41, 4, pp. 1476-1482, (2014)
[10]  
Chaurasia V., Pal S., Data Mining techniques: to predict and resolve breast cancer survivability, IJCSMC, 3, pp. 10-22, (2014)