Neural network and support vector machine for the prediction of chronic kidney disease: A comparative study.
Journal:
Computers in biology and medicine
Published Date:
Apr 25, 2019
Abstract
This paper aims to assist in the prevention of Chronic Kidney Disease (CKD) by utilizing machine learning techniques to diagnose CKD at an early stage. Kidney diseases are disorders that disrupt the normal function of the kidney. As the percentage of patients affected by CKD is significantly increasing, effective prediction procedures should be considered. In this paper, we focus on applying different machine learning classification algorithms to a dataset of 400 patients and 24 attributes related to diagnosis of chronic kidney disease. The classification techniques used in this study include Artificial Neural Network (ANN) and Support Vector Machine (SVM). To perform experiments, all missing values in the dataset were replaced by the mean of the corresponding attributes. Then, the optimized parameters for the Artificial Neural Network (ANN) and Support Vector Machine (SVM) techniques were determined by tuning the parameters and performing several experiments. The final models of the two proposed techniques were developed using the best-obtained parameters and features. The empirical results from the experiments indicated that ANN performed better than SVM, with accuracies of 99.75% and 97.75%, respectively, indicating that the outcome of this study is very promising.