Skip to main content

ORIGINAL RESEARCH article

Front. Med., 28 September 2022
Sec. Intensive Care Medicine and Anesthesiology
This article is part of the Research Topic Clinical Application of Artificial Intelligence in Emergency and Critical Care Medicine, Volume III View all 13 articles

Explainable time-series deep learning models for the prediction of mortality, prolonged length of stay and 30-day readmission in intensive care patients

\r\nYuhan DengYuhan Deng1Shuang LiuShuang Liu1Ziyao WangZiyao Wang1Yuxin WangYuxin Wang1Yong Jiang,*Yong Jiang2,3*Baohua Liu*Baohua Liu1*
  • 1School of Public Health, Peking University, Beijing, China
  • 2Department of Neurology, Beijing Tiantan Hospital, Capital Medical University, Beijing, China
  • 3China National Clinical Research Center for Neurological Diseases, Beijing, China

Background: In-hospital mortality, prolonged length of stay (LOS), and 30-day readmission are common outcomes in the intensive care unit (ICU). Traditional scoring systems and machine learning models for predicting these outcomes usually ignore the characteristics of ICU data, which are time-series forms. We aimed to use time-series deep learning models with the selective combination of three widely used scoring systems to predict these outcomes.

Materials and methods: A retrospective cohort study was conducted on 40,083 patients in ICU from the Medical Information Mart for Intensive Care-IV (MIMIC-IV) database. Three deep learning models, namely, recurrent neural network (RNN), gated recurrent unit (GRU), and long short-term memory (LSTM) with attention mechanisms, were trained for the prediction of in-hospital mortality, prolonged LOS, and 30-day readmission with variables collected during the initial 24 h after ICU admission or the last 24 h before discharge. The inclusion of variables was based on three widely used scoring systems, namely, APACHE II, SOFA, and SAPS II, and the predictors consisted of time-series vital signs, laboratory tests, medication, and procedures. The patients were randomly divided into a training set (80%) and a test set (20%), which were used for model development and model evaluation, respectively. The area under the receiver operating characteristic curve (AUC), sensitivity, specificity, and Brier scores were used to evaluate model performance. Variable significance was identified through attention mechanisms.

Results: A total of 33 variables for 40,083 patients were enrolled for mortality and prolonged LOS prediction and 36,180 for readmission prediction. The rates of occurrence of the three outcomes were 9.74%, 27.54%, and 11.79%, respectively. In each of the three outcomes, the performance of RNN, GRU, and LSTM did not differ greatly. Mortality prediction models, prolonged LOS prediction models, and readmission prediction models achieved AUCs of 0.870 ± 0.001, 0.765 ± 0.003, and 0.635 ± 0.018, respectively. The top significant variables co-selected by the three deep learning models were Glasgow Coma Scale (GCS), age, blood urea nitrogen, and norepinephrine for mortality; GCS, invasive ventilation, and blood urea nitrogen for prolonged LOS; and blood urea nitrogen, GCS, and ethnicity for readmission.

Conclusion: The prognostic prediction models established in our study achieved good performance in predicting common outcomes of patients in ICU, especially in mortality prediction. In addition, GCS and blood urea nitrogen were identified as the most important factors strongly associated with adverse ICU events.

Introduction

Patients in the intensive care unit (ICU) are usually critically ill, presenting a high mortality risk compared with other departments in the hospital (1). In addition, readmission and prolonged length of stay (LOS) are both common clinical outcomes indicating patients’ health conditions (2, 3), critical care quality (4, 5), and medical efficiency (6). Thus, early identification of seriously ill patients and those with prolonged LOS and readmission risk and subsequent management is exceedingly important in improving patient outcomes and providing optimal allocation of medical resources.

However, traditional scoring systems, even some machine learning methods in predicting these outcomes, especially in stratifying the risk of readmission, have shown only modest results (710). Although part of the existing work based on machine learning models seems promising (1113), few of them are able to take advantage of the characteristics of features collected in the ICU, which are time-series forms. Presently, these time-series problems can be approached with deep learning-based models, such as recurrent neural network (RNN) and its derived models, namely, gated recurrent unit (GRU) (14) and long short-term memory (LSTM) (15), which can learn valuable information from a large number of rapidly changing variables, making it possible to make full use of ICU data collected at a high frequency (16). Based on these advanced models, several studies have conducted prognostic prediction of patients in ICU, but most were disease-specific or ICU-specific (1720), the clinical use of which was restricted to a specific group. To the best of our knowledge, no studies have ever predicted common outcomes while maximizing the value of these models of patients in general ICU. Furthermore, because of the complexity of these deep learning models, they are not easy to interpret, which restricts their practical application to clinical decisions (21, 22). Therefore, transparency and explainability must be considered when constructing prediction models. Recently, several methods have been introduced to improve model interpretability; among them, attention mechanisms seem to be one of the most prospective approaches (23), which have been proven to provide the foundation for clinical interpretation (24). Through explainable prediction models, significant factors can be identified at an early stage to help clinicians offer better medical interventions.

