Real-Data Comparison of Data Mining Methods in Prediction of Diabetes in Iran
2
조회
0
다운로드
3
인용
0
참고문헌
키워드
초록
Objectives: Diabetes is one of the most common non-communicable diseases in developing countries. Early screening and diagnosis play an important role in effective prevention strategies. This study compared two traditional classification methods (logistic regression and Fisher linear discriminant analysis) and four machine-learning classifiers (neural networks, support vector machines, fuzzy c-mean, and random forests) to classify persons with and without diabetes. Methods: The data set used in this study included 6,500 subjects from the Iranian national non-communicable diseases risk factors surveillance obtained through a cross-sectional survey. The obtained sample was based on cluster sampling of the Iran population which was conducted in 2005–2009 to assess the prevalence of major non-communicable disease risk factors. Ten risk factors that are commonly associated with diabetes were selected to compare the performance of six classifiers in terms of sensitivity, specificity, total accuracy, and area under the receiver operating characteristic (ROC) curve criteria. Results: Support vector machines showed the highest total accuracy (0.986) as well as area under the ROC (0.979). Also, this method showed high specificity (1.000) and sensitivity (0.820). All other methods produced total accuracy of more than 85%, but for all methods, the sensitivity values were very low (less than 0.350). Conclusions: The results of this study indicate that, in terms of sensitivity, specificity, and overall classification accuracy, the support vector machine model ranks first among all the classifiers tested in the prediction of diabetes. Therefore, this approach is a promising classifier for predicting diabetes, and it should be further investigated for the prediction of other diseases.
Abstract
Objectives: Diabetes is one of the most common non-communicable diseases in developing countries. Early screening and diagnosis play an important role in effective prevention strategies. This study compared two traditional classification methods (logistic regression and Fisher linear discriminant analysis) and four machine-learning classifiers (neural networks, support vector machines, fuzzy c-mean, and random forests) to classify persons with and without diabetes. Methods: The data set used in this study included 6,500 subjects from the Iranian national non-communicable diseases risk factors surveillance obtained through a cross-sectional survey. The obtained sample was based on cluster sampling of the Iran population which was conducted in 2005–2009 to assess the prevalence of major non-communicable disease risk factors. Ten risk factors that are commonly associated with diabetes were selected to compare the performance of six classifiers in terms of sensitivity, specificity, total accuracy, and area under the receiver operating characteristic (ROC) curve criteria. Results: Support vector machines showed the highest total accuracy (0.986) as well as area under the ROC (0.979). Also, this method showed high specificity (1.000) and sensitivity (0.820). All other methods produced total accuracy of more than 85%, but for all methods, the sensitivity values were very low (less than 0.350). Conclusions: The results of this study indicate that, in terms of sensitivity, specificity, and overall classification accuracy, the support vector machine model ranks first among all the classifiers tested in the prediction of diabetes. Therefore, this approach is a promising classifier for predicting diabetes, and it should be further investigated for the prediction of other diseases.
참고문헌 (31)
- "IDF Diabetes Atlas: the global burden" (2013)[URL]
- Priya R, "Review of automated diagnosis of diabetic retinopathy using the support vector machine", Int J Appl Eng Res (Dindigul), 1(4), pp.844~863 (2011)
- Calder R, "Cardiovascular disease in people with diabetes mellitus", Pract Diabetol, 19(4), pp.7~18 (2000)
- Barr EL, "Risk of cardiovascular and all-cause mortality in individuals with diabetes mellitus, impaired fasting glucose, and impaired glucose tolerance: the Australian Diabetes, Obesity, and Lifestyle Study (AusDiab)", Circulation, 116(2), pp.151~157 (2007)
- Pi-Sunyer FX, "How effective are lifestyle changes in the prevention of type 2 diabetes mellitus?", Nutr Rev, 65(3), pp.101~110 (2007)
- IDF Clinical Guidelines Task Force, "Global Guideline for Type 2 Diabetes: recommendations for standard, comprehensive, and minimal care", Diabet Med, 23(6), pp.579~593 (2006)
- Thomas C, "Type 2 diabetes mellitus in midlife estimated from the Cambridge Risk Score and body mass index", Arch Intern Med, 166(6), pp.682~688 (2006)
- Maroco J, "Data mining methods in the prediction of Dementia: a real-data comparison of the accuracy, sensitivity and specificity of linear discriminant analysis, logistic regression, neural networks, support vector machines, classification trees and random forests", BMC Res Notes, 4, pp.299~ (2011)
- Yu W, "Application of support vector machine modeling for prediction of common diseases: the case of diabetes and pre-diabetes", BMC Med Inform Decis Mak, 10, pp.16~ (2010)
- Sujin Kim, "A Comparison of Intensive Care Unit Mortality Prediction Models through the Use of Data Mining Techniques", Healthcare Informatics Research, 17(4), pp.232~243 (2011), 대한의료정보학회
- Soo-Kyoung Lee, "Application of Support Vector Machine for Prediction of Medication Adherence in Heart Failure Patients", Healthcare Informatics Research, 16(4), pp.253~259 (2010), 대한의료정보학회
- 이수경, "Predictors of Medication Adherence in Elderly Patients with Chronic Diseases Using Support Vector Machine Models", Healthcare Informatics Research, 19(1), pp.33~41 (2013), 대한의료정보학회[DOI]
- Lehmann C, "Application and comparison of classification algorithms for recognition of Alzheimer's disease in electrical brain activity (EEG)", J Neurosci Methods, 161(2), pp.342~350 (2007)
- Peyman Rezaei Hachesu, "Use of Data Mining Techniques to Determine and Predict Length of Stay of Cardiac Patients", Healthcare Informatics Research, 19(2), pp.121~129 (2013), 대한의료정보학회[DOI]
- Priya R, "SVM and neural network based diagnosis of diabetic retinopathy", Int J Comput Appl, 41(1), pp.6~12 (2012)
- Finch H, "Classification accuracy of neural networks vs. discriminant analysis, logistic regression, and classification and regression trees: three- and five-group cases", Methodology (Gott), 3(2), pp.47~57 (2007)
- Austin PC, "A comparison of regression trees, logistic regression, generalized additive models, and multivariate adaptive regression splines for predicting AMI mortality", Stat Med, 26(15), pp.2937~2957 (2007)
- Gelnarova E, "Comparison of three statistical classifiers on a prostate cancer data", Neural Netw World, 15(4), pp.311~318 (2005)
- Green M, "Comparison between neural networks and multiple logistic regression to predict acute coronary syndrome in the emergency room", Artif Intell Med, 38(3), pp.305~318 (2006)
- Meyer D, "The support vector machine under test", Neurocomputing, 55(1-2), pp.169~186 (2003)
- Poorolajal J, "Five-year evaluation of chronic diseases in Hamadan, Iran: 2005-2009", Iran J Public Health, 41(3), pp.71~81 (2012)
- Little RJ, "A test of missing completely at random for multivariate data with missing values", J Am Stat Assoc, 83(404), pp.1198~1202 (1988)
- Yoon H, "A comparative study of artificial neural networks and support vector machines for predicting groundwater levels in a coastal aquifer", J Hydrol, 396(1-2), pp.128~138 (2011)
- "The Comprehensive R Archive Network (CRAN) package"[URL]
- Schwarzer G, "On the misuses of artificial neural networks for prognostic and diagnostic classification in oncology", Stat Med, 19(4), pp.541~561 (2000)
- Oliveira PP Jr, "Use of SVM methods with surfacebased cortical and volumetric subcortical measurements to detect Alzheimer's disease", J Alzheimers Dis, 19(4), pp.1263~1272 (2010)
- Zhu Y, "Feature selection and performance evaluation of support vector machine (SVM)-based classifier for differentiating benign and malignant pulmonary nodules by computed tomography", J Digit Imaging, 23(1), pp.51~65 (2010)
- Abbasimehr H, "A neuro-fuzzy classifier for customer churn prediction", Int J Comput Appl, 19(8), pp.35~41 (2011)
- Breiman L, "Random forests", Mach Learn, 45(1), pp.5~32 (2001)
- Smith A, "Novel application of a statistical technique, random forests, in a bacterial source tracking study", Water Res, 44(14), pp.4067~4076 (2010)
- Support vector machines (SVM) as a technique for solvency analysis, "Auria L", Deutsches Institut fur Wirtschaftsforschung, ~ (2008)