Using machine learning methods to predict 28-day mortality in patients with hepatic encephalopathy

BACKGROUND: Hepatic encephalopathy (HE) is associated with marked increases in morbidity and mortality for cirrhosis patients. This study aimed to develop and validate machine learning (ML) models to predict 28-day mortality for patients with HE. METHODS: A retrospective cohort study was conducted in the Medical Information Mart for Intensive Care (MIMIC)-IV database. Patients from MIMIC-IV were randomized into training and validation cohorts in a ratio of 7:3. Training cohort was used for establishing the model while validation cohort was used for validation. The outcome was defined as 28-day mortality. Predictors were identified by recursive feature elimination (RFE) within 24 h of intensive care unit (ICU) admission. The area under the curve (AUC) and calibration curve were used to determine the predictive performance of different ML models. RESULTS: In the MIMIC-IV database, 601 patients were eventually diagnosed with HE. Of these, 112 (18.64%) experienced death within 28 days. Acute physiology score III (APSIII), sepsis related organ failure assessment (SOFA), international normalized ratio (INR), total bilirubin (TBIL), albumin, blood urea nitrogen (BUN), acute kidney injury (AKI) and mechanical ventilation were identified as independent risk factors. Validation set indicated that the artificial neural network (NNET) model had the highest AUC of 0.837 (95% CI:0.774–0.901). Furthermore, in the calibration curve, the NNET model was also well-calibrated (P = 0.323), which means that it can better predict the 28-day mortality in HE patients. Additionally, the performance of the NNET is superior to existing scores, including Model for End-Stage Liver Disease (MELD) and Model for End-Stage Liver Disease-Sodium (MELD-Na). CONCLUSIONS: In this study, the NNET model demonstrated better discrimination in predicting 28-day mortality as compared to other models. This developed model could potentially improve the early detection of HE with high mortality, subsequently improving clinical outcomes in these patients with HE, but further external prospective validation is still required. SUPPLEMENTARY INFORMATION: The online version contains supplementary material available at 10.1186/s12876-023-02753-z.