In this study, we aimed to apply three time-series deep learning models for predicting three common ICU outcomes, namely, mortality, prolonged LOS, and readmission, of patients in ICU from the Medical Information Mart for Intensive Care-IV (MIMIC-IV) database and identified predictors of high importance based on attention mechanisms to facilitate model interpretability.

Materials and methods

Data source and study participants

Patient information was extracted from the MIMIC-IV database (25) to conduct a retrospective cohort study. The MIMIC-IV database contains real medical records with comprehensive information for each patient, ranging from demographic information, vital signs, and laboratory tests to medication administration. All patient information was collected from those who were admitted to the emergency departments and ICU of a tertiary academic medical center in Boston, MA, United States, from 2008 to 2019. The database involves a total of 53,150 patients admitted to the ICU, and all patients’ information was de-identified.

A total of 40,083 patients were included in our study. Patients were excluded for the following reasons: (1) age ≤ 18 years or ≥ 90 years and (2) stay in the ICU for less than 24 h. In addition, we only included the first admission record if a patient was admitted to the ICU more than once, so the admission records and subject IDs corresponded.

Predictors and outcomes

We extracted the following data from the MIMIC-IV database upon the initial 24 h of ICU admission and the last 24 h before discharge, and all of the variables were selected according to three conventional scoring systems [APACHE II (26), SOFA (27) and SAPS II (28)]: (1) basic information: age, sex, admission type, ethnicity; (2) diagnosis: AIDS, hematologic malignancy, metastatic cancer; (3) laboratory measurements: serum sodium, serum potassium, serum creatinine, hematocrit, white blood cell count, blood urea nitrogen (BUN), serum bicarbonate, bilirubin, platelets; (4) vital signs: temperature, mean arterial pressure, systolic blood pressure, heart rate, respiratory rate, PaO2, Glasgow coma score (GCS); (5) medication administration: dopamine, dobutamine, epinephrine, norepinephrine; (6) output: urinary output; (7) surgical procedures: invasive mechanical ventilation, non-invasive mechanical ventilation.

Three primary outcomes were needed for prediction in our study. One is the occurrence of death in the hospital, which was defined as whether the patient died during hospitalization, and this information can be extracted from hospital_expire_flag in the admissions table in the MIMIC-IV database. Another is the occurrence of prolonged LOS, a binary variable with a cutoff point of 75th percentile LOS of the study participants, which was 4 days in our study. Thus, patients with LOS for more than 4 days were labeled as 1, and those with LOS for less than 4 days were labeled as 0. Prolonged LOS information was calculated from the icustays table. The other outcome is readmission, which was defined as whether the patient was recorded as having full-cause readmission within 30 days after hospital discharge.

Data extracted from the initial 24 h after ICU admission were used to predict mortality and prolonged LOS, while data derived from the last 24 h before discharge were used to predict the risk of 30-day readmission.

Data preprocessing and statistical analysis

Continuous variables are presented as the means ± SDs or medians and interquartile ranges and are compared using Student’s t-test or Wilcoxon rank-sum test according to their normality test results. Categorical variables are presented as counts and percentages and compared through the Chi-square test or Fisher’s exact test with significant p-values < 0.05.

According to recording frequencies, predictors can be classified into dynamic predictors and static predictors. Dynamic variables were those recorded more than once during ICU hospitalization, mostly consisting of vital signs and laboratory tests. Static variables, which included demographic information such as age, sex, and admission type, were all constant and did not change over time. The initial 24 h of ICU admission and the last 24 h before discharge were divided into a time-series of 24 steps, and all variables were obtained for each 1 h window to generate a complete dataset. For static variables, the same value of each patient was recorded 24 times. For dynamic variables, if a variable was recorded more than once in an hour, its mean value was used for aggregation, and then the last observation carried forward (LOCF) was conducted to impute missing values of time-series data. After the first missingness imputation, variables with missing rates of more than 30% were excluded. All categorical variables were one-hot encoded, so the final number of predictors was 33.

All participants were randomly split into a training set (80%) and a test set (20%). The mean value of each continuous variable in the training set was used to impute the remaining missing values in both the training set and the test set. Three deep learning models, RNN, GRU, and LSTM, were used for model development in the training set, and model performance was evaluated in terms of AUC, sensitivity, specificity, and Brier score in the test set. Variable importance according to the attention mechanism was also produced from the test set.

All data analysis procedures were conducted with SAS 9.4 and Python 3.7.

Recurrent neural network

The mechanism of RNN to tackle time-series problems is that it includes a hidden layer, which incorporates information from all former steps, and with the extension of each time step, the hidden layer iteratively updates, and stores new memory. As shown in Figure 1A, Xt represents input variables of the present time step, while Ht–1 is the hidden layer of the previous time step, two of which co-determine the hidden layer Ht of the present time step, so Ht contains all information of both the previous time steps and the present time step.

FIGURE 1
www.frontiersin.org

Figure 1. Model diagram of a single cell. (A) RNN; (B) GRU; (C) LSTM.

Gated recurrent unit

