Predicting the Presence of Traumatic Chest Injuries Using Machine Learning Algorithm
Various tools have been developed to determine the priority of radiography in trauma patients. Thisstudy aimed to investigate the role of machine learning models in predicting chest injuries following multiple trauma.
We used the database of a comprehensive cross-sectional survey conducted in 2015. Eight machine learningmodels were developed using demographic characteristics, physical exam findings, and radiologic results of 2860 pa-tients.
Area under the receiver operating characteristic curve (AUC) was greater than 0.96 in Random Forest,Gradient Boosting, XGBoost, Decision Tree, Support Vector Machine (SVM), Logistic Regression, K-Nearest Neighbors(KNN), and Neural Network models. The random forest model, XGBoost and Gradient Boosting had the highest accu-racy (0.99). Sensitivity was also highest in the Gradient Boosting, XGBoost and KNN models (0.99). The specificity ofall of the models in predicting chest radiography outcomes of multiple trauma patients was higher than 0.97, exceptfor logistic regression and SVM (0.912 and 0.885 respectively).
Our study highlights the strong potential ofmachine learning models, especially Random Forest and Gradient Boosting, in predicting chest trauma outcomes withhigh accuracy and sensitivity