A Semi-Supervised Learning Framework to Leverage Proxy Information for Stroke MRI Analysis

Jennifer Polson,Haoyue Zhang,Kambiz Nael,Noriko Salamon,Bryan Yoo,Namkug Kim,Dong-Wha Kang,William Speier,Corey W Arnold
DOI: https://doi.org/10.1109/EMBC46164.2021.9631098
Abstract:Treating acute ischemic stroke (AIS) patients is a time-sensitive endeavor, as therapies target areas experiencing ischemia to prevent irreversible damage to brain tissue. Depending on how an AIS is progressing, thrombolytics such as tissue-plasminogen activator (tPA) may be administered within a short therapeutic window. The underlying conditions for optimal treatment are varied. While previous clinical guidelines only permitted tPA to be administered to patients with a known onset within 4.5 hours, clinical trials demonstrated that patients with signal intensity differences between diffusion-weighted imaging (DWI) and fluid-attenuated inversion recovery (FLAIR) sequences in an MRI study can benefit from thrombolytic therapy. This intensity difference, known as DWI-FLAIR mismatch, is prone to high inter-reader variability. Thus, a paradigm exists where onset time serves as a weak proxy for DWI-FLAIR mismatch. In this study, we sought to detect DWI-FLAIR mismatch in an automated fashion, and we compared this to assessments done by three expert neuroradiologists. Our approach involved training a deep learning model on MRI to classify tissue clock and leveraging time clock as a weak proxy label to supplement training in a semi-supervised learning (SSL) framework. We evaluate our deep learning model by testing it on an unseen dataset from an external institution. In total, our proposed framework was able to improve detection of DWI-FLAIR mismatch, achieving a top ROC-AUC of 74.30%. Our study illustrated that incorporating clinical proxy information into SSL can improve model optimization by increasing the fidelity of unlabeled samples included in the training process.
What problem does this paper attempt to address?