A Hybrid Machine Learning Framework for Early Prediction of Chronic Kidney Disease Progression Using Longitudinal Claims Data: An XGBoost-LSTM Ensemble with Temporal Attention
SAXENA, J. N.; Potturu, D. V. P.; Nagraj, A.
Show abstract
Background: Chronic kidney disease (CKD) affects approximately 850 million individuals worldwide and remains a leading cause of morbidity, premature mortality, and escalating healthcare costs. Despite the availability of clinical biomarkers, CKD progression to end stage renal disease (ESRD) is frequently identified late, limiting opportunities for preventive intervention. Conventional predictive models have relied predominantly on static cross sectional laboratory values, failing to capture the temporal dynamics of disease trajectory that longitudinal claims data can provide. Objective: This study proposes a novel hybrid machine learning framework: XGBoost LSTM Attention (XLA), that integrates gradient boosted feature selection with long short-term memory (LSTM) networks and a temporal attention mechanism to improve early prediction of CKD progression from Stage 3 to Stages 4/5 or ESRD using longitudinal claims based features. Methods: We conducted two complementary analyses. Primary analysis: a cross sectional validation using real NHANES 2015 to 2018 data (n=701 CKD Stage 3 adults) predicting significant proteinuria (UACR greater than or equal to 30 mg/g) from clinical features excluding UACR. Supplementary analysis: an NHANES-calibrated longitudinal cohort (n=8,412) with simulated quarterly measurements demonstrated XLA performance under real world longitudinal data conditions. All models were evaluated using 5-fold stratified cross-validation. Results: In the primary NHANES cross sectional analysis, the XLA framework achieved AUC ROC of 0.684 (95% CI: 0.641 to 0.727), with all models performing comparably (AUC 0.684 to 0.710), confirming that cross sectional clinical features alone provide limited signal for proteinuria prediction and underscoring the necessity of UACR measurement. In the longitudinal supplementary analysis, XLA achieved AUC ROC of 0.994 versus 0.939 for the best cross-sectional baseline (+5.5%), demonstrating that temporal trajectory features particularly eGFR slope and RAAS adherence trends: confer substantial incremental predictive value when longitudinal data are available. Conclusion: The XLA framework demonstrates meaningful advantages over traditional approaches when applied to longitudinal claims data. Cross sectional findings highlight the irreplaceable role of direct UACR measurement in CKD risk stratification. Together, these results provide actionable evidence for both the limitations of static prediction and the promise of trajectory based approaches in value based care programs managing large CKD populations. Keywords: chronic kidney disease, CKD progression, machine learning, XGBoost, LSTM, temporal attention, claims data, NHANES, proteinuria, healthcare informatics, value based care.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Correlating Deep Learning-Based Automated Reference Kidney Histomorphometry with Patient Demographics and Creatinine 94%
- Artificial Intelligence for COVID-19 Risk Classification in Kidney Disease: Can Technology Unmask an Unseen Disease? 94%
- Estimating and predicting kidney function decline in the general population 93%
Similar papers in this journal
- Prognostic Utility of Total Kidney Volume for Chronic Kidney Disease Risk Prediction: An Observational and Mendelian Randomization Study 93%
- Clonal hematopoiesis of indeterminate potential contributes to accelerated chronic kidney disease progression 93%
- Clonal hematopoiesis of indeterminate potential is associated with worse kidney function and anemia in a cohort of patients with advanced chronic kidney disease 92%
Similar papers in this journal
- External validation of six clinical models for prediction of unknown chronic kidney disease in a German population 93%
- Multimodal predictions of end stage chronic kidney disease from asymptomatic individuals for discovery of genomic biomarkers 92%
- Kidney Shape Statistical Analysis: Associations with Disease and Anthropometric Factors 92%
Similar papers in this journal
- Evaluating the kidney disease progression using a comprehensive patient profiling algorithm: A hybrid clustering approach 94%
- Cardiovascular disease protein biomarkers are associated with kidney function: the Framingham Heart Study 92%
- Chronic Kidney Disease in Ecuador: An Epidemiological and Health System Analysis of an Emerging Public Health Crisis 91%
Similar papers in this journal
- Comparison of low eGFR prevalence and prediction for mortality using 2009 and 2021 CKD-EPI equations in Mexican adults 94%
- Circulating Plasma Biomarkers in Biopsy-Confirmed Kidney Disease: Results from the Boston Kidney Biopsy Cohort 92%
- Preprint server use in kidney disease research: a rapid review 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.