Gated recurrent unit enriches the structure of RNN with gating systems (an update gate and a reset gate) to solve the problem of too much information kept in the hidden layer when time sequences are too long, in which the update gate (Zt) decides how much information to forget and how much information to keep and the reset gate (Rt) determines how much information on former steps to forget, as shown in Figure 1B.

Long short-term memory

Long short-term memory is more complicated than GRU. It has three gates, an input gate (It), a forget gate (Ft), and an output gate (Ot) addition with a memory cell Ct. The three gates are all generated by Xt and Xt–1, and they separately decide how much present input information to keep, how much previous information to forget, and how much total information to output. The schematic diagram of an LSTM cell is shown in Figure 1C.

Attention mechanism

Considering the complexity of the three deep learning models, especially LSTM, which has relatively more parameters, it would be very difficult to explain the contribution of each variable from these prediction models. Hence, an additional layer was added to each of the three models at the level of input variables; specifically, each variable of each time step (33 × 24 time-specific variables in all) was given an attention weight, which can be represented as at = softmax(xtWt), and the sum of the weight of each time step was equal to 1 (|at| = 1), so the new input variable was represented as Xnew = AX. As a result, we ignored the possibly different contributions of each time step but focused on the contribution of each variable. Through the aggregation of all time steps, the global contribution of each variable can be generated.

Results

Patient characteristics

A total of 40,083 patients were included in our study for the prediction of mortality and prolonged LOS after excluding those who did not meet the selection criteria, and 36,180 of them were included to predict readmission, as shown in Figure 2. Among these patients, 3,903 (9.74%) deaths occurred during hospitalization, and 11,038 (27.54%) underwent prolonged LOS. After excluding 3,903 patients who died in the hospital, 4,268 (11.79%) were readmitted to the hospital within 30 days after discharge. The comparison of basic information of these patients stratified by outcomes is shown in Table 1. Patients with in-hospital death, compared with those without, were older (P < 0.001), comprised more women (P < 0.001) and more other or unknown ethnicity (P < 0.001), and were more likely to be admitted to the emergency room and transferred from the hospital (P < 0.001), had a longer LOS in the ICU (P < 0.001), and were more likely to be diagnosed with metastatic cancer (P < 0.001) and hematologic malignancy (P < 0.001). Patients with prolonged LOS were also comprised of more women (P < 0.015) and other or unknown ethnicity (P < 0.001), more transferred from the hospital (P < 0.001), and more were diagnosed with hematologic malignancy (P < 0.048), while fewer were diagnosed with metastatic cancer (P = 0.025). Patients who were readmitted were also older (P < 0.001), comprised of more white people and fewer other or unknown ethnicity (P < 0.001), more were transferred from the hospital (P < 0.001) and diagnosed with metastatic cancer (P < 0.001) and hematologic malignancy (P < 0.001). The diagnosis of AIDS showed similar results between both patients with and without in-hospital death (P = 0.777), prolonged LOS (P = 0.985), and readmission (P = 0.146).

FIGURE 2
www.frontiersin.org

Figure 2. Flow chart depicting the inclusion of study participants.

TABLE 1
www.frontiersin.org

Table 1. Characteristics of study participants grouped by outcomes.

Model performance

The receiver operating characteristic (ROC) curves of the three prediction models in predicting in-hospital mortality, prolonged LOS, and 30-day readmission are shown in Figures 3A–C. The AUCs of RNN, GRU, and LSTM in predicting mortality were 0.862 ± 0.001, 0.870 ± 0.001, and 0.869 ± 0.002, respectively, and those in prolonged LOS prediction were 0.761 ± 0.002, 0.757 ± 0.011, and 0.765 ± 0.003, respectively. The AUCs of readmission prediction reached only 0.625 ± 0.008, 0.631 ± 0.011, and 0.635 ± 0.018 for the three deep learning models. Other performance metrics, namely, sensitivity, specificity, and Brier score, are shown in Table 2.

FIGURE 3
www.frontiersin.org

Figure 3. ROC curves of RNN, GRU, and LSTM. (A) Mortality prediction; (B) prolonged LOS prediction; (C) 30-day readmission prediction.

TABLE 2
www.frontiersin.org

Table 2. Model performance in predicting hospital mortality, PLOS, and 30-day readmission of patients in ICU.

Variable significance

The significance of the variables is shown in Figures 46. All three prediction models (RNN, GRU, and LSTM) indicated the important roles of GCS, age, blood urea nitrogen, and administration of norepinephrine in predicting mortality. GCS, invasive ventilation, and blood urea nitrogen were all among the top five significant predictors for prolonged LOS prediction. Blood urea nitrogen, GCS score, and ethnicity were strong predictors for 30-day readmission prediction.

FIGURE 4
www.frontiersin.org

Figure 4. Variable importance generated by mortality prediction models. (A) RNN; (B) GRU; (C) LSTM.

FIGURE 5
www.frontiersin.org

Figure 5. Variable importance generated by prolonged LOS prediction models. (A) RNN; (B) GRU; (C) LSTM.

FIGURE 6
www.frontiersin.org

Figure 6. Variable importance generated by 30-day readmission prediction models. (A) RNN; (B) GRU; (C) LSTM.

Discussion

