Hybrid deep learning and metaheuristic model based stroke diagnosis system using electroencephalogram (EEG)

Abstract

Over the last few decades, there has been a significant increase in the average lifespan. Consequently, the number of elderly people suffering from strokes has also risen. As a result, strokes and their treatments have become crucial subjects of research, particularly for the application of machine learning. One of the primary factors in stroke treatment is the speed of response. Currently, both computed tomography (CT) and magnetic resonance imaging (MRI) are used to diagnose strokes. However, CT takes eight hours before an accurate diagnosis can be made, and MRI is expensive and not available in all hospitals. Therefore, there is a growing need for novel approaches to identifying strokes based on electroencephalogram (EEG) signals. In this paper, a hybrid model of deep learning and metaheuristic was developed in the offline stage to classify strokes. Since EEG data is a time series with frequencies, a hybrid model was deemed appropriate. This hybrid model combined a Convolutional Neural Network (CNN) with bidirectional Gated Recurrent Unit (BiGRU). The performance of this model surpassed that of other comparable models. Given the paramount importance of speed and accuracy in this work, the harmony search (HS) algorithm, which is specialized in handling frequencies, was used for feature selection. HS outperformed all similar algorithms when applied to the CNN-BiGRU hybrid model. Additionally, for the optimization of continuous hyperparameters, the multiverse optimization (MVO) algorithm was employed, which proved to be the most effective when compared to another similar algorithm for validation purposes. The new model, CNN-BiGRU-HS-MVO, was applied to analyze the data collected from Al Bashir Hospital using the MUSE-2 portable device, resulting in an impressive prediction accuracy of 99.991%. Moreover, it demonstrated an 11.08% improvement over the results from the paper titled “Predicting stroke severity with a 3-min recording from the Muse portable EEG study”. Furthermore, a decision support system was built on the cloud computing environment based on the hybrid model. This system allows for the diagnosis of patients anytime and from anywhere within minutes, with the authorized person receiving the diagnosis results through SMS notification.