Abstract:Rationale: The current mortality risk scores for hemodialysis (HD) patients employ regression techniques and have limited applicability and robustness. Objective: We aimed to develop a machine learning model utilizing clinical factors to predict first-year mortality in HD patients that could assist physicians in classifying high-risk patients.Methods: Training and testing cohorts consisted of 5351 incident HD patients from a single center and 5828 incident HD patients from 97 renal centers, respectively. The outcome was all-cause mortality during the first year of dialysis. eXtreme Gradient Boosting (XGBoost) was used for algorithm training and validation. Two models were established based on the data obtained at dialysis initiation (model 1) and data from 0-3 months after dialysis initiation (model 2). Ten-fold cross-validation was applied to each model. The C statistic (area under the receiver operating characteristic curve, AUC) was used to assess the predictive ability of the models.Results: In the training and testing cohorts, 585 (10.93%) and 764 (13.11%) patients, respectively, died during the first-year follow-up. Fifteen most important features were selected from 42 candidate features. The performance of model 1 (C statistic of 0.8318, 95% CI 0.7771 - 0.8374) was similar to that of model 2 (C statistic of 0.8530, 95% CI 0.8057 - 0.8593).Conclusions: We developed and validated two machine learning models to predict first-year mortality in HD patients. These models could be used to stratify high-risk patients in the early stages of dialysis.Funding Statement: This work is supported by National Key Research and Development Projects of China (2018YFC1314003). Declaration of Interests: The authors declare that they have no other relevant financial interests.Ethics Approval Statement: This study followed the tenets of the Declaration of Helsinki and was approved by the ethics committee of the First Affiliated Hospital of Zhejiang University (IIT20200088A), Hangzhou, China. Written informed consent was obtained for all study participants.

Optimization of Dry Weight Assessment in Hemodialysis Patients via Reinforcement Learning

A Practical Electronic Health Record-Based Dry Weight Supervision Model for Hemodialysis Patients.

Optimization of anemia treatment in hemodialysis patients via reinforcement learning

Multiple Laplacian Regularized RBF Neural Network for Assessing Dry Weight of Patients With End-Stage Renal Disease

Optimal Treatment Strategies for Critical Patients with Deep Reinforcement Learning

Reinforcement Learning in Clinical Medicine: a Method to Optimize Dynamic Treatment Regime over Time.

Machine Learning for the Prediction of First-Year Mortality in Incident Hemodialysis Patients

Mortality Prediction with Adaptive Feature Importance Recalibration for Peritoneal Dialysis Patients: a deep-learning-based study on a real-world longitudinal follow-up dataset

Deep Reinforcement Learning for Cost-Effective Medical Diagnosis

Development and validation of a new dry weight estimation method using single frequency bioimpedance in hemodialysis patients.

Machine learning models to predict post-dialysis blood pressure in children and young adults on maintenance hemodialysis

A value-based deep reinforcement learning model with human expertise in optimal treatment of sepsis

A Hemodialysis Mortality Prediction Model Based on Active Contrastive Learning.

Assessing Dry Weight of Hemodialysis Patients Via Sparse Laplacian Regularized RVFL Neural Network with L2,1-Norm

Reinforcement Learning in Dynamic Treatment Regimes Needs Critical Reexamination

Dialysate Decision Support Method Based on Deep Learning

Enhancing End Stage Renal Disease Outcome Prediction: A Multi-Sourced Data-Driven Approach

Deep Reinforcement Learning for Dynamic Treatment Regimes on Medical Registry Data

An optimized machine learning framework for predicting intradialytic hypotension using indexes of chronic kidney disease-mineral and bone disorders

Improvement of Dialysis Dosing Using Big Data Analytics

Robust Hybrid Learning for Estimating Personalized Dynamic Treatment Regimens