SASSO: Design of a Signature-Based Feature Selection Model with Ensemble Deep Learning Using Bi-LSTM and Bi-GRU for Chronic Kidney Disease Classification

Authors

  • Anindita Khade, Avaneesh Karthikeyan Iyer

Keywords:

: Machine Learning, Deep Learning, Feature Selection, Bi-LSTM, Bi-GRU, SES, LASSO

Abstract

Chronic kidney disease (CKD) involves numerous variables, but only a few are significant for classification. The SES method, inspired by constraint-based learning in Bayesian networks, identifies essential features in CKD. Unlike traditional feature selection methods, which concentrate on a single set of features with the greatest predictive potential, the SES method can identify multiple predictive feature subsets with comparable performance. Most feature selection (FS) classifiers work better with strongly correlated data, making FS challenging in high-throughput data for finding important features and choosing the best classifier. This study is conducted on a real time data available from hospitals. This study suggests the use of the Least Absolute Shrinkage and Selection Operator (LASSO) in conjunction with the SES method, which is abbreviated as SASSO, to identify CKD features. Subsequently, a combination of Bidirectional Long Short-Term Memory (Bi-LSTM) and Bidirectional Gated Recurrent Unit (Bi-GRU) ensemble deep-learning models is proposed for CKD classification. The model's performance is measured using accuracy, precision, recall, and the F1-score. The experimental results are compared to individual classifiers, including Random Forest (RF), Naïve Bayes (NB), XGBoost (XGB), and Artificial Neural Networks (ANN).

The findings show a 6% improvement in classification accuracy with the proposed hybrid feature selection approach and the Bi-LSTM-Bi-GRU ensemble model.

Downloads

Download data is not yet available.

References

Khade, A. A., Vidhate, A. V., & Vidhate, D. A Comparative Analysis of Applied AI Techniques for an Early Prediction of Chronic Kidney Disease. Proceedings - 2nd International Conference on Smart Electronics and Communication, ICOSEC 2021; 1386–1392. https://doi.org/10.1109/ICOSEC51865.2021.9591869

Chittora, P., Chaurasia, S., Chakrabarti, P., Kumawat, G., Chakrabarti, T., Leonowicz, Z., Jasinski, M., Jasinski, L., Gono, R., Jasinska, E., & Bolshev, V. Prediction of Chronic Kidney Disease - A Machine Learning Perspective. IEEE Access, 9(January),2021; 17312–17334. https://doi.org/10.1109/ACCESS.2021.3053763

Dare, A. J., Fu, S. H., Patra, J., Rodriguez, P. S., Thakur, J. S., & Jha, P. Renal failure deaths and their risk factors in India 2001–13: nationally representative estimates from the Million Death Study. The Lancet Global Health, 5(1), 2017; e89–e95. https://doi.org/10.1016/S2214-109X(16)30308- 4

Ifraz, G. M., Rashid, M. H., Tazin, T., Bourouis, S., & Khan, M. M. Comparative Analysis for Prediction of Kidney Disease Using Intelligent Machine Learning Methods. Computational and Mathematical Methods in Medicine, 2021; https://doi.org/10.1155/2021/6141470

Yuan, Q., Zhang, H., Deng, T., Tang, S., Yuan, X., Tang, W., Xie, Y., Ge, H., Wang, X., Zhou, Q., & Xiao, X.. Role of artificial intelligence in kidney disease. International Journal of Medical Sciences, 17(7),2020; 970–984. https://doi.org/10.7150/ijms.42078

Bhaskar, N., & Suchetha, M. A Computationally Efficient Correlational Neural Network for Automated Prediction of Chronic Kidney Disease. IRBM, 42(4), 2020;268–276. https://doi.org/10.1016/j.irbm.2020.07.002

Muslim, M. A., Kurniawati, I., & Sugiharti, E. Expert system diagnosis chronic kidney disease based on Mamdani fuzzy inference system. Journal of Theoretical and Applied Information Technology,2015; 78(1), 70–75.

