Developing practical machine learning survival models to identify high-risk patients for in-hospital mortality following traumatic brain injury.

ABSTRACT: Machine learning (ML) offers precise predictions and could improve patient care, potentially replacing traditional scoring systems. A retrospective study at Emtiaz Hospital analyzed 3,180 traumatic brain injury (TBI) patients. Nineteen variables were assessed using ML algorithms to predict outcomes. Data preparation addressed missing values and balancing methods corrected imbalances. Model building involved training-test splits, survival analysis, and ML algorithms like Random Survival Forest (RSF) and Gradient Boosting. Feature importance was examined, with patient risk stratification guiding survival analysis. The best-performing model, RSF with ROS resampling, achieved the highest mean AUC of 0.80, the lowest IBS of 0.11, and IPCW c-index of 0.79, maintaining strong predictive ability o

SUBMITTER: Andishgar A 

PROVIDER: S-EPMC11836111 | biostudies-literature | 2025 Feb

REPOSITORIES: biostudies-literature