Ensemble Learning-based Brain Stroke Prediction Model Using Magnetic Resonance Imaging

Abstract

Brain stroke (BS) imposes a substantial burden on healthcare systems due to the long-term care and high expenditure. Earlier detection and intervention can reduce the impact of BS. Magnetic resonance imaging (MRI) is commonly applied for BS detection. Deep learning techniques can employ MRI images to identify the BS risks in the initial stages. This study developed a BS detection model using an ensemble learning approach that combines the predictions of the base models. A MobileNet V3 model backbone was used to extract the intricate patterns of BS from MRI images. LightGBM and CatBoost models were used as base models to predict BS using the extracted features. In addition, the random forest model was used to integrate the predictions of base models to identify BS. The proposed model was generalized on a public MRI dataset that covers 2888 clinical MRI images. The experimental outcomes showed the effectiveness of the suggested BS detection model. The proposed model has obtained an accuracy of 98.7%, an area under the receiver operating characteristic score of 0.95, and an area under the precision–recall curve of 0.92. The recommended model is believed to be deployed in real-time healthcare settings to assist radiologists and clinicians in making effective decisions.