HE-Mind: A model for automatically predicting hematoma expansion after spontaneous intracerebral hemorrhage

Abstract

PurposeTo develop and validate an end-to-end model for automatically predicting hematoma expansion (HE) after spontaneous intracerebral hemorrhage (sICH) using a novel deep learning framework. MethodsThis multicenter retrospective study collected cranial noncontrast computed tomography (NCCT) images of 490 patients with sICH at admission for model training (n = 236), internal testing (n = 60), and external testing (n = 194). A HE-Mind model was designed to predict HE, which consists of a densely connected U-net for segmentation process, a multi-instance learning strategy for resolving label ambiguity and a Siamese network for classification process. Two radiomics models based on support vector machine or logistic regression and two deep learning models based on residual network or Swin transformer were developed for performance comparison. Reader experiments including physician diagnosis mode and artificial intelligence mode were conducted for efficiency comparison. ResultsThe HE-Mind model showed better performance compared to the comparative models in predicting HE, with areas under the curve of 0.849 and 0.809 in the internal and external test sets respectively. With the assistance of the HE-Mind model, the predictive accuracy and work efficiency of the emergency physician, junior radiologist, and senior radiologist were significantly improved, with accuracies of 0.768, 0.789, and 0.809 respectively, and reporting times of 7.26 s, 5.08 s, and 3.99 s respectively. ConclusionsThe HE-Mind model could rapidly and automatically process the NCCT data and predict HE after sICH within three seconds, indicating its potential to assist physicians in the clinical diagnosis workflow of HE.