Machine learning models for predicting early hemorrhage progression in traumatic brain injury
Heui Seung Lee,Ji Hee Kim,Jiye Son,Hyeryun Park,Jinwook Choi
DOI: https://doi.org/10.1038/s41598-024-61739-3
IF: 4.6
2024-05-24
Scientific Reports
Abstract:This study explores the progression of intracerebral hemorrhage (ICH) in patients with mild to moderate traumatic brain injury (TBI). It aims to predict the risk of ICH progression using initial CT scans and identify clinical factors associated with this progression. A retrospective analysis of TBI patients between January 2010 and December 2021 was performed, focusing on initial CT evaluations and demographic, comorbid, and medical history data. ICH was categorized into intraparenchymal hemorrhage (IPH), petechial hemorrhage (PH), and subarachnoid hemorrhage (SAH). Within our study cohort, we identified a 22.2% progression rate of ICH among 650 TBI patients. The Random Forest algorithm identified variables such as petechial hemorrhage (PH) and countercoup injury as significant predictors of ICH progression. The XGBoost algorithm, incorporating key variables identified through SHAP values, demonstrated robust performance, achieving an AUC of 0.9. Additionally, an individual risk assessment diagram, utilizing significant SHAP values, visually represented the impact of each variable on the risk of ICH progression, providing personalized risk profiles. This approach, highlighted by an AUC of 0.913, underscores the model's precision in predicting ICH progression, marking a significant step towards enhancing TBI patient management through early identification of ICH progression risks.
multidisciplinary sciences
What problem does this paper attempt to address?
This paper aims to solve the problem of risk prediction for the progression of intracranial hemorrhage (ICH) in patients with traumatic brain injury (TBI). Specifically, the objective of the study is to use the initial CT scan to predict the risk of ICH progression and identify the clinical factors associated with this progression. The study retrospectively analyzed the data of TBI patients from January 2010 to December 2021, focusing on the initial CT assessment as well as demographic, comorbidity, and medical history data. The study classified ICH into intraparenchymal hemorrhage (IPH), punctate hemorrhage (PH), and subarachnoid hemorrhage (SAH). Among 650 TBI patients, 22.2% had ICH progression. The random forest algorithm identified punctate hemorrhage (PH) and contrecoup injury as important predictors of ICH progression. The XGBoost algorithm combined with the SHAP values of key variables demonstrated strong performance, achieving an AUC value of 0.9. In addition, the individual risk assessment map based on significant SHAP values intuitively shows the influence of each variable on the risk of ICH progression, providing a personalized risk profile. This method significantly improves the management of TBI patients by early identification of the risk of ICH progression.