Detection and delineation of acute cerebral infarct on DWI using weakly supervised machine learning

Abstract

Improved outcome in patients with ischemic stroke is achieved through acute diagnosis and early restoration of cerebral flow in appropriate patients. Diffusion-weighted MR imaging (DWI) plays a central role in these efforts by enabling rapid early localization and quantification of ischemic lesions. Automated detection and quantification can potentially accelerate diagnosis, improve treatment safety and efficacy and reduce costs. However, the manual quantification of acute ischemic stroke volumes for algorithm training is time consuming and imprecise. We present YNet as a novel fully-automated deep learning algorithm for detection and volumetric segmentation and quantification of acute cerebral ischemic lesions from DWI. The algorithm is a semi-supervised multi-tasking deep neural network architecture we developed that enables the combination of both weak labels derived from radiology report classification and manually delineated pixel level training data. The model is trained on a very large dataset of 10000 studies, achieves detection sensitivity 0.981, detection specificity 0.980 and segmentation Dice score 0.623 on a heterogeneous test set.