Predicting hematoma expansion using machine learning: An exploratory analysis of the ATACH 2 trial.

ABSTRACT

INTRODUCTION: Hematoma expansion (HE) in patients with intracerebral hemorrhage (ICH) is a key predictor of poor prognosis and potentially amenable to treatment. This study aimed to build a classification model to predict HE in patients with ICH using deep learning algorithms without using advanced radiological features. METHODS: Data from the ATACH-2 trial (Antihypertensive Treatment of Acute Cerebral Hemorrhage) was utilized. Variables included in the models were chosen as per literature consensus on salient variables associated with HE. HE was defined as increase in either >33% or 6 mL in hematoma volume in the first 24 h. Multiple machine learning algorithms were employed using iterative feature selection and outcome balancing methods. 70% of patients were used for training and 30% for internal validation. We compared the ML models to a logistic regression model and calculated AUC, accuracy, sensitivity and specificity for the internal validation models respective models. RESULTS: Among 1000 patients included in the ATACH-2 trial, 924 had the complete parameters which were included in the analytical cohort. The median [interquartile range (IQR)] initial hematoma volume was 9.93.mm3 [5.03-18.17] and 25.2% had HE. The best performing model across all feature selection groups and sampling cohorts was using an artificial neural network (ANN) for HE in the testing cohort with AUC 0.702 [95% CI, 0.631-0.774] with 8 hidden layer nodes The traditional logistic regression yielded AUC 0.658 [95% CI, 0.641-0.675]. All other models performed with less accuracy and lower AUC. Initial hematoma volume, time to initial CT head, and initial SBP emerged as most relevant variables across all best performing models. CONCLUSION: We developed multiple ML algorithms to predict HE with the ANN classifying the best without advanced radiographic features, although the AUC was only modestly better than other models. A larger, more heterogenous dataset is needed to further build and better generalize the models.