Chronic kidney disease (CKD) is recognized as a major global public health challenge, with substantial impacts on morbidity and health care utilization worldwide [1,2]. In Taiwan, CKD affects approximately 10% of the adult population. The consistently high end-stage renal disease (ESRD) incidence imposes a substantial burden on the health care system and is accompanied by steadily increasing financial pressure [3,4]. This local epidemiologic context supports the need to improve risk stratification for kidney failure and mortality using Taiwan-based real-world data. Despite advancements in CKD research and management, accurately predicting progression to ESRD remains challenging. CKD is heterogeneous, with diverse etiologies, clinical presentations, and pathological features, and progression rates can vary substantially even among patients with identical primary kidney diseases [5]. Numerous clinical factors, including baseline estimated glomerular filtration rate (eGFR), proteinuria levels, age, and comorbidities, have been identified as contributors to CKD progression [6,7]. However, these factors and traditional risk equations, such as the kidney failure risk equation (KFRE), may not fully capture the complex, multidimensional nature of CKD, in which clinical, laboratory, and demographic variables can interact nonlinearly.

In recent years, advancements in artificial intelligence have significantly impacted the medical field. The integration of machine learning and deep learning with big data has found extensive applications across various medical disciplines, including image interpretation, disease diagnosis, and risk prediction [8]. Previous research has highlighted the potential of artificial intelligence for predicting the risk of patients with diabetes developing diabetic kidney disease or progressing to ESRD [9,10]. By incorporating a comprehensive range of clinical, demographic, and laboratory variables, these machine learning models have demonstrated considerable promise for achieving more precise risk stratification compared to traditional methods. Despite these advantages, several challenges remain in translating machine learning models into clinical practice. Many models have limited transparency, which can reduce interpretability and clinician trust [11]. In addition, differences in data sources, preprocessing strategies, and modeling pipelines may affect reproducibility across studies [12,13]. Furthermore, a substantial proportion of published machine learning prediction models lack external validation, which raises concerns about their generalizability and real-world clinical usefulness [14]. To bridge this gap between predictive modeling and clinical application, there is a critical need for transparent algorithms that can accurately handle the complex, time-to-event nature of chronic disease progression. Integrating tools such as Shapley additive explanations (SHAP) and survival-based machine learning techniques can demystify these models, providing clinicians with actionable insights into individual risk trajectories over time.

The aim of this study was to develop and validate machine learning–based models to predict the risks of ESRD and all-cause mortality in patients with CKD using a large-scale single-center cohort. We hypothesized that these models would enhance the precision of risk stratification, identify clinically meaningful predictors to inform clinical decision-making, and ultimately facilitate timely interventions to improve patient outcomes.

In this study, we developed and validated machine learning models to predict the risk of ESRD and all-cause mortality in a large cohort of patients with CKD. Our findings showed that the evaluated machine learning algorithms, including XGBoost, LightGBM, CatBoost, random forest, and a stacking classifier, achieved good discrimination for ESRD prediction, with a maximum AUROC of 0.894. Predictive performance for all-cause mortality was more modest, with a maximum AUROC of 0.774. In addition, calibration and decision-curve analyses supported the overall reliability and clinical utility of the model. A previous study developed predictive models for CKD progression using traditional statistical methods and clinical variables from Canadian cohorts collected between 2001 and 2008 [7]. These models, based on Cox proportional hazards regression, showed reasonable predictive performance by incorporating key demographic and laboratory parameters, such as eGFR and albuminuria. However, their reliance on conventional statistical techniques and a limited number of variables may restrict their ability to fully capture complex relationships among CKD progression risk factors. Furthermore, evolving clinical practices, therapeutic strategies, and diagnostic methods over time might limit their applicability to contemporary patient populations.

Using advanced machine learning techniques allows the integration of a broader range of clinical, demographic, and laboratory features into predictive models, enabling more precise risk assessment for CKD progression [16,17]. Unlike traditional statistical methods, which are often constrained by linear assumptions and limited predictors, machine learning algorithms can capture complex, nonlinear interactions among variables. Additionally, these models manage high-dimensional data, autonomously identify relevant features, and continuously adapt to evolving clinical practices and patient populations. The incorporation of explainability tools, such as SHAP, further enhances interpretability by clarifying individual risk factor contributions [18]. A recent study by Tangri et al [19] validated the Klinrisk machine learning model using data from the CANVAS (Canagliflozin Cardiovascular Assessment Study) and CREDENCE (Canagliflozin and Renal Events in Diabetes With Established Nephropathy Clinical Evaluation) clinical trials, demonstrating superior performance compared to KDIGO (Kidney Disease: Improving Global Outcomes) risk classification, with an AUROC of 0.88 at 3 years. In addition, our survival-based comparative analysis showed that RSF outperformed the KFRE at both the 2- and 5-year horizons, suggesting that machine learning approaches incorporating time-to-event information may provide improved discrimination over conventional risk equations.

Another study used machine learning models to predict ESRD risk in newly diagnosed patients with type 2 diabetes mellitus (T2DM), illustrating the effectiveness of these approaches in risk stratification [10]. Among the evaluated algorithms, XGBoost achieved the highest predictive performance. This study also used SHAP to interpret predictions, identifying baseline serum creatinine, mean serum creatinine within 1 year before T2DM diagnosis, high-sensitivity C-reactive protein, and UPCR as key predictors. However, its limitation to patients with diabetes restricts generalizability to broader CKD populations. Given CKD’s heterogeneity, with significantly varying etiologies and progression rates between patients with diabetes and those without diabetes, models trained exclusively on T2DM populations might not fully capture CKD’s complexity. Our study extends the application of machine learning in CKD risk prediction by analyzing CKD cohort with varied etiologies, offering a comprehensive ESRD risk assessment beyond diabetes-related kidney disease. Previous machine learning studies predicting CKD progression often had limited sample sizes and narrow feature sets [20-22]. Our study, in contrast, uses a significantly larger cohort and broader clinical, laboratory, and demographic features, likely enhancing observed model performance. This comprehensive approach improves the applicability of machine learning–based models in clinical settings, aiding early identification and intervention for patients at high risk of ESRD.

