Optimizing Early Diagnosis of Chronic Kidney Disease: A Machine Learning-Based Predictive Model

Abstract: Chronic Kidney Disease (CKD) is a progressive condition that often remains undetected until its later stages, leading to severe health complications and increased mortality. Therefore, this study presents the implementation of a machine learning-based system for early CKD prediction using the logistic regression algorithm.

The study addresses the problem of delayed CKD diagnosis due to vague early symptoms and limited screening efficiency in traditional clinical workflows. The development of the system utilizes a clinical dataset from the UCI Machine Learning Repository made up of 400 patient records with 24 features, including demographic, clinical and laboratory parameters. Data preprocessing techniques were employed including label encoding, imputation of missing values and feature scaling for improving the quality of the data. Then, feature selection was conducted using the SelectKBest method with mutual information to identify the most relevant predictors. The logistic regression model was trained and evaluated using an 80:20 data split and the result of the implementation presents that the model achieved an accuracy of 97.5%, precision of 96.7%, recall of 98.2%, F1-score of 97.4% and a ROC-AUC score of 0.99. This work underscores the effectiveness of logistic regression in medical diagnostics and highlights the value of machine learning in facilitating early detection and timely treatment of CKD.

Chronic Kidney Disease (CKD), Logistic Regression, Machine Learning, Medical Diagnosis, Feature Selection

1. Arif, M. S., Rehman, A. U., & Asif, D. (2024). Explainable machine learning model for chronic kidney disease prediction. Algorithms, 17(10), 443. https://doi.org/10.3390/a17100443 [Google Scholar] [Crossref]

2. Asif, D., Bibi, M., Arif, M. S., & Mukheimer, A. (2023). Enhancing heart disease prediction through ensemble learning techniques with hyperparameter optimization. Algorithms, 16(6), 308. https://doi.org/10.3390/a16060308 [Google Scholar] [Crossref]

3. Bellazzi, R., & Zupan, B. (2008). Predictive data mining in clinical medicine: Current issues and guidelines. International Journal of Medical Informatics, 77(2), 81–97. https://doi.org/10.1016/j.ijmedinf.2006.11.006 [Google Scholar] [Crossref]

4. Beretta, L., & Santaniello, A. (2016). Nearest neighbour imputation algorithms: A critical evaluation. BMC Medical Informatics and Decision Making, 16, Article 197. https://doi.org/10.1186/s12911-016-0405-2 [Google Scholar] [Crossref]

5. Burnier, M., & Damianaki, A. (2023). Hypertension as cardiovascular risk factor in chronic kidney disease. Circulation Research, 132(8), 1050–1063. https://doi.org/10.1161/CIRCRESAHA.122.322027 [Google Scholar] [Crossref]

6. Carpenter, J. R., & Smuk, M. (2021). Missing data: A statistical framework for practice. Biometrical Journal, 63(5), 915–947. https://doi.org/10.1002/bimj.202000318 [Google Scholar] [Crossref]

7. Dong, Y., & Peng, C. Y. J. (2013). Principled missing data methods for researchers. SpringerPlus, 2, Article 222. https://doi.org/10.1186/2193-1801-2-222 [Google Scholar] [Crossref]

8. Emmanuel, T., Maupong, T., Mpoeleng, D., Semong, T., Mphago, B., & Tabona, O. (2021). A survey on missing data in machine learning. Journal of Big Data, 8, Article 140. https://doi.org/10.1186/s40537-021-00505-7 [Google Scholar] [Crossref]

9. García, S., Ramírez-Gallego, S., Luengo, J., Benítez, J. M., & Herrera, F. (2016). Big data preprocessing: Methods and prospects. Big Data Analytics, 1, Article 9. https://doi.org/10.1186/s41044-016-0014-0 [Google Scholar] [Crossref]

10. Ghazal, T. M., Hasan, M. K., Alshurideh, M. T., Alzoubi, H. M., Ahmad, M., Akbar, S. S., Al Kurdi, B., & Akour, I. A. (2021). IoT for smart cities: Machine learning approaches in smart healthcare—A review. Future Internet, 13(8), 218. https://doi.org/10.3390/fi13080218 [Google Scholar] [Crossref]

11. Greener, J. G., Kandathil, S. M., Moffat, L., & Jones, D. T. (2022). A guide to machine learning for biologists. Nature Reviews Molecular Cell Biology, 23(1), 40–55. https://doi.org/10.1038/s41580-021-00428-3 [Google Scholar] [Crossref]

12. Hassija, V., Chamola, V., Mahapatra, A., Singal, A., Goel, D., Huang, K., Scardapane, S., Spinelli, I., Mahmud, M., & Hussain, A. (2024). Interpreting black-box models: A review on explainable artificial intelligence. Cognitive Computation, 16, 45–74. https://doi.org/10.1007/s12559-023-10144-3 [Google Scholar] [Crossref]

13. Ho, Y. S., Fülöp, T., Krisanapan, P., Soliman, K. M., & Cheungpasitporn, W. (2024). Artificial intelligence and machine learning trends in kidney care. The American Journal of the Medical Sciences, 367(4), 281–295. https://doi.org/10.1016/j.amjms.2023.09.005 [Google Scholar] [Crossref]

14. Hussain, S., Jamali, M. C., Habib, A., Hussain, M. S., Akhtar, M., & Najmi, A. K. (2021). Diabetic kidney disease: An overview of prevalence, risk factors, and biomarkers. Clinical Epidemiology and Global Health, 9, 2–6. https://doi.org/10.1016/j.cegh.2020.06.014 [Google Scholar] [Crossref]

15. Inker, L. A., & Titan, S. (2021). Measurement and estimation of GFR for use in clinical practice: Core curriculum 2021. American Journal of Kidney Diseases, 78(5), 736–749. https://doi.org/10.1053/j.ajkd.2021.07.007 [Google Scholar] [Crossref]

16. Jager, K. J., Kovesdy, C., Langham, R., Rosenberg, M., Jha, V., & Zoccali, C. (2019). A single number for advocacy and communication—Worldwide more than 850 million individuals have kidney diseases. Nephrology Dialysis Transplantation, 34(11), 1803–1805. https://doi.org/10.1093/ndt/gfz174 [Google Scholar] [Crossref]

17. Kalantar-Zadeh, K., Jafar, T. H., Nitsch, D., Neuen, B. L., & Perkovic, V. (2021). Chronic kidney disease. The Lancet, 398(10302), 786–802. https://doi.org/10.1016/S0140-6736(21)00519-5 [Google Scholar] [Crossref]

18. Kovesdy, C. P. (2022). Epidemiology of chronic kidney disease: An update 2022. Kidney International Supplements, 12(1), 7–11. [Google Scholar] [Crossref]

19. Levey, A. S., & Inker, L. A. (2017). Assessment of glomerular filtration rate in health and disease: A state-of-the-art review. Clinical Pharmacology & Therapeutics, 102(3), 405–419. https://doi.org/10.1002/cpt.729 [Google Scholar] [Crossref]

20. Martin, H. (2011). Laboratory measurement of urine albumin and urine total protein in screening for proteinuria in chronic kidney disease. Clinical Biochemistry Reviews, 32(3), 97. [Google Scholar] [Crossref]

21. Podkowińska, A., & Formanowicz, D. (2020). Chronic kidney disease as oxidative stress- and inflammatory-mediated cardiovascular disease. Antioxidants, 9(8), 752. https://doi.org/10.3390/antiox9080752 [Google Scholar] [Crossref]

22. Pudjihartono, N., Fadason, T., Kempa-Liehr, A. W., & O’Sullivan, J. M. (2022). A review of feature selection methods for machine learning based disease risk prediction. Frontiers in Bioinformatics, 2, Article 927312. https://doi.org/10.3389/fbinf.2022.927312 [Google Scholar] [Crossref]

23. Romagnani, P., Remuzzi, G., Glassock, R., Levin, A., Jager, K. J., Tonelli, M., Massy, Z., Wanner, C., & Anders, H. J. (2017). Chronic kidney disease. Nature Reviews Disease Primers, 3, Article 17088. https://doi.org/10.1038/nrdp.2017.88 [Google Scholar] [Crossref]

24. Rubini, L., Soundarapandian, P., & Eswaran, P. (2015). Chronic Kidney Disease. UCI Machine Learning Repository. https://archive.ics.uci.edu/dataset/336/chronic+kidney+disease [Google Scholar] [Crossref]

25. Saeed, M. H., & Hama, J. I. (2023). Cardiac disease prediction using AI algorithms with SelectKBest. Medical & Biological Engineering & Computing, 61(11), 3397–3408. https://doi.org/10.1007/s11517-023-02852-5 [Google Scholar] [Crossref]

26. Vassalotti, J. A., Centor, R., Turner, B. J., Greer, R. C., Choi, M., Sequist, T. D., & National Kidney Foundation Kidney Disease Outcomes Quality Initiative. (2016). Practical approach to detection and management of chronic kidney disease for the primary care clinician. The American Journal of Medicine, 129(2), 153–162. https://doi.org/10.1016/j.amjmed.2015.08.025 [Google Scholar] [Crossref]

27. Wadei, H. M., & Textor, S. C. (2012). The role of the kidney in regulating arterial blood pressure. Nature Reviews Nephrology, 8(10), 602–609. https://doi.org/10.1038/nrneph.2012.179 [Google Scholar] [Crossref]

28. Webster, A. C., Nagler, E. V., Morton, R. L., & Masson, P. (2017). Chronic kidney disease. The Lancet, 389(10075), 1238–1252. https://doi.org/10.1016/S0140-6736(16)32064-5 [Google Scholar] [Crossref]

29. Zhang, S. (2012). Nearest neighbour selection for iteratively KNN imputation. Journal of Systems and Software, 85(11), 2541–2552. https://doi.org/10.1016/j.jss.2012.05.072 [Google Scholar] [Crossref]

• Most Advanced Construction Material for Typical Boilers
• Nutrigenetic Impact of PEMT Gene Polymorphism Rs7946 On Choline Metabolism and Its Role in Personalised Nutrition
• Evaluating the Influence Factors of Media on Investor Sentiment Towards Stock Market Investments.
• To Compare the Effectiveness of Manual Traction and Mechanical Traction in Patient with Cervicogenic Headache
• Is Green Human Resource Management the Newest Trend or A Strategic Necessity?