In this study, three time-series deep learning models were applied to predict in-hospital mortality, prolonged LOS, and 30-day readmission with conventional and easily available variables in ICU settings, and influential factors associated with the three outcomes were identified through attention mechanisms to enhance model interpretability.

Our study focused on the outcome prediction of general patients without distinguishing their diseases, and the results showed in-hospital mortality of 9.74%, a prolonged LOS of 27.54%, and 30-day readmission of 11.79%, which were roughly consistent with previous studies (29, 30). For better practical use in clinical settings, we only included variables that are commonly used and easily available according to three traditional scoring systems [APACHE II (26), SOFA (27), and SAPS II (28)] and collected within 24 h, so compared with other similar studies, the number of variables in this study was relatively small, which partly explained the not very outstanding performance of our prediction models. For example, in Golas’s study, 3,512 variables were included (31) and in Sherman’s study, 165 variables were included (32), while in our study, only 33 variables were included, which were all among the common clinical measurement indicators. Specifically, the values of AUC indicated good discrimination capability in mortality prediction, moderate in prolonged LOS, and not as good in readmission. All prediction models were trained with a 24-h time window, which was a comprehensive consideration of various conditions, such as the significance of each period, the complexity of variable availability, and the missingness rate. Nevertheless, the length of the time window can also have a certain impact on model performance. In Na’s study, the best-performing model (GRU) was trained with 8/16/24/48-h time windows, and the overall tendency indicated that the extended time window corresponded to better predictive performance (33). In addition, the performance of the readmission prediction model may be strongly affected by the period of readmission, ranging from 24 h to 30 days in existing studies (3436); usually, the shorter the time interval is the better the prediction capability. Thus, using a relatively narrow time window, which is 24 h, to predict long-term outcomes theoretically resulted in a weak predictive capability. However, the result is still competitive in all three outcome predictions because of the application of deep learning models with a small quantity of time-series variables (8, 9, 31, 37).

The results of the performances of the three deep learning modes (RNN, GRU, and LSTM) did not differ greatly in predicting outcomes, and this was inconsistent with what was obtained by Na’s study (33). For a similar task (mortality prediction using RNN, GRU, and LSTM with variables collected within a 48-h observation window), GRU and LSTM performed better than RNN. In their study, the observation window was double-length, which may be related to the difference in the results. The superiority of LSTM and GRU is that their additional gate systems can better select important information stored in hidden layers on each time step, so when the time window is too short, the information contained is more likely to be undiscardable so that the advantages of LSTM and GRU cannot be reflected (23).

Attention mechanisms allowed us to identify important features used by three different models in prediction, and the influential variables of each outcome selected by different deep learning models also did not differ greatly. The GCS was identified as the top important factor for mortality, prolonged LOS, and readmission prediction, and the same results can also be extracted from other similar studies. For example, some studies have concluded that GCS is an independent mortality-related factor and has the most significant feature importance in some specific diseases (38, 39). This variable was also demonstrated to be one of the most important determinants of prolonged LOS in patients with traumatic brain injury (40). Moreover, in Oh’s study, 2.28-fold higher unplanned 2-day readmissions were associated with GCS scores less than 13 (41). A lower GCS score indicates more severely impaired consciousness, which may lead to a poor outcome if timely medical intervention is not conducted (42). Age was also demonstrated to have a strong relationship with in-hospital mortality in the ICU by previous studies (43, 44), with a higher mortality rate occurring among elderly patients. These patients generally have reduced immunity, underlying chronic diseases, and worse recovery ability, which may complicate their health status and result in adverse outcomes (45, 46). In Martin’s study, BUN was discovered to have a significant association with 28-day mortality (47), and Jamshid’s study identified BUN as one of the factors with the highest predictive values to predict the risk of mortality from patients with severe COVID-19 (48), which also provides support for our results. BUN was also identified as a significant variable for prolonged LOS and readmission prediction, and the same results can also be found in homogeneous studies (49, 50). The increased level of BUN is associated with kidney damage, which is supported by multiple mechanisms (51). We also included some medication administration information following SOFA scoring systems (27), and the results showed that norepinephrine, which was recommended as first-line therapy for cardiogenic shock (52), had decisive implications on mortality prediction. This result was also generated by Lu’s study, which concluded that patients in cardiogenic shock treated with norepinephrine had significantly increased short-term mortality rates (53). These patients, especially those in refractory shock, usually had an extremely poor prognosis, which lead to higher mortality (54). We also found that invasive ventilation was a decisive predictor for prolonged LOS, a risk factor also suggested by a meta-analysis containing 28 articles (3). In the prediction of readmission, the results showed that ethnicity was a decisive predictor, with the white people owning an increased probability for readmission and other/unknown ethnicity decreasing. In Mukhopadhyay’s study, the results also showed that ethnicity was independently associated with hospital readmissions (55).

There are several limitations to our study. First, we excluded some variables that may have predictive values because of high missingness rates, such as the mean arterial blood pressure and bilirubin, and the insurance variable, which may influence LOS, was also not included considering that more than half of the insurance type was labeled “Others.” Second, as a single-center study, the generalizability and representation of our conclusion still need to be demonstrated by other data sources. Third, the alternative variables may still be not comprehensive. For example, the diagnosis at ICU admission was not considered a predictor in our study, which may affect the application and generalization of this model in different patient groups. More variables that are easily available need to be explored to further improve model performance.

