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 Disease Progression Prediction Based on Reinforcement Reasoning over a Knowledge Graph

Interpretable Disease Prediction based on Reinforcement Path Reasoning over Knowledge Graphs

Deep Knowledge Reasoning Guided Disease Prediction.

RDKG: A Reinforcement Learning Framework for Disease Diagnosis on Knowledge Graph

Towards Explainable Machine Learning for Prediction of Disease Progression

A Knowledge-Guided Method for Disease Prediction Based on Attention Mechanism

Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community Retrieval

Context-aware and Time-aware Attention-based Model for Disease Risk Prediction with Interpretability

Interpretable Representation Learning for Healthcare via Capturing Disease Progression through Time

An Explainable and Personalized Cognitive Reasoning Model Based on Knowledge Graph: Toward Decision Making for General Practice

Causal Reasoning Model Based on Medical Knowledge Graph for Disease Diagnosis

Predictive Modeling of Clinical Events with Mutual Enhancement Between Longitudinal Patient Records and Medical Knowledge Graph

Mdpg: a novel multi-disease diagnosis prediction method based on patient knowledge graphs

Enhancing Personalized Healthcare via Capturing Disease Severity, Interaction, and Progression

A Probabilistic Disease Progression Model for Predicting Future Clinical Outcome

Interpretable machine learning for predicting chronic kidney disease progression risk

CTP:A Causal Interpretable Model for Non-Communicable Disease Progression Prediction

Marrying Medical Domain Knowledge With Deep Learning on Electronic Health Records: A Deep Visual Analytics Approach

Learning by Reasoning: an Explainable Hierarchical Association Regularized Deep Learning Method for Disease Prediction.

Leveraging A Medical Knowledge Graph into Large Language Models for Diagnosis Prediction