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.

Interpretable machine learning framework to predict gout associated with dietary fiber and triglyceride-glucose index

Creating machine learning models that interpretably link systemic inflammatory index, sex steroid hormones, and dietary antioxidants to identify gout using the SHAP (SHapley Additive exPlanations) method

Interpretable machine learning identifies metabolites associated with glomerular filtration rate in type 2 diabetes patients

Advancing non-alcoholic fatty liver disease prediction: a comprehensive machine learning approach integrating SHAP interpretability and multi-cohort validation

Multimodal Machine Learning-Based Marker Enables Early Detection and Prognosis Prediction for Hyperuricemia

A machine learning-based prediction model for gout in hyperuricemics: a nationwide cohort study

Prediction of hyperuricemia in people taking low-dose aspirin using a machine learning algorithm: a cross-sectional study of the National Health and Nutrition Examination Survey

Early Prediction of Gestational Diabetes Mellitus in the Chinese Population Via Advanced Machine Learning

Interpretable machine learning framework reveals novel gut microbiome features in predicting type 2 diabetes

Interpretable Machine Learning Algorithm Reveals Novel Gut Microbiome Features in Predicting Type 2 Diabetes

Exploration of Machine Learning for Hyperuricemia Prediction Models Based on Basic Health Checkup Tests

Interpretable prediction model for assessing diabetes complication risks in Chinese sufferers

Interpretable machine learning for predicting chronic kidney disease progression risk

Metabolomics and Machine Learning Identify Metabolic Differences and Potential Biomarkers for Frequent versus Infrequent Gout Flares

Investigation on explainable machine learning models to predict chronic kidney diseases

Interpretable machine learning model for predicting long-term cardiovascular mortality risk in chronic kidney disease patients using SHAP explainers

Application of machine learning algorithm incorporating dietary intake in prediction of gestational diabetes mellitus

A Machine Learning-Assisted Model for Renal Urate Underexcretion with Genetic and Clinical Variables among Chinese Men with Gout

Predictive model and risk analysis for peripheral vascular disease in type 2 diabetes mellitus patients using machine learning and shapley additive explanation

Prediction of Fatty Liver Disease in a Chinese Population Using Machine-Learning Algorithms

Explainable Machine Learning-Based Prediction Model for Diabetic Nephropathy