Conclusion

Three time-series deep learning models were applied for the prediction of three common ICU outcomes, namely, mortality, prolonged LOS, and readmission. The prediction models reached good performance, especially in mortality prediction, which is of great value in clinical settings considering the conventional and easily available variables incorporated. Our results also indicate that GCS and blood urea nitrogen were highly associated with adverse outcomes of patients in ICU, and focusing on these variables can better assist clinical decisions.

Data availability statement

All data analyzed in this study were obtained from the MIMIC-IV database, which can be found at https://physionet.org/about/database/.

Ethics statement

Ethical review and informed consent were not required for this study as the study database, the MIMIC-IV database is publicly available, and all patient data are de-identified.

Author contributions

YD designed the study and wrote the manuscript draft. BL and YJ critically revised the manuscript. SL assisted with the study protocol and data analysis. ZW contributed to manuscript editing and model explanation. YW helped with manuscript revision. All authors read and approved the final manuscript.

Funding

This study received financial support from the National Key Research and Development Program of China (No. 2018YFC1311700).

Conflict of interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

Publisher’s note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

References

1. Awad A, Bader-El-Den M, McNicholas J, Briggs J. Early hospital mortality prediction of intensive care unit patients using an ensemble learning approach. Int J Med Inform. (2017) 108:185–95. doi: 10.1016/j.ijmedinf.2017.10.002

PubMed Abstract | CrossRef Full Text | Google Scholar

2. Rosenberg AL, Watts C. Patients readmitted to ICUs: a systematic review of risk factors and outcomes. Chest. (2000) 118:492–502. doi: 10.1378/chest.118.2.492

PubMed Abstract | CrossRef Full Text | Google Scholar

3. Peres IT, Hamacher S, Oliveira FLC, Thomé AMT, Bozza FA. What factors predict length of stay in the intensive care unit? Systematic review and meta-analysis. J Crit Care. (2020) 60:183–94. doi: 10.1016/j.jcrc.2020.08.003

PubMed Abstract | CrossRef Full Text | Google Scholar

4. Ramos JGR, Forte DN. Accountability for reasonableness and criteria for admission, triage and discharge in intensive care units: an analysis of current ethical recommendations. Rev Bras Ter Intensiva. (2021) 33:38–47. doi: 10.5935/0103-507X.20210004

PubMed Abstract | CrossRef Full Text | Google Scholar

5. Kılıç M, Yüzkat N, Soyalp C, Gülhaş N. Cost analysis on intensive care unit costs based on the length of stay. Turk J Anaesthesiol Reanim. (2019) 47:142. doi: 10.5152/TJAR.2019.80445

PubMed Abstract | CrossRef Full Text | Google Scholar

6. Verburg IWM, Atashi A, Eslami S, Holman R, Abu-Hanna A, de Jonge E, et al. Which models can i use to predict adult icu length of stay? A systematic review. Crit Care Med. (2017) 45:e222–31. doi: 10.1097/CCM.0000000000002054

PubMed Abstract | CrossRef Full Text | Google Scholar

7. Lee H, Lim CW, Hong HP, Ju JW, Jeon YT, Hwang JW, et al. Efficacy of the APACHE II score at ICU discharge in predicting post-ICU mortality and ICU readmission in critically ill surgical patients. Anaesth Intensive Care. (2015) 43:175–86. doi: 10.1177/0310057X1504300206

PubMed Abstract | CrossRef Full Text | Google Scholar

8. Wu J, Lin Y, Li P, Hu Y, Zhang L, Kong G. Predicting prolonged length of icu stay through machine learning. Diagnostics. (2021) 11:2242. doi: 10.3390/diagnostics11122242

PubMed Abstract | CrossRef Full Text | Google Scholar

9. Lineback CM, Garg R, Oh E, Naidech AM, Holl JL, Prabhakaran S. Prediction of 30-day readmission after stroke using machine learning and natural language processing. Front Neurol. (2021) 12:649521. doi: 10.3389/fneur.2021.649521

PubMed Abstract | CrossRef Full Text | Google Scholar

10. Su L, Xu Z, Chang F, Ma Y, Liu S, Jiang H, et al. Early prediction of mortality, severity, and length of stay in the intensive care unit of sepsis patients based on sepsis 3.0 by machine learning models. Front Med. (2021) 28:664966. doi: 10.3389/fmed.2021.664966

PubMed Abstract | CrossRef Full Text | Google Scholar

11. Loreto M, Lisboa T, Moreira VP. Early prediction of ICU readmissions using classification algorithms. Comput Biol Med. (2020) 118:103636. doi: 10.1016/j.compbiomed.2020.103636

PubMed Abstract | CrossRef Full Text | Google Scholar

