Predictive Analysis of Chronic Kidney Disease using Explainable Ensemble Learning Models
DOI:
https://doi.org/10.63503/acset.118Keywords:
CKD, Machine Learning Ensemble models, Explainable AI, InterpretabilityAbstract
Chronic kidney disease (CKD) progresses and is potentially fatal. Early detection of CKD is essential to prevent severe effects and kidney failure. This work provides an intelligent and comprehensible method of predicting CKD using ML and ensemble methods. It works with the CKD dataset from the UCI ML Repository, comprising 400 patient records from Apollo Hospital in India and containing 25 numeric and nominal attributes. We apply statistical and model-based imputation to replace missing values, and SMOTE to address class imbalance (CKD vs. non-CKD). We identified six clinically significant items, including haemoglobin, serum creatinine, albumin, high blood pressure, age, and diabetes mellitus. A combination of preprocessing, feature engineering, and feature selection techniques, such as Mutual Information, Variance Thresholding, RFE, and Sequential Feature Selection, was used to identify these. Various models are evaluated using Stratified K-Fold cross-validation. These are LR, Gaussian NB, SVM, DT, RF and AdaBoost. A mixed Voting Classifier composed of RF and AdaBoost achieves an F1-score of 99.3% and 99.4% accuracy, precision, and recall. The model is easier to comprehend because individual forecasts and feature contributions are explained using LIME and SHAP. The web application, developed using Flask, allows users to register and log in, enter data securely, make real-time predictions, and receive interpretable results. It provides outcomes such as “CKD detected” and “no CKD detected” that help doctors make informed decisions
References
[1] P. A. Moreno-Sanchez, “Data-Driven Early Diagnosis of Chronic Kidney Disease: Development and Evaluation of an Explainable AI Model,” IEEE Access, vol. 11, pp. 38359–38369, 2023, doi: 10.1109/ACCESS.2023.3264270.
[2] K. M. T. Jawad, A. Verma, F. Amsaad, and L. Ashraf, “AI-Driven Predictive Analytics Approach for Early Prognosis of Chronic Kidney Disease Using Ensemble Learning and Explainable AI,” Jan. 2025. [Online]. Available: http://arxiv.org/abs/2406.06728
[3] S. K. Ghosh and A. H. Khandoker, “Investigation on explainable machine learning models to predict chronic kidney diseases,” Scientific Reports, vol. 14, no. 1, Dec. 2024, doi: 10.1038/s41598-024-54375-4.
[4] K. Jhumka, M. M. Auzine, M. Heenaye-Mamode Khan, S. M. Casseem, S. A. Fedally, and Z. Mungloo-Dilmohamud, “Explainable Chronic Kidney Disease (CKD) Prediction using Deep Learning and Shapley Additive Explanations (SHAP),” in ACM International Conference Proceeding Series, Association for Computing Machinery, Oct. 2023, pp. 29–33, doi: 10.1145/3633598.3633604.
[5] S. K. Ghosh, N. Widatalla, and A. H. Khandoker, “Machine Learning Framework for Early Detection of Chronic Kidney Disease Stages Using Optimized Estimated Glomerular Filtration Rate,” IEEE Access, vol. 13, pp. 78057–78072, 2025, doi: 10.1109/ACCESS.2025.3565549.
[6] M. V. N., A. N., and K. S. K., “Decision Tree-Based Explainable AI for Diagnosis of Chronic Kidney Disease,” in 2023 5th International Conference on Inventive Research in Computing Applications (ICIRCA), 2023, pp. 947–952, doi: 10.1109/ICIRCA57980.2023.10220774.
[7] F. Sanmarchi, C. Fanconi, D. Golinelli, D. Gori, T. Hernandez-Boussard, and A. Capodici, “Predict, diagnose, and treat chronic kidney disease with machine learning: a systematic literature review,” May 01, 2023, Springer Science and Business Media Deutschland GmbH, doi: 10.1007/s40620-023-01573-4.
[8] A. Holzinger, C. Biemann, C. S. Pattichis, and D. B. Kell, “What do we need to build explainable AI systems for the medical domain?,” Dec. 2017. [Online]. Available: http://arxiv.org/abs/1712.09923
[9] A. BAYRAM, C. Gurkan, A. BUDAK, and H. KARATAŞ, “A Detection and Prediction Model Based on Deep Learning Assisted by Explainable Artificial Intelligence for Kidney Diseases,” European Journal of Science and Technology, Mar. 2022, doi: 10.31590/ejosat.1171777.
[10] A. C. M. da Silveira, Á. Sobrinho, L. D. da Silva, E. de B. Costa, M. E. Pinheiro, and A. Perkusich, “Exploring Early Prediction of Chronic Kidney Disease Using Machine Learning Algorithms for Small and Imbalanced Datasets,” Applied Sciences (Switzerland), vol. 12, no. 7, Apr. 2022, doi: 10.3390/app12073673.
[11] N. Mehrabi, F. Morstatter, N. Saxena, K. Lerman, and A. Galstyan, “A Survey on Bias and Fairness in Machine Learning,” Jan. 2022. [Online]. Available: http://arxiv.org/abs/1908.09635
[12] P. Ventrella, G. Delgrossi, G. Ferrario, M. Righetti, and M. Masseroli, “Supervised machine learning for the assessment of Chronic Kidney Disease advancement,” Computer Methods and Programs in Biomedicine, vol. 209, p. 106329, 2021, doi: 10.1016/j.cmpb.2021.106329.
[13] M. Nishat et al., “A Comprehensive Analysis on Detecting Chronic Kidney Disease by Employing Machine Learning Algorithms,” EAI Endorsed Transactions on Pervasive Health and Technology, pp. 1–12, Mar. 2021, doi: 10.4108/eai.13-8-2021.170671.
[14] P. Chittora et al., “Prediction of Chronic Kidney Disease - A Machine Learning Perspective,” IEEE Access, vol. 9, pp. 17312–17334, 2021, doi: 10.1109/ACCESS.2021.3053763.
[15] D. Dave, H. Naik, S. Singhal, and P. Patel, “Explainable AI meets Healthcare: A Study on Heart Disease Dataset,” Nov. 2020. [Online]. Available: http://arxiv.org/abs/2011.03195
[16] P. Ghosh, F. M. Javed Mehedi Shamrat, S. Shultana, S. Afrin, A. A. Anjum, and A. A. Khan, “Optimization of Prediction Method of Chronic Kidney Disease Using Machine Learning Algorithm,” in 2020 15th International Joint Symposium on Artificial Intelligence and Natural Language Processing (iSAI-NLP), 2020, pp. 1–6, doi: 10.1109/iSAI-NLP51646.2020.9376787.
[17] A. J. Aljaaf et al., “Early Prediction of Chronic Kidney Disease Using Machine Learning Supported by Predictive Analytics,” in 2018 IEEE Congress on Evolutionary Computation (CEC), 2018, pp. 1–9, doi: 10.1109/CEC.2018.8477876.
[18] A. C. Webster, E. V. Nagler, R. L. Morton, and P. Masson, “Chronic Kidney Disease,” The Lancet, vol. 389, no. 10075, pp. 1238–1252, Mar. 2017, doi: 10.1016/S0140-6736(16)32064-5.
[19] M. Chen, Y. Hao, K. Hwang, L. Wang, and L. Wang, “Disease Prediction by Machine Learning Over Big Data From Healthcare Communities,” IEEE Access, vol. 5, pp. 8869–8879, 2017, doi: 10.1109/ACCESS.2017.2694446.
[20] Elslvler, I. D. Penman, S. H. Ralston, M. W. J. Strachan, and R. P. Hobson, “24th Edition.”
[21] T. J. Hoerger et al., “The future burden of CKD in the United States: A simulation model for the CDC CKD initiative,” American Journal of Kidney Diseases, vol. 65, no. 3, pp. 403–411, Mar. 2015, doi: 10.1053/j.ajkd.2014.09.023.
[22] H.-Y. Kim, “Statistical notes for clinical researchers: Chi-squared test and Fisher’s exact test,” Restorative Dentistry & Endodontics, vol. 42, no. 2, p. 152, 2017, doi: 10.5395/rde.2017.42.2.152.
[23] E. Seker, J. R. Talburt, and M. L. Greer, “Preprocessing to Address Bias in Healthcare Data,” in Studies in Health Technology and Informatics, IOS Press BV, May 2022, pp. 327–331, doi: 10.3233/SHTI220468.
Downloads
Published
Conference Proceedings Volume
Section
License
Copyright (c) 2026 Adroid Conference Series: Engineering and Technology
This work is licensed under a Creative Commons Attribution 4.0 International License.
