Abstract:Objective Chronic kidney disease (CKD) poses a major global health burden. Early CKD risk prediction enables timely interventions, but conventional models have limited accuracy. Machine learning (ML) enhances prediction, but interpretability is needed to support clinical usage with both in diagnostic and decision-making. Methods A cohort of 491 patients with clinical data was collected for this study. The dataset was randomly split into an 80% training set and a 20% testing set. To achieve the first objective, we developed four ML algorithms (logistic regression, random forests, neural networks, and eXtreme Gradient Boosting (XGBoost)) to classify patients into two classes—those who progressed to CKD stages 3–5 during follow-up (positive class) and those who did not (negative class). For the classification task, the area under the receiver operating characteristic curve (AUC-ROC) was used to evaluate model performance in discriminating between the two classes. For survival analysis, Cox proportional hazards regression (COX) and random survival forests (RSFs) were employed to predict CKD progression, and the concordance index (C-index) and integrated Brier score were used for model evaluation. Furthermore, variable importance, partial dependence plots, and restrict cubic splines were used to interpret the models’ results. Results XGBOOST demonstrated the best predictive performance for CKD progression in the classification task, with an AUC-ROC of 0.867 (95% confidence interval (CI): 0.728–0.100), outperforming the other ML algorithms. In survival analysis, RSF showed slightly better discrimination and calibration on the test set compared to COX, indicating better generalization to new data. Variable importance analysis identified estimated glomerular filtration rate, age, and creatinine as the most important predictors for CKD survival analysis. Further analysis revealed non-linear associations between age and CKD progression, suggesting higher risks in patients aged 52–55 and 65–66 years. The association between cholesterol levels and CKD progression was also non-linear, with lower risks observed when cholesterol levels were in the range of 5.8–6.4 mmol/L. Conclusions Our study demonstrated the effectiveness of interpretable ML models for predicting CKD progression. The comparison between COX and RSF highlighted the advantages of ML in survival analysis, particularly in handling non-linearity and high-dimensional data. By leveraging interpretable ML for unraveling risk factor relationships, contrasting predictive techniques, and exposing non-linear associations, this study significantly advances CKD risk prediction to enable enhanced clinical decision-making.

Integrated machine learning and deep learning for predicting diabetic nephropathy model construction, validation, and interpretability

A SuperLearner approach for predicting diabetic kidney disease upon the initial diagnosis of T2DM in hospital

Explainable Machine Learning-Based Prediction Model for Diabetic Nephropathy

Machine Learning-Based Predictive Modeling of Diabetic Nephropathy in Type 2 Diabetes Using Integrated Biomarkers: A Single-Center Retrospective Study

Development and external validation of a machine learning model to predict diabetic nephropathy in T1DM patients in the real-world

Study on risk factors of diabetic peripheral neuropathy and establishment of a prediction model by machine learning

A 10-year retrospective cohort of diabetic patients in a large medical institution: Utilizing multiple machine learning models for diabetic kidney disease prediction

Prediction of Chronic Kidney Disease Risk Using Multimodal Data

Prediction of diabetic kidney disease risk using machine learning models: a population-based cohort study of Asian adults

A Deep Learning Model Incorporating Knowledge Representation Vectors and Its Application in Diabetes Prediction

Development of a predictive model for the risk of microalbuminuria: comparison of 2 machine learning algorithms

Deep Learning Identifies Intelligible Predictors of Poor Prognosis in Chronic Kidney Disease

Predicting diabetic kidney disease for type 2 diabetes mellitus by machine learning in the real world: a multicenter retrospective study

Prediction and analysis of risk factors for diabetic retinopathy based on machine learning and interpretable models

Construction of Risk Prediction Model of Type 2 Diabetic Kidney Disease Based on Deep Learning

A Machine Learning Approach for Prediction of Diabetes Mellitus

Predicting diabetic retinopathy and identifying interpretable biomedical features using machine learning algorithms

Multi‐feature, Chinese–Western medicine‐integrated prediction model for diabetic peripheral neuropathy based on machine learning and SHAP

Prediction of diabetic kidney disease with machine learning algorithms, upon the initial diagnosis of type 2 diabetes mellitus

Development and internal validation of machine learning algorithms for end-stage renal disease risk prediction model of people with type 2 diabetes mellitus and diabetic kidney disease

Interpretable machine learning for predicting chronic kidney disease progression risk