12. Hou N, Li M, He L, Xie B, Wang L, Zhang R, et al. Predicting 30-days mortality for MIMIC-III patients with sepsis-3: a machine learning approach using XGboost. J Transl Med. (2020) 18:462. doi: 10.1186/s12967-020-02620-5

PubMed Abstract | CrossRef Full Text | Google Scholar

13. Alsinglawi B, Alshari O, Alorjani M, Mubin O, Alnajjar F, Novoa M, et al. An explainable machine learning framework for lung cancer hospital length of stay prediction. Sci Rep. (2022) 12:607. doi: 10.1038/s41598-021-04608-7

PubMed Abstract | CrossRef Full Text | Google Scholar

14. Chung J, Gulcehre C, Cho K, Bengio Y. Empirical evaluation of gated recurrent neural networks on sequence modeling. Neural Evol Comput.. (2014): doi: 10.48550/arXiv.1412.3555

PubMed Abstract | CrossRef Full Text | Google Scholar

15. Hochreiter S, Schmidhuber J. Long short-term memory. Neural Comput. (1997) 9:1735–80. doi: 10.1162/neco.1997.9.8.1735

PubMed Abstract | CrossRef Full Text | Google Scholar

16. Nemati S, Holder A, Razmi F, Stanley MD, Clifford GD, Buchman TG. An interpretable machine learning model for accurate prediction of sepsis in the ICU. Crit Care Med. (2018) 46:547–53. doi: 10.1097/CCM.0000000000002936

PubMed Abstract | CrossRef Full Text | Google Scholar

17. Zhao Y, Zhang R, Zhong Y, Wang J, Weng Z, Luo H, et al. Statistical analysis and machine learning prediction of disease outcomes for COVID-19 and pneumonia patients. Front Cell Infect Microbiol. (2022) 12:838749. doi: 10.3389/fcimb.2022.838749

PubMed Abstract | CrossRef Full Text | Google Scholar

18. Sun Y, Kaur R, Gupta S, Paul R, Das R, Cho SJ, et al. Development and validation of high definition phenotype-based mortality prediction in critical care units. JAMIA Open. (2021) 4:ooab004. doi: 10.1093/jamiaopen/ooab004

PubMed Abstract | CrossRef Full Text | Google Scholar

19. Wernly B, Mamandipoor B, Baldia P, Jung C, Osmani V. Machine learning predicts mortality in septic patients using only routinely available ABG variables: a multi-centre evaluation. Int J Med Inform. (2021) 145:104312. doi: 10.1016/j.ijmedinf.2020.104312

PubMed Abstract | CrossRef Full Text | Google Scholar

20. Maheshwari S, Agarwal A, Shukla A, Tiwari R. A comprehensive evaluation for the prediction of mortality in intensive care units with LSTM networks: patients with cardiovascular disease. Biomed Tech. (2020) 65:435–46. doi: 10.1515/bmt-2018-0206

PubMed Abstract | CrossRef Full Text | Google Scholar

21. London AJ. Artificial intelligence and black-box medical decisions: accuracy versus explainability. Hastings Cent Rep. (2019) 49:15–21. doi: 10.1002/hast.973

PubMed Abstract | CrossRef Full Text | Google Scholar

22. Lim WX, Chen Z, Ahmed A. The adoption of deep learning interpretability techniques on diabetic retinopathy analysis: a review. Med Biol Eng Comput. (2022) 60:633–42. doi: 10.1007/s11517-021-02487-8

PubMed Abstract | CrossRef Full Text | Google Scholar

23. Gandin I, Scagnetto A, Romani S, Barbati G. Interpretability of time-series deep learning models: a study in cardiovascular patients admitted to Intensive care unit. J Biomed Inform. (2021) 121:103876. doi: 10.1016/j.jbi.2021.103876

PubMed Abstract | CrossRef Full Text | Google Scholar

24. Song H, Rajan D, Thiagarajan JJ, Spanias A. Attend and diagnose: clinical time series analysis using attention models. arXiv [Preprint]. (2017): doi: 10.48550/arXiv.1711.03905

PubMed Abstract | CrossRef Full Text | Google Scholar

25. Johnson A, Bulgarelli L, Pollard T, Horng S, Celi LA, Mark R. MIMIC-IV (version 0.4). PhysioNet. (2020). doi: 10.13026/a3wn-hq05

CrossRef Full Text | Google Scholar

26. Knaus WA, Draper EA, Wagner DP, Zimmerman JE. APACHE II: a severity of disease classification system. Crit Care Med. (1985) 13:818–29.

Google Scholar

27. Vincent JL, Moreno R, Takala J, Willatts S, De Mendonça A, Bruining H, et al. The SOFA (Sepsis-related Organ Failure Assessment) score to describe organ dysfunction/failure. on behalf of the working group on sepsis-related problems of the european society of intensive care medicine. Intensive Care Med. (1996) 22:707–10. doi: 10.1007/BF01709751

PubMed Abstract | CrossRef Full Text | Google Scholar

28. Le Gall JR, Lemeshow S, Saulnier F. A new simplified acute physiology score (SAPS II) based on a European/North American multicenter study. JAMA. (1993) 270:2957–63. doi: 10.1001/jama.270.24.2957

