Early Prediction of Chronic Kidney Disease: A Comprehensive Performance Analysis of Deep Learning Models

Chronic kidney disease (CKD) is one of the most life-threatening disorders. To improve survivability, early discovery and good management are encouraged. In this paper, CKD was diagnosed using multiple optimized neural networks against traditional neural networks on the UCI machine learning dataset, to identify the most efficient model for the task. The study works on the binary classification of CKD from 24 attributes. For classification, optimized CNN (OCNN), ANN (OANN), and LSTM (OLSTM) models were used as well as traditional CNN, ANN, and LSTM models. With various performance matrixes, error measures, loss values, AUC values, and compilation time, the implemented models are compared to identify the most competent model for the classification of CKD. It is observed that, overall, the optimized models have better performance compared to the traditional models. The highest validation accuracy among the tradition models were achieved from CNN with 92.71%, whereas OCNN, OANN, and OLSTM have higher accuracies of 98.75%, 96.25%, and 98.5%, respectively. Additionally, OCNN has the highest AUC score of 0.99 and the lowest compilation time for classification with 0.00447 s, making it the most efficient model for the diagnosis of CKD.

[1]  Khan Md. Hasib,et al.  SkinNet-16: A deep learning approach to identify benign and malignant skin lesions , 2022, Frontiers in Oncology.

[2]  Suliman A. Alsuhibany,et al.  Ensemble of Deep Learning Based Clinical Decision Support System for Chronic Kidney Disease Diagnosis in Medical Internet of Things Environment , 2021, Comput. Intell. Neurosci..

[3]  Asif Karim,et al.  A Performance Based Study on Deep Learning Algorithms in the Effective Prediction of Breast Cancer , 2021, 2021 International Joint Conference on Neural Networks (IJCNN).

[4]  Yuan Zhang,et al.  An Early Prediction Model for Chronic Kidney Disease , 2021, SSRN Electronic Journal.

[5]  Xinlin Tang,et al.  Multi-disease prediction using LSTM recurrent neural networks , 2021, Expert Syst. Appl..

[6]  Aman Verma,et al.  Chronic Kidney Disease Prediction Using Artificial Neural Network , 2020, Proceedings of International Conference on Big Data, Machine Learning and their Applications.

[7]  Iliyas Ibrahim Iliyas,et al.  Prediction of Chronic Kidney Disease Using Deep Neural Network , 2020, FUDMA JOURNAL OF SCIENCES.

[8]  Aditya Khamparia,et al.  KDSAE: Chronic kidney disease classification with multimedia data learning using deep stacked autoencoder network , 2019, Multimedia Tools and Applications.

[9]  Tao Sun,et al.  Detection and diagnosis of chronic kidney disease using deep learning-based heterogeneous modified artificial neural network , 2020, Future Gener. Comput. Syst..

[10]  M. Suchetha,et al.  A Computationally Efficient Correlational Neural Network for Automated Prediction of Chronic Kidney Disease , 2020 .

[11]  Chalumuru Suresh,et al.  A Neural Network based Model for Predicting Chronic Kidney Diseases , 2020, 2020 Second International Conference on Inventive Research in Computing Applications (ICIRCA).

[12]  S. Bandyopadhyay Chronic Kidney Disease Prediction Using Neural Approach , 2020, medRxiv.

[13]  Jamal Alhiyafi,et al.  Neural network and support vector machine for the prediction of chronic kidney disease: A comparative study , 2019, Comput. Biol. Medicine.

[14]  B.V. Ravindra,et al.  Chronic Kidney Disease Detection Using Back Propagation Neural Network Classifier , 2018, 2018 International Conference on Communication, Computing and Internet of Things (IC3IoT).

[15]  B. Surarso,et al.  Chronic Kidney Disease Diagnosis System using Sequential Backward Feature Selection and Artificial Neural Network , 2021, E3S Web of Conferences.