Accepted for/Published in: JMIR Medical Informatics
Date Submitted: May 28, 2021
Date Accepted: Jul 27, 2021
Date Submitted to PubMed: Aug 4, 2021
XGBoost Algorithm Prediction of Critical Care Outcome for Adult Patients Presenting to Emergency Department Using Initial Triage Information
ABSTRACT
Background:
Emergency department (ED) triage system to classify and prioritize patients at high risk from less urgent continues to be a challenge.
Objective:
This study, comprising 80,433 patients, aims to develop a machine learning algorithm prediction model of critical care outcome for adult patients using information collected during ED triage and compare the performance with that of the baseline model using Korean Triage and Acuity Scale (KTAS).
Methods:
To predict the need of critical care, we used 13 predictors from triage information: age, gender, mode of ED arrival, time interval between onset and ED arrival, reason of ED visit, chief complaints, systolic blood pressure, diastolic blood pressure, pulse rate, respiratory rate, body temperature, oxygen saturation and level of consciousness. The baseline model with KTAS was developed using logistic regression and the machine learning model with 13 variables was generated using extreme gradient boosting (XGB) and deep neural network (DNN) algorithms. The discrimination was measured by area under the receiver operating characteristic curve(AUC). The ability of calibration with Hosmer–Lemeshow test and reclassification with net reclassification index (NRI) were evaluated. The calibration plot and partial dependence plot were used in analysis.
Results:
The AUC of the model with the full set of variables (0.833–0.861) was better than that of the baseline model (0.796). The XGB model of AUC 0.861 (0.848, 0.874, 95% CI) showed a higher discriminative performance than DNN model of 0.833(0.819, 0.848). The XGB and DNN models proved better reclassification than the baseline model with positive net reclassification index. The XGB models was well calibrated (Hosmer-Lemeshow test p>0.05); however, the DNN showed poor calibration power (H-L test p<0.001). We further interpreted non-linear association between variables and critical care prediction.
Conclusions:
Our study demonstrated that the performance of the XGB model using initial information at ED triage for predicting patients in need of critical care outperformed the conventional model with KTAS.
Citation
Request queued. Please wait while the file is being generated. It may take some time.
Copyright
© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.