Dataset Information

Predicting mortality in hemodialysis patients using machine learning analysis.

ABSTRACT:

Background

Besides the classic logistic regression analysis, non-parametric methods based on machine learning techniques such as random forest are presently used to generate predictive models. The aim of this study was to evaluate random forest mortality prediction models in haemodialysis patients.

Methods

Data were acquired from incident haemodialysis patients between 1995 and 2015. Prediction of mortality at 6 months, 1 year and 2 years of haemodialysis was calculated using random forest and the accuracy was compared with logistic regression. Baseline data were constructed with the information obtained during the initial period of regular haemodialysis. Aiming to increase accuracy concerning baseline information of each patient, the period of time used to collect data was set at 30, 60 and 90 days after the first haemodialysis session.

Results

There were 1571 incident haemodialysis patients included. The mean age was 62.3 years and the average Charlson comorbidity index was 5.99. The mortality prediction models obtained by random forest appear to be adequate in terms of accuracy [area under the curve (AUC) 0.68-0.73] and superior to logistic regression models (ΔAUC 0.007-0.046). Results indicate that both random forest and logistic regression develop mortality prediction models using different variables.

Conclusions

Random forest is an adequate method, and superior to logistic regression, to generate mortality prediction models in haemodialysis patients.

SUBMITTER: Garcia-Montemayor V

PROVIDER: S-EPMC8247746 | biostudies-literature |

REPOSITORIES: biostudies-literature

ACCESS DATA

Similar Datasets

Project description:BackgroundSepsis is a life-threatening condition, causing almost one fifth of all deaths worldwide. The aim of the current study was to identify variables predictive of 7- and 30-day mortality among variables reflective of the presentation of septic patients arriving to the emergency department (ED) using machine learning.MethodsRetrospective cross-sectional design, including all patients arriving to the ED at Södersjukhuset in Sweden during 2013 and discharged with an International Classification of Diseases (ICD)-10 code corresponding to sepsis. All predictions were made using a Balanced Random Forest Classifier and 91 variables reflecting ED presentation. An exhaustive search was used to remove unnecessary variables in the final model. A 10-fold cross validation was performed and the accuracy was described using the mean value of the following: AUC, sensitivity, specificity, PPV, NPV, positive LR and negative LR.ResultsThe study population included 445 septic patients, randomised to a training (n = 356, 80%) and a validation set (n = 89, 20%). The six most important variables for predicting 7-day mortality were: "fever", "abnormal verbal response", "low saturation", "arrival by emergency medical services (EMS)", "abnormal behaviour or level of consciousness" and "chills". The model including these variables had an AUC of 0.83 (95% CI: 0.80-0.86). The final model predicting 30-day mortality used similar six variables, however, including "breathing difficulties" instead of "abnormal behaviour or level of consciousness". This model achieved an AUC = 0.80 (CI 95%, 0.78-0.82).ConclusionsThe results suggest that six specific variables were predictive of 7- and 30-day mortality with good accuracy which suggests that these symptoms, observations and mode of arrival may be important components to include along with vital signs in a future prediction tool of mortality among septic patients presenting to the ED. In addition, the Random Forests appears to be a suitable machine learning method on which to build future studies.