Title
Prediction of Extubation Failure for Intensive Care Unit Patients Using Light Gradient Boosting Machine
Abstract
Extubation failure is a complex and ongoing problem in the intensive care unit (ICU). It refers to the patients who require re-intubation after extubation (namely disconnection from mechanical ventilation). In these patients, extubation failure leads to severe risks associated with re-intubation and is associated with increased mortalities, longer stay in ICU and also higher health care costs. Many studies have been proposed to analyze the problem of extubation failure and identify possible factors or indices that may predict extubation failure. However, these studies used a small number of patients for extubation failure and limited their features to several vital signs or main characteristics. We argue that these are insufficient and less accurate for the prediction of extubation failure. In this paper, we analyze 3636 adult patient records in the MIMIC-III clinical database and apply the Light Gradient Boosting Machine (LightGBM) to predict extubation failure. Also, we perform feature importance analysis according to the result of LightGBM and interpret these features using <underline>SH</underline>apley <underline>A</underline>dditive ex<underline>P</underline>lanations (SHAP). Experimental results show that our LightGBM method is effective in predicting extubation failure and outperform other machine learning methods such as artificial neural network (ANN), logistic regression (LR) and support vector machine (SVM). The results of feature importance and SHAP analysis are also proved effective and accurate.
Year
DOI
Venue
2019
10.1109/ACCESS.2019.2946980
IEEE ACCESS
Keywords
DocType
Volume
Ventilation,Predictive models,Feature extraction,Diseases,Blood pressure,Blood,MIMICs,Extubation failure prediction,feature importance,light gradient boosting machine,shapley additive explanations
Journal
7
ISSN
Citations 
PageRank 
2169-3536
0
0.34
References 
Authors
0
8
Name
Order
Citations
PageRank
Tingting Chen184.49
Jun Xu200.34
Haochao Ying37310.03
Xiaojun Chen400.34
Ruiwei Feng524.41
Xueling Fang600.34
Honghao Gao721745.24
Jian Wu893395.62