PubMed Abstract | CrossRef Full Text | Google Scholar

29. Grieshop S. Post-intensive care syndrome. Am J Crit Care. (2022) 31:145. doi: 10.4037/ajcc2022899

PubMed Abstract | CrossRef Full Text | Google Scholar

30. Hunter A, Johnson L, Coustasse A. Reduction of intensive care unit length of stay: the case of early mobilization. Health Care Manag. (2014) 33:128–35. doi: 10.1097/HCM.0000000000000006

CrossRef Full Text | Google Scholar

31. Golas SB, Shibahara T, Agboola S, Otaki H, Sato J, Nakae T, et al. A machine learning model to predict the risk of 30-day readmissions in patients with heart failure: a retrospective analysis of electronic medical records data. BMC Med Inform Decis Mak. (2018) 18:44. doi: 10.1186/s12911-018-0620-z

PubMed Abstract | CrossRef Full Text | Google Scholar

32. Sherman E, Alejo D, Wood-Doughty Z, Sussman M, Schena S, Ong CS, et al. Leveraging machine learning to predict 30-day hospital readmission after cardiac surgery. Ann Thorac Surg. (2021). [Epub ahead of print]. doi: 10.1016/j.athoracsur.2021.11.011

PubMed Abstract | CrossRef Full Text | Google Scholar

33. Na Pattalung T, Ingviya T, Chaichulee S. Feature explanations in recurrent neural networks for predicting risk of mortality in intensive care patients. J Pers Med. (2021) 11:934. doi: 10.3390/jpm11090934

PubMed Abstract | CrossRef Full Text | Google Scholar

34. Ofoma UR, Chandra S, Kashyap R, Herasevich V, Ahmed A, Gajic O, et al.. Findings from the implementation of a validated readmission predictive tool in the discharge workflow of a medical intensive care unit. Ann Am Thorac Soc. (2014) 11:737–43. doi: 10.1513/AnnalsATS.201312-436OC

PubMed Abstract | CrossRef Full Text | Google Scholar

35. Martin LA, Kilpatrick JA, Al-Dulaimi R, Mone MC, Tonna JE, Barton RG, et al. Predicting ICU readmission among surgical ICU patients: development and validation of a clinical nomogram. Surgery. (2019) 165:373–80. doi: 10.1016/j.surg.2018.06.053

PubMed Abstract | CrossRef Full Text | Google Scholar

36. Salet N, Stangenberger VA, Eijkenaar F, Schut FT, Schut MC, Bremmer RH, et al. Identifying prognostic factors for clinical outcomes and costs in four high-volume surgical treatments using routinely collected hospital data. Sci Rep. (2022) 12:5902. doi: 10.1038/s41598-022-09972-6

PubMed Abstract | CrossRef Full Text | Google Scholar

37. Li F, Xin H, Zhang J, Fu M, Zhou J, Lian Z. Prediction model of in-hospital mortality in intensive care unit patients with heart failure: machine learning-based, retrospective analysis of the MIMIC-III database. BMJ Open. (2021) 11:e044779. doi: 10.1136/bmjopen-2020-044779

PubMed Abstract | CrossRef Full Text | Google Scholar

38. Hu H, Lai X, Tan C, Yao N, Yan L. Factors associated with in-patient mortality in the rapid assessment of adult earthquake trauma patients. Prehosp Disaster Med. (2022) 37:299–305. doi: 10.1017/S1049023X22000693

PubMed Abstract | CrossRef Full Text | Google Scholar

39. Li X, Wang L, Zhang J, He M, Xu J. XGBoost machine learning algorism performed better than regression models in predicting mortality of moderate to severe traumatic brain injury. World Neurosurg. (2022) 163:e617–22. doi: 10.1016/j.wneu.2022.04.044

PubMed Abstract | CrossRef Full Text | Google Scholar

40. Tardif PA, Moore L, Boutin A, Dufresne P, Omar M, Bourgeois G, et al. Hospital length of stay following admission for traumatic brain injury in a Canadian integrated trauma system: a retrospective multicenter cohort study. Injury. (2017) 48:94–100. doi: 10.1016/j.injury.2016.10.042

PubMed Abstract | CrossRef Full Text | Google Scholar

41. Oh TK, Song IA, Jeon YT. Impact of Glasgow Coma Scale scores on unplanned intensive care unit readmissions among surgical patients. Ann Transl Med. (2019) 7:520. doi: 10.21037/atm.2019.10.06

PubMed Abstract | CrossRef Full Text | Google Scholar

42. Jones C. Glasgow coma scale. Am J Nurs. (1979) 79:1551–3.

Google Scholar

43. Kukoè A, Mihelèiæ A, Miko I, Romiæ A, Pražetina M, Tipura D, et al. Clinical and laboratory predictors at ICU admission affecting course of illness and mortality rates in a tertiary COVID-19 center. Heart Lung. (2022) 53:1–10. doi: 10.1016/j.hrtlng.2022.01.013

PubMed Abstract | CrossRef Full Text | Google Scholar

44. Arvaniti K, Dimopoulos G, Antonelli M, Blot K, Creagh-Brown B. Epidemiology and age-related mortality in critically Ill patients with intra-abdominal infection or sepsis: an international cohort study. Int J Antimicrob Agents. (2022) 20:106591. doi: 10.1016/j.ijantimicag.2022.106591

PubMed Abstract | CrossRef Full Text | Google Scholar

45. Sofu H, Üçpunar H, Çamurcu Y, Duman S, Konya MN, Gürsu S, et al. Predictive factors for early hospital readmission and 1-year mortality in elder patients following surgical treatment of a hip fracture. Ulus Travma Acil Cerrahi Derg. (2017) 23:245–50. doi: 10.5505/tjtes.2016.84404

PubMed Abstract | CrossRef Full Text | Google Scholar

46. Nasa P, Juneja D, Singh O. Severe sepsis and septic shock in the elderly: an overview. World J Crit Care Med. (2012) 1:23–30. doi: 10.5492/wjccm.v1.i1.23

PubMed Abstract | CrossRef Full Text | Google Scholar

47. Harazim M, Tan K, Nalos M, Matejovic M. Blood urea nitrogen - independent marker of mortality in sepsis. Biomed Pap Med Fac Univ Palacky Olomouc Czech Repub. (2022). [Epub ahead of print]. doi: 10.5507/bp.2022.015

PubMed Abstract | CrossRef Full Text | Google Scholar

48. Jamshidi E, Asgary A, Tavakoli N, Zali A, Setareh S, Esmaily H, et al. Using Machine Learning to Predict Mortality for COVID-19 Patients on Day 0 in the ICU. Front Digit Health. (2022) 3:681608. doi: 10.3389/fdgth.2021.681608

PubMed Abstract | CrossRef Full Text | Google Scholar

49. Omar HR, Guglin M. Longer-than-average length of stay in acute heart failure : determinants and outcomes. Herz. (2018) 43:131–9. doi: 10.1007/s00059-016-4532-3

PubMed Abstract | CrossRef Full Text | Google Scholar

50. Gao S, Yin G, Xia Q, Wu G, Zhu J, Lu N, et al. Development and validation of a nomogram to predict the 180-day readmission risk for chronic heart failure: a multicenter prospective study. Front Cardiovasc Med. (2021) 8:731730. doi: 10.3389/fcvm.2021.731730

PubMed Abstract | CrossRef Full Text | Google Scholar

51. Li X, Zheng R, Zhang T, Zeng Z, Li H, Liu J. Association between blood urea nitrogen and 30-day mortality in patients with sepsis: a retrospective analysis. Ann Palliat Med. (2021) 10:11653–63. doi: 10.21037/apm-21-2937

PubMed Abstract | CrossRef Full Text | Google Scholar

52. Thiele H, Ohman EM, de Waha-Thiele S, Zeymer U, Desch S. Management of cardiogenic shock complicating myocardial infarction: an update 2019. Eur Heart J. (2019) 40:2671–83. doi: 10.1093/eurheartj/ehz363

PubMed Abstract | CrossRef Full Text | Google Scholar

53. Lu X, Wang X, Gao Y, Walline JH, Yu S, Ge Z, et al. Norepinephrine use in cardiogenic shock patients is associated with increased 30 day mortality. ESC Heart Fail. (2022) 9:1875–83. doi: 10.1002/ehf2.13893

PubMed Abstract | CrossRef Full Text | Google Scholar

54. Singer KE, Sussman JE, Kodali RA, Winer LK, Heh V, Hanseman D, et al. Hitting the vasopressor ceiling: finding norepinephrine associated mortality in the critically Ill. J Surg Res. (2021) 265:139–46. doi: 10.1016/j.jss.2021.03.042

PubMed Abstract | CrossRef Full Text | Google Scholar

55. Mukhopadhyay A, Mohankumar B, Chong LS, Hildon ZJL, Tai BC, Quek SC. Factors and experiences associated with unscheduled 30-day hospital readmission: a mixed method study. Ann Acad Med Singap. (2021) 50:751–64. doi: 10.47102/annals-acadmedsg.2020522

PubMed Abstract | CrossRef Full Text | Google Scholar

Keywords: intensive care unit (ICU), mortality, length of stay, readmission, prognostic prediction, deep learning

Citation: Deng Y, Liu S, Wang Z, Wang Y, Jiang Y and Liu B (2022) Explainable time-series deep learning models for the prediction of mortality, prolonged length of stay and 30-day readmission in intensive care patients. Front. Med. 9:933037. doi: 10.3389/fmed.2022.933037

Received: 30 April 2022; Accepted: 01 September 2022;
Published: 28 September 2022.

Edited by:

Longxiang Su, Peking Union Medical College Hospital (CAMS), China

Reviewed by:

Fady Alnajjar, United Arab Emirates University, United Arab Emirates
Prashant Nasa, NMC Specialty Hospital Al Nahda, United Arab Emirates

Copyright © 2022 Deng, Liu, Wang, Wang, Jiang and Liu. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

*Correspondence: Baohua Liu, baohualiu@bjmu.edu.cn; Yong Jiang, jy78@vip.sina.com

Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.