Background: Stroke infarct volume predicts patient disability and has utility for clinical trial outcomes. Accurate infarct volume measurement requires manual segmentation of stroke boundaries in diffusion-weighted magnetic resonance imaging scans which is time-consuming and subject to variability. Automatic infarct segmentation should be robust to rotation and reflection; however, prior work has not encoded this property into deep learning architecture. Here, we use rotation-reflection equivariance and train a deep learning model to segment stroke volumes in a large cohort of well-characterized patients with acute ischemic stroke in different vascular territories.
Methods: In this retrospective study, patients were selected from a stroke registry at Houston Methodist Hospital. Eight hundred seventy-five patients with acute ischemic stroke in any brain area who had magnetic resonance imaging with diffusion-weighted imaging were included for analysis and split 80/20 for training/testing. Infarct volumes were manually segmented by consensus of 3 independent clinical experts and cross-referenced against radiology reports. A rotation-reflection equivariant model was developed based on U-Net and grouped convolutions. Segmentation performance was evaluated using Dice score, precision, and recall. Ninety-day modified Rankin Scale outcome prediction was also evaluated using clinical variables and segmented stroke volumes in different brain regions.
Results: Segmentation model Dice scores are 0.88 (95% CI, 0.87-0.89; training) and 0.85 (0.82-0.88; testing). The modified Rankin Scale outcome prediction AUC using stroke volume in 30 refined brain regions based upon modified Rankin Scale-relevance areas adjusted for clinical variables was 0.80 (0.76-0.83) with an accuracy of 0.75 (0.72-0.78).
Conclusions: We trained a deep learning model with encoded rotation-reflection equivariance to segment acute ischemic stroke lesions in diffusion- weighted imaging using a large data set from the Houston Methodist stroke center. The model achieved competitive performance in 175 well-balanced hold-out testing cases that include strokes from different vascular territories. Furthermore, the location specific stroke volume segmentations from the deep learning model combined with clinical factors demonstrated high AUC and accuracy for 90-day modified Rankin Scale in an outcome prediction model.
Keywords: brain ischemia; deep learning; prognosis; retrospective studies; stroke volume.