Unlocking Precision Medicine for Prognosis of Chronic Kidney Disease Using Machine Learning

Diagnostics (Basel). 2023 Oct 8;13(19):3151. doi: 10.3390/diagnostics13193151.

Abstract

Chronic kidney disease (CKD) is a significant global health challenge that requires timely detection and accurate prognosis for effective treatment and management. The application of machine learning (ML) algorithms for CKD detection and prediction holds promising potential for improving patient outcomes. By incorporating key features which contribute to CKD, these algorithms enhance our ability to identify high-risk individuals and initiate timely interventions. This research highlights the importance of leveraging machine learning techniques to augment existing medical knowledge and improve the identification and management of kidney disease. In this paper, we explore the utilization of diverse ML algorithms, including gradient boost (GB), decision tree (DT), K-nearest neighbor (KNN), random forest (RF), histogram boost (HB), and XGBoost (XGB) to detect and predict chronic kidney disease (CKD). The aim is to improve early detection and prognosis, enhancing patient outcomes and reducing the burden on healthcare systems. We evaluated the performance of the ML algorithms using key metrics like accuracy, precision, recall, and F1 score. Additionally, we conducted feature significance analysis to identify the most influential characteristics in the detection and prediction of kidney disease. The dataset used for training and evaluation contained various clinical and demographic attributes of patients, including serum creatinine level, blood pressure, and age, among others. The proficiency analysis of the ML algorithms revealed consistent predictors across all models, with serum creatinine level, blood pressure, and age emerging as particularly effective in identifying individuals at risk of kidney disease. These findings align with established medical knowledge and emphasize the pivotal role of these attributes in early detection and prognosis. In conclusion, our study demonstrates the effectiveness of diverse machine learning algorithms in detecting and predicting kidney disease. The identification of influential predictors, such as serum creatinine level, blood pressure, and age, underscores their significance in early detection and prognosis. By leveraging machine learning techniques, we can enhance the accuracy and efficiency of kidney disease diagnosis and treatment, ultimately improving patient outcomes and healthcare system effectiveness.

Keywords: K-nearest neighbors (KNN); XGBoost (XGB); chronic kidney disease (CKD); decision tree (DT); gradient boost (GB); histogram boost (HB); machine learning (ML); prognosis; random forest (RF).

Grants and funding

This research received no external funding.