[8] Hu, D., Nie, F., & Li, X. Deep linear discriminant analysis hashing. Scientia Sinica Information is 51(2), 2021;279–293. https://doi.org/10.1360/SSI-2019-0175

[9] Singh, V., Asari, V. K., & Rajasekaran, R. A Deep Neural Network for Early Detection and Prediction of Chronic Kidney Disease. Diagnostics, 2022; 12(1),1–22.https://doi.org/10.3390/diagnostics12010116

Almansour, N. A., Syed, H. F., Khayat, N. R., Altheeb, R. K., Juri, R. E., Alhiyafi, J., Alrashed, S., & Olatunji, S. O. Neural network and support vector machine for the prediction of chronic kidney disease: A comparative study. Computers in Biology and Medicine, 2019; 109(April), 101–111. https://doi.org/10.1016/j.compbiomed.2019.04.017

Senan, E. M., Al-Adhaileh, M. H., Alsaade, et al. Diagnosis of Chronic Kidney Disease Using Effective Classification Algorithms and Recursive Feature Elimination Techniques. Journal of Healthcare Engineering, 2021;. https://doi.org/10.1155/2021/1004767

Rady, E. H. A., & Anwar, A. S. Prediction of kidney disease stages using data mining algorithms. Informatics in Medicine Unlocked, 15(December 2018), 2019; 100178. https://doi.org/10.1016/j.imu.2019.100178

Khamparia, A., Saini, G., Pandey, B., Tiwari, S., Gupta, D., & Khanna, A. KDSAE: Chronic kidney disease classification with multimedia data learning using deep stacked autoencoder network. Multimedia Tools and Applications, 79(47–48), 2020; 35425–35440. https://doi.org/10.1007/s11042-019-07839-z

Song, X., Waitman, L. R., Yu, A. S. L., et al, Longitudinal risk prediction of chronic kidney disease in diabetic patients using a temporal-enhanced gradient boosting machine: Retrospective cohort study. JMIR Medical Informatics, 2020;8(1), 1–16. https://doi.org/10.2196/15510

Pasadana, I. A., Hartama, D., Zarlis, M., et al. Chronic Kidney Disease Prediction by Using Different Decision Tree Techniques. Journal of Physics: Conference Series,2019; 1255(1). https://doi.org/10.1088/1742-6596/1255/1/012024

Krishnamurthy, S., Kapeleshh, K. S., Dovgan, E., et al. Machine learning prediction models for chronic kidney disease using national health insurance claim data in Taiwan. Healthcare (Switzerland), 2021;9(5), 1–13. https://doi.org/10.3390/healthcare9050546

Neves, J., Martins, M. R., Vilhena, J., Neves, J., Gomes, S., Abelha, A., Machado, J., & Vicente,

H. A Soft Computing Approach to Kidney Diseases Evaluation. Journal of Medical Systems, 2015;39(10). https://doi.org/10.1007/s10916-015-0313-4

Fokas, A. S., Dikaios, N., & Kastis, G. A. Mathematical models and deep learning for predicting the number of individuals reported to be infected with SARS-CoV-2. Journal of the Royal Society Interface,2020; 17(169). https://doi.org/10.1098/rsif.2020.0494

Chen, X., & Jeong, J. C. Enhanced Recursive Feature Elimination. January 2008.2014; https://doi.org/10.1109/ICMLA.2007.35

Shastri, S., Kour, P., Kumar, S., Singh, K., & Mansotra, V. XGBoost: A novel Grading- Ada Boostensemble approach for automatic identification of erythemato-squamous disease. International Journal of Information Technology (Singapore), 2021; 13(3), 959–971. https://doi.org/10.1007/s41870-020-00589-4

Khanday, A. M. U. D., Rabani, S. T., Khan, Q. R., Rouf, N., & Mohi Ud Din, M. Machine learning-based approaches for detecting COVID-19 using clinical text data. International Journal of Information Technology (Singapore), 2020;12(3), 731–739. https://doi.org/10.1007/s41870-020-00495-9

Dasari, S. K., & Prasad, V. A novel and proposed comprehensive methodology using deep convolutional neural networks for flue-cured tobacco leaves classification. International Journal of Information Technology (Singapore), 2019;11(1), 107–117. https://doi.org/10.1007/s41870-018-0174-4

[23] Khade, A. A. and Vidhate, A. V. Application of Artificial Intelligence Techniques in the Early-Stage Detection of Chronic Kidney Disease, Data Science-Techniques and Intelligent Applications(NewYork), 2022.

H.-C. Lee et al., “Prediction of Acute Kidney Injury after Liver Transplantation: Machine Learning Approaches vs. Logistic Regression Model,” Journal of Clinical Medicine, vol. 7, no. 11, p. 428, Nov. 2018, doi: 10.3390/jcm7110428

Elhoseny, M.; Shankar, K.; Uthayakumar, J. Intelligent diagnostic prediction and classification system for chronic kidney disease. Sci. Rep. 2019, 9, 9583. [CrossRef] [PubMed].

Vasquez-Morales, G.R.; Martinez-Monterrubio, S.M.; Moreno-Ger, P.; Recio-Garcia, J.A. Explainable Prediction of Chronic Renal Disease in the Colombian Population Using Neural Networks and Case-Based Reasoning. IEEE Access 2019, 7, 152900–152910.

Senan, E.M.; Al-Adhaileh, M.H.; Alsaade, F.W.; Aldhyani, T.H.H.; Alqarni, A.A.; Alsharif, N.; Uddin, M.I.; Alahmadi, A.H.;

Jadhav, M.E.; Alzahrani, M.Y. Diagnosis of Chronic Kidney Disease Using Effective Classification Algorithms and Recursive

Feature Elimination Techniques. J. Healthc. Eng. 2021, 2021, 1004767.

Krishnamurthy, S.; Ks, K.; Dovgan, E.; Luštrek, M.; Piletiˇc, B.G.; Srinivasan, K.; Li, Y.-C.; Gradišek, A.; Syed-Abdul, S. Machine learning prediction models for chronic kidney disease using national health insurance claim data in Taiwan. Healthcare 2021, 9, 546.

Polat, H.; Mehr, H.D.; Cetin, A. Diagnosis of chronic kidney disease based on support vector machine by feature selection methods. J. Med. Syst. 2020, 41, 55..

S. K. Agarwal, S. C. Dash, M. Irshad, S. Raju, R. Singh, and R. M. Pandey, “Prevalence of chronic renal failure in adults in Delhi, India,” Nephrology Dialysis Transplantation, vol. 20, no. 8, pp. 1638–1642, Apr. 2005, doi: 10.1093/ndt/gfh855.

S. Singh, S. Shreevastava, T. Som, and G. Somani, “A fuzzy similarity-based rough set approach for attribute selection in set valued information systems,” Soft Computing, vol. 24, no. 6, pp. 4675–4691, Jul. 2019, doi: 10.1007/s00500-019-04228-4.

Khade, A., Vidhate, A.V. & Vidhate, D. FFN-XGB- design of a hybrid feed forward neural network and extreme gradient boosting model for early prediction of chronic kidney disease. Int J Syst Assur Eng Manag 2023. https://doi.org/10.1007/s13198-023-01993-2

Khade, A.., Vidhate, A. V., & Vidhate, D., Design of an Optimized Self-Acclimation Graded Boolean PSO with Back Propagation Model and Cuckoo Search Heuristics for Automatic Prediction of Chronic Kidney Disease. Journal of Mobile Multimedia, 19(06), 1395–1414. https://doi.org/10.13052/jmm1550-4646.1962

Downloads

Published

09.07.2024

How to Cite

Anindita Khade. (2024). SASSO: Design of a Signature-Based Feature Selection Model with Ensemble Deep Learning Using Bi-LSTM and Bi-GRU for Chronic Kidney Disease Classification. International Journal of Intelligent Systems and Applications in Engineering, 12(22s), 1435 –. Retrieved from https://ijisae.org/index.php/IJISAE/article/view/6665

Issue

Section

Research Article