Abstract:Chronic kidney disease (CKD) poses a major global public health burden, with approximately 7 million affected. Early identification of those in whom disease is likely to progress enables timely therapeutic interventions to delay advancement to kidney failure. This study developed explainable machine learning models leveraging pathology data to accurately predict CKD trajectory, targeting improved prognostic capability even in early stages using limited datasets. Key variables used in this study include age, gender, most recent estimated glomerular filtration rate (eGFR), mean eGFR, and eGFR slope over time prior to incidence of kidney failure. Supervised classification modelling techniques included decision tree and random forest algorithms selected for interpretability. Internal validation on an Australian tertiary centre cohort (n=706; 353 with kidney failure and 353 without) achieved exceptional predictive accuracy, with the area under the receiver operating characteristic curve (ROC-AUC) reaching 0.94 and 0.98 on the binary task of predicting kidney failure for decision tree and random forest, respectively. To address the inherent class imbalance, centroid-cluster-based under-sampling was applied to the Australian dataset. To externally validate the performance of the model, we applied the model to a dataset (n=597 adults) sourced from a Japanese CKD registry. To overcome risks of overfitting on small sample sizes, transfer learning was subsequently employed by fine-tuned machine learning models on 15% of the external dataset (n=89) before evaluating the remaining 508 patients. This external validation demonstrated performant results with an ROC-AUC of 0.88 for the decision tree and 0.93 for the random forest model. Decision tree model analysis revealed the most recent eGFR and eGFR slope as the most informative variables for prediction in the Japanese cohort, aligning with the underlying pathophysiology. The research highlights the utility of deploying explainable machine learning techniques to forecast CKD trajectory even in the early stages utilising limited real-world datasets.

Prediction for chronic kidney disease by categorical and non_categorical attributes using different machine learning algorithms

Advancing Chronic Kidney Disease Prediction: Comparative Analysis of Machine Learning Algorithms and a Hybrid Model

Chronic kidney disease prediction based on machine learning algorithms

Prediction of Chronic Kidney Disease Risk Using Multimodal Data

Chronic Kidney Diseases Prediction Using Machine Learning Algorithms

Chronic Kidney Disease Prediction using Machine Learning

Chronic kidney disease prediction using machine learning techniques

Diagnosis of Chronic Kidney Disease Using Machine Learning Algorithm

A Robust Chronic Kidney Disease Classifier Using Machine Learning

A Machine Learning Methodology for Diagnosing Chronic Kidney Disease

An Ensemble Learning Approach for detection of Chronic Kidney Disease (CKD)

A Comparative Study, Prediction and Development of Chronic Kidney Disease Using Machine Learning on Patients Clinical Records

Neural network and support vector machine for the prediction of chronic kidney disease: A comparative study

Machine Learning Techniques for Chronic Kidney Disease Risk Prediction

Machine Learning Models for the Prediction of Renal Failure in Chronic Kidney Disease: A Retrospective Cohort Study

Chronic Kidney Disease Prediction by using Naive Bayes

A Detailed Analysis on Kidney and Heart Disease Prediction using Machine Learning

Investigation on explainable machine learning models to predict chronic kidney diseases

Exploring Early Prediction of Chronic Kidney Disease Using Machine Learning Algorithms for Small and Imbalanced Datasets

Predicting chronic kidney disease progression using small pathology datasets and explainable machine learning models

A comparative assessment of artificial intelligence models used for early prediction and evaluation of chronic kidney disease