Machine learning for classifying chronic kidney disease and predicting creatinine levels using at-home measurements

Abstract

AbstractBackgroundChronic kidney disease (CKD) is a global health concern with early detection playing a pivotal role in effective management. Machine learning models demonstrate promise in CKD detection, yet the impact on detection and classification using different sets of clinical features remains under-explored.MethodsIn this study, we focus on CKD classification and creatinine prediction using three sets of features; at-home, monitoring, and laboratory. We employ artificial neural networks (ANNs) and random forests (RFs) on a dataset of 400 patients with 25 input features, which we divide into three feature sets. Using 10-fold cross-validation, we calculate metrics such as accuracy, true positive rate (TPR), true negative rate (TNR), and mean squared error.ResultsOur results reveal RF achieves superior accuracy (92.5%) in at-home CKD classification over ANNs (82.9%). ANNs achieve a higher TPR (92.0%) but a lower TNR (67.9%) compared with RFs (90.0% and 95.8%, respectively). For monitoring and laboratory features, both methods achieve accuracies exceeding 98%. The R2 score for creatinine regression is approximately 0.3 higher with laboratory features than at-home features. Feature importance analysis identifies key clinical variables hemoglobin and blood urea, and key comorbidities hypertension and diabetes mellitus, in agreement with previous studies.ConclusionsMachine learning models, particularly RFs, exhibit promise in CKD diagnosis and highlight significant features in CKD detection. Moreover, such models may assist in screening a general population using at-home features—potentially increasing early detection of CKD, thus improving patient care and offering hope for a more effective approach to managing this prevalent health condition.