Extensive Multilabel Classification of Brain MRI Scans for Infarcts Using the Swin UNETR Architecture in Deep Learning Applications

OBJECTIVE

To distinguish infarct location and type with the utmost precision using the advantages of the Swin UNEt TRansformers (Swin UNETR) architecture.

METHODS

The research employed a two-phase training approach. In the first phase, the Swin UNETR model was trained using the Ischemic Stroke Lesion Segmentation Challenge (ISLES) 2022 dataset, which included cases of acute and subacute infarcts. The second phase involved training with data from 309 patients. The 110 categories result from classifying infarcts based on 22 specific brain regions. Each region is divided into right and left sides, and each side includes four types of infarcts (acute, acute lacunar, subacute, subacute lacunar). The unique architecture of Swin UNETR, integrating elements of both the transformer and u-net designs with a hierarchical transformer computed with shifted windows, played a crucial role in the study.

RESULTS

During Swin UNETR training with the ISLES 2022 dataset, batch loss decreased to 0.8885±0.1897, with training and validation dice scores reaching 0.4224±0.0710 and 0.4827±0.0607, respectively. The optimal model weight had a validation dice score of 0.5747. In the patient data model, batch loss decreased to 0.0565±0.0427, with final training and validation accuracies of 0.9842±0.0005 and 0.9837±0.0010.

CONCLUSION

The results of this study surpass the accuracy of similar studies, but they involve the issue of overfitting, highlighting the need for future efforts to improve generalizability. Such detailed classifications could significantly aid physicians in diagnosing infarcts in clinical settings.

Stroke classification and treatment support system artificial intelligence for usefulness of stroke diagnosis

Background and aims

It is important to diagnose cerebral infarction at an early stage and select an appropriate treatment method. The number of stroke-trained physicians is unevenly distributed; thus, a shortage of specialists is a major problem in some regions. In this retrospective design study, we tested whether an artificial intelligence (AI) we built using computer-aided detection/diagnosis may help medical physicians to classify stroke for the appropriate treatment.

Methods

To build the Stroke Classification and Treatment Support System AI, the clinical data of 231 hospitalized patients with ischemic stroke from January 2016 to December 2017 were used for training the AI. To verify the diagnostic accuracy, 151 patients who were admitted for stroke between January 2018 and December 2018 were also enrolled.

Results

By utilizing multimodal data, such as DWI and ADC map images, as well as patient examination data, we were able to construct an AI that can explain the analysis results with a small amount of training data. Furthermore, the AI was able to classify with high accuracy (Cohort 1, evaluation data 88.7%; Cohort 2, validation data 86.1%).

Conclusion

In recent years, the treatment options for cerebral infarction have increased in number and complexity, making it even more important to provide appropriate treatment according to the initial diagnosis. This system could be used for initial treatment to automatically diagnose and classify strokes in hospitals where stroke-trained physicians are not available and improve the prognosis of cerebral infarction.

Application of Machine Learning Techniques for Characterization of Ischemic Stroke with MRI Images: A Review

Magnetic resonance imaging (MRI) is a standard tool for the diagnosis of stroke, but its manual interpretation by experts is arduous and time-consuming. Thus, there is a need for computer-aided-diagnosis (CAD) models for the automatic segmentation and classification of stroke on brain MRI. The heterogeneity of stroke pathogenesis, morphology, image acquisition modalities, sequences, and intralesional tissue signal intensity, as well as lesion-to-normal tissue contrast, pose significant challenges to the development of such systems. Machine learning (ML) is increasingly being used in predictive neuroimaging diagnosis and prognostication. This paper reviews image processing and machine learning techniques that have been applied to detect ischemic stroke on brain MRI, including details on image acquisition, pre-processing, techniques to segment, extraction of features, and classification into stroke types. The main objective of this work is to find the state-of-art machine learning techniques used to predict the ischemic stroke and their application in clinical set-up. The article selection is performed according to PRISMA guideline. The state-of-the-art on automated MRI stroke diagnosis, with a focus on machine learning, is discussed, along with its advantages and limitations. We found that the various machine learning models discussed in this article are able to detect the infarcts with an acceptable accuracy of 70–90%. However, no one has highlighted the time complexity to predict the stroke in the model developed, which is an important factor. The work concludes with proposals for future recommendations for building efficient and robust deep learning (DL) models for quantitative brain MRI analysis. In recent work, with the application of DL approaches, using large datasets to train the models has improved the detection accuracy and reduced computational complexity. We suggest that the design of a decision support system based on artificial intelligence (AI) and clinical data presenting symptoms is essential to support clinicians to accelerate diagnosis and timeous therapy in the emergency management of stroke.