A machine learning approach for predicting perihematomal edema expansion in patients with intracerebral hemorrhage

Objectives Preventing the expansion of perihematomal edema (PHE) represents a novel strategy for the improvement of neurological outcomes in intracerebral hemorrhage (ICH) patients. Our goal was to predict early and delayed PHE expansion using a machine learning approach. Methods We enrolled 550 patients with spontaneous ICH to study early PHE expansion, and 389 patients to study delayed expansion. Two imaging researchers rated the shape and density of hematoma in non-contrast computed tomography (NCCT). We trained a radiological machine learning (ML) model, a radiomics ML model, and a combined ML model, using data from radiomics, traditional imaging, and clinical indicators. We then validated these models on an independent dataset by using a nested 4-fold cross-validation approach. We compared models with respect to their predictive performance, which was assessed using the receiver operating characteristic curve. Results For both early and delayed PHE expansion, the combined ML model was most predictive (early/delayed AUC values were 0.840/0.705), followed by the radiomics ML model (0.799/0.663), the radiological ML model (0.779/0.631), and the imaging readers (reader 1: 0.668/0.565, reader 2: 0.700/0.617). Conclusion We validated a machine learning approach with high interpretability for the prediction of early and delayed PHE expansion. This new technique may assist clinical practice for the management of neurocritical patients with ICH. Key Points • This is the first study to use artificial intelligence technology for the prediction of perihematomal edema expansion. • A combined machine learning model, trained on data from radiomics, clinical indicators, and imaging features associated with hematoma expansion, outperformed all other methods.