An Online Machine Learning Algorithm-Based Prognostic Predictive Model for Maintenance Hemodialysis Patients

High mortality rates among maintenance hemodialysis (MHD) patients necessitate more precise predictive tools. Existing models are limited by the accuracy and clinical usability. This study aimed to construct a precise and user-friendly machine learning (ML)-based mortality risk predictive model for MHD patients.

A total of 601 MHD patients from Shantou Central Hospital were enrolled in this study. Clinical and laboratory data were meticulously gathered and assessed. Patients were divided randomly into Training (70%) and Test (30%) cohort. Six types of ML algorithms based predictive models were constructed for prognostic prediction. The predictive accuracy of the model was evaluated by calculating the area under the receiver operating characteristic curve (AUROC) and the area under the precision-recall curve (AUPRC). Additionally, an online predictive model application was developed for practical clinical application.

The Training and Test cohort exhibited comparable demographic and clinical traits. Age, body mass index, hemoglobin, cholesterol, aspartate aminotransferase, and serum albumin levels emerged as significant independent predictors of prognosis. The Extreme Gradient Boosting (XGBoost) based model predictive performance measures included with AUROC 0.831 and AUPRC 0.310 in the Test cohort. The XGBoost-based model was selected as the definitive predictive tool and was made accessible via a web application.

We successfully developed a machine ML predictive model to predict the risk factors of MHD patients, which was then integrated into a user-friendly web application. This predictive tool could help identify MHD patients at high risk of mortality in clinical practice.