We identified baseline serum creatinine, UACR, and UPCR as top predictors of ESRD, consistent with established CKD progression knowledge. These biomarkers have long been recognized as key indicators of kidney function decline, and their prominence reinforces their critical role in risk assessment. Hemoglobin also emerged as a strong predictor, suggesting anemia’s potentially significant role in CKD progression. Anemia is common in CKD, primarily due to reduced erythropoietin production, iron deficiency, and chronic inflammation []. It has been linked to increased cardiovascular risk and impaired quality of life, particularly among patients with CKD [23-25]. A study using data from the SOLVD trial, enrolling patients with heart failure with significantly reduced left ventricular ejection fractions of 35% or less, showed anemia significantly associated with accelerated eGFR decline, especially pronounced in patients with baseline CKD [26]. The complex interaction among anemia, cardiovascular disease, and renal function decline underscores the importance of comprehensive clinical management in CKD [27].

The predictive performance for all-cause mortality was lower than that for ESRD, likely reflecting the greater heterogeneity of mortality as an outcome. Unlike ESRD, which is more directly linked to renal function and proteinuria-related measures, all-cause mortality is influenced by a broader range of renal and nonrenal factors. Routinely collected structured EHR variables may not fully capture important dimensions of mortality risk, such as frailty, functional status, nutritional reserve, cognitive impairment, and overall disease severity. Incorporating these variables may improve mortality prediction in future studies. Top predictors identified by SHAP analysis were hemoglobin, loop diuretic use, age, systolic blood pressure, cancer, and albumin. In patients with CKD, lower hemoglobin levels may reflect reduced erythropoietin production, chronic inflammation, malnutrition, and a greater comorbidity burden, all of which can contribute to vulnerability and adverse outcomes [28]. Loop diuretic use may indicate underlying volume overload or more advanced heart or kidney disease. Lower systolic blood pressure also emerged as an important predictor of mortality. This finding is consistent with prior studies in older adults with CKD and may be especially relevant in our cohort, given its advanced age distribution [29]. In this setting, lower systolic blood pressure may reflect frailty, impaired cardiac function, malnutrition, or overall physiologic vulnerability. Serum albumin is a well-established marker of both nutritional status and systemic inflammation [30,31]. Hypoalbuminemia has been consistently linked to increased short- and long-term mortality across diverse patient populations, including those with and without chronic kidney disease [32-34]. Proteinuria, commonly used to assess renal progression, also reflects systemic endothelial dysfunction and microvascular damage, contributing to increased cardiovascular morbidity and mortality. Studies have linked proteinuria with increased risk of coronary artery disease, stroke, heart failure, and arrhythmias [35-39]. Even low-grade albuminuria has been associated with adverse cardiac remodeling and unfavorable cardiovascular outcomes. A recent large-scale cohort study further demonstrated that proteinuria significantly increases the risk of sudden cardiac arrest in young adults, particularly in those with CKD [40]. Despite its well-established prognostic value in cardiovascular disease, proteinuria remains underutilized in routine cardiovascular risk stratification [41]. Our findings reinforce the clinical importance of proteinuria and support its broader integration into multidisciplinary risk evaluation strategies, especially within CKD populations.

From a clinical perspective, the practical value of these machine learning models lies in their potential integration into routine electronic health record workflows. Such risk estimates may help clinicians identify patients at high risk for ESRD or death who may benefit from closer follow-up, earlier nephrology referral, intensified management of modifiable risk factors, and more timely preparation for kidney replacement therapy. In addition, the decision-curve analysis suggests that the model may support risk-based clinical decision-making across a range of threshold probabilities. However, the optimal thresholds for intervention and the clinical impact of implementing such a model should be established in future prospective studies.

The strengths of our study include a large real-world CKD dataset with comprehensive clinical and laboratory information. Multiple machine learning algorithms, enhanced by advanced techniques such as SMOTE and SHAP, strengthen our findings’ robustness and applicability. Consistency in model performance across different frameworks further supports reliability. However, several limitations should be acknowledged. First, this was a retrospective, single-center study, which may introduce selection bias and limit generalizability. In particular, the cohort had a relatively advanced age distribution, likely reflecting the older patient population served by our institution. Nevertheless, age-stratified subgroup analysis showed similar model discrimination in patients aged ≥60 and <60 years, although external validation in younger populations remains necessary. Second, although we included a broad set of routinely available clinical variables, some potentially relevant factors, such as genetic markers and socioeconomic influences, were not included because they were not consistently available in structured form in our EHRs. Future studies incorporating multiomics data and social determinants may enhance predictive accuracy. Third, preprocessing methods may have introduced bias. KNN imputation may smooth extreme laboratory values, and SMOTE may generate synthetic minority samples that do not fully reflect the true clinical distribution, particularly for laboratory-based predictors. However, sensitivity analyses without imputation showed comparable model discrimination, suggesting that the overall results were broadly similar without imputation. Finally, the relative homogeneity of this hospital-based cohort may limit broader generalizability, and external validation in independent and more diverse CKD populations remains necessary.

In conclusion, machine learning models may provide useful tools for risk stratification of ESRD and all-cause mortality in patients with CKD. However, further validation is required before broader clinical application.