Machine Learning for Predicting In-Hospital Mortality After Traumatic Brain Injury in Both High-Income and Low- and Middle-Income Countries

Abstract

Background: Machine learning (ML) holds promise as a tool to guide clinical decision making by predicting in-hospital mortality for patients with traumatic brain injury (TBI). Previous models such as the international mission for prognosis and clinical trials in TBI (IMPACT) and the corticosteroid randomization after significant head injury (CRASH) prognosis calculators can potentially be improved with expanded clinical features and newer ML approaches.

Objective: To develop ML models to predict in-hospital mortality for both the high-income country (HIC) and the low- and middle-income country (LMIC) settings.

Methods: We used the Duke University Medical Center National Trauma Data Bank and Mulago National Referral Hospital (MNRH) registry to predict in-hospital mortality for the HIC and LMIC settings, respectively. Six ML models were built on each data set, and the best model was chosen through nested cross-validation. The CRASH and IMPACT models were externally validated on the MNRH database.

Results: ML models built on National Trauma Data Bank (n = 5393, 84 predictors) demonstrated an area under the receiver operating curve (AUROC) of 0.91 (95% CI: 0.85-0.97) while models constructed on MNRH (n = 877, 31 predictors) demonstrated an AUROC of 0.89 (95% CI: 0.81-0.97). Direct comparison with CRASH and IMPACT models showed significant improvement of the proposed LMIC models regarding AUROC (P = .038).

Conclusion: We developed high-performing well-calibrated ML models for predicting in-hospital mortality for both the HIC and LMIC settings that have the potential to influence clinical management and traumatic brain injury patient trajectories.