[go: up one dir, main page]

Next Article in Journal
Development of a Transient Magnetic Field Sensor Based on Digital Integration and Frequency Equalization
Previous Article in Journal
Towards Semantic Integration of Machine Vision Systems to Aid Manufacturing Event Understanding
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Deep Learning-Based Stroke Disease Prediction System Using Real-Time Bio Signals

1
KEPCO Research Institute, Korea Electric Power Corporation, 105 Munji-ro Yuseong-gu, Daejeon 34056, Korea
2
Research Team for Health & Safety Convergence, Korea Research Institute of Standards and Science (KRISS), Daejeon 34113, Korea
3
Department of KSB Convergence Research, Electronics and Telecommunications Research Institute (ETRI), Daejeon 34129, Korea
4
Department of Rehabilitation Medicine, Chungnam National University College of Medicine, 266 Munhwa-ro Jung-gu, Daejeon 35015, Korea
5
School of Creative Convergence, Andong National University, 1375 Gyeongdong-ro (Songcheon-dong), Andong, Gyeongsangbuk-do 36729, Korea
*
Authors to whom correspondence should be addressed.
Sensors 2021, 21(13), 4269; https://doi.org/10.3390/s21134269
Submission received: 10 May 2021 / Revised: 15 June 2021 / Accepted: 16 June 2021 / Published: 22 June 2021
(This article belongs to the Section Sensing and Imaging)

Abstract

:
The emergence of an aging society is inevitable due to the continued increases in life expectancy and decreases in birth rate. These social changes require new smart healthcare services for use in daily life, and COVID-19 has also led to a contactless trend necessitating more non-face-to-face health services. Due to the improvements that have been achieved in healthcare technologies, an increasing number of studies have attempted to predict and analyze certain diseases in advance. Research on stroke diseases is actively underway, particularly with the aging population. Stroke, which is fatal to the elderly, is a disease that requires continuous medical observation and monitoring, as its recurrence rate and mortality rate are very high. Most studies examining stroke disease to date have used MRI or CT images for simple classification. This clinical approach (imaging) is expensive and time-consuming while requiring bulky equipment. Recently, there has been increasing interest in using non-invasive measurable EEGs to compensate for these shortcomings. However, the prediction algorithms and processing procedures are both time-consuming because the raw data needs to be separated before the specific attributes can be obtained. Therefore, in this paper, we propose a new methodology that allows for the immediate application of deep learning models on raw EEG data without using the frequency properties of EEG. This proposed deep learning-based stroke disease prediction model was developed and trained with data collected from real-time EEG sensors. We implemented and compared different deep-learning models (LSTM, Bidirectional LSTM, CNN-LSTM, and CNN-Bidirectional LSTM) that are specialized in time series data classification and prediction. The experimental results confirmed that the raw EEG data, when wielded by the CNN-bidirectional LSTM model, can predict stroke with 94.0% accuracy with low FPR (6.0%) and FNR (5.7%), thus showing high confidence in our system. These experimental results demonstrate the feasibility of non-invasive methods that can easily measure brain waves alone to predict and monitor stroke diseases in real time during daily life. These findings are expected to lead to significant improvements for early stroke detection with reduced cost and discomfort compared to other measuring techniques.

1. Introduction

Stroke is a condition involving abnormalities in the brain blood vessels that result in dysfunction in certain brain locations [1]. According to a 2016 report by the World Health Organization (WHO), stroke is the second most common global cause of death in the world and the third most common global cause of disability [2]. The incidence of stroke in developing countries has more than doubled over the past 40 years [3]. Since a suitable treatment for stroke has yet to be found, early detection is paramount. CT and MRI techniques are the most common detection methods for stroke disease. However, CT and MRI are expensive and may not be suitable for developing countries or low-income earners. With stroke disease emerging as an important disease worldwide, particularly among low-income and elderly people, healthcare services desperately need a solution to help them accurately and quickly detect stroke diseases at a low cost.
Studies on the early detection and prediction of stroke are actively underway. The 2019 Global Burden Disease (GBD) study estimated cardiovascular disease incidence and patient mortality from 204 countries and regions from 1990 to 2019 [4]. According to that report, the number of deaths from cardiovascular disease in 2019 accounted for one-third of all deaths. While the death toll according to cardiovascular index rose from 11.1 million in 1990 to 18.6 million in 2019, many of the causes from cardiovascular disease were attributed to ischemic stroke. Hier et al. also reported that ischemic stroke had a very high recurrence rate of 14.1% within two years [5]. Further, recent studies have shown a link between COVID-19 and stroke, which is expected to increase the number of stroke deaths [6,7]. Kummer et al. [6] informs us that patients admitted with COVID-19 who had a stroke history were much more likely to die than those without a stroke history, while Zhang et al. [7] reported that patients with a stroke prognosis had a higher incidence of severe pneumonia and subsequent mortality according to Cox regression. Aside from these prior studies, there is still a significant lack of understanding between experimental data and data collected in real-time for stroke.
Stroke disease can be identified using blood tests; brain imaging such as CT, MRI, and X-ray; ECG and EEG; and neurological physiological methods such as induced potential tests [8]. Among these techniques, CT and MRI are most often used to determine stroke, but these involve risks such as exposure to radiation or potential allergic reactions to the contrast agents used. These tools can also be inconvenient as they involve confined spaces and require constant monitoring while also incurring separate medical costs for each examination, all of which increase the difficulty of diagnosis. New wearable electrodes provide an opportunity to measure EEG in the comfort of a participant’s home. These electrodes are attached to the head and measure the activity of brain nerve cells in a more natural setting. Brain signals are also recorded during the different sleep states, thereby allowing for pain-free and rapid examination. EEG data can be contaminated by the patient’s movement as well as environmental noises. However, it is possible to collect and test EEG data in real time and in a low-cost manner with fewer side effects than the aforementioned imaging techniques. Due to these advantages, 24-h EEG measurements are considered a useful, low-cost method for monitoring stroke disease with high recurrence rates in daily life.
According to the literature review, a number of studies have analyzed diseases such as seizure, Alzheimer’s, and stroke using EEG, while other studies have also correlated the level of sleep and emotions using EEG testing [9,10,11,12,13]. However, most of these studies focused on simple classifications or used preset frequency features extracted for experimentation. Thus, additional time and cost are needed to separate raw data into frequency domains, meaning that real-time health monitoring is not currently feasible with the technology that has been reported to date. A recent study was able to classify seizure patients using raw EEG data alone, representing a promising step toward real-time health monitoring using EEG [14,15]. Therefore, in this paper, we propose a real-time stroke monitoring system to predict the degree of risk of stroke in real time by collecting raw EEG data.
Using real-time EEG data, our system can predict stroke diseases in elderly Koreans. To test this system, we developed a walking protocol that reflects the everyday life of an elderly Korean. The EEG data used in this work were measured and collected from Korean seniors aged 65 or older. To compare between deep learning models and machine learning models, the collected EEG data were separated into raw data and frequency domain-extracted data, respectively. Each set of raw EEG data from six channels (Fz, T7, C1, C2, T8, and Oz) undergoes a fast Fourier transform (FFT), and 66 values in total were extracted and used in the experiment. The initial results show that the raw data alone could be used to predict stroke disease with high accuracy. To determine which deep learning model is the most suitable for real-time EEG data, a comparison of the predictive accuracies of LSTM, bidirectional LSTM, CNN-LSTM, and CNN-bidirectional LSTM models was conducted. These algorithms were chosen because they are known to be suitable for real-time data learning based on the characteristics of EEG data. Our experiments showed 94.0% accuracy for CNN-Bidirectional LSTM models with very low false negative rate (FNR) and false positive rate (FPR) at 5.7% and 6.0%, respectively, confirming high confidence in the results. Meanwhile, using power value, we showed 81.4% accuracy along with 18.5% FPR and 17.3% FNR in CNN-bidirectional LSTM models. Further, experiments using relative value showed 89.2% accuracy as well as FPR and FNR of 12.5% and 8.4%, respectively, in CNN-LSTM models. These experimental results indicate that raw EEG data alone can be used to accurately detect and predict stroke diseases without separating the frequency attribute values. We also confirmed that the system proposed in this paper accurately predicts the precursor symptoms of stroke disease with very high mortality and recurrence rates in real time, and thus represents a low-cost method that enables health monitoring during the daily lives of the elderly.
The paper is organized as follows. Section 2 examines prior research involved in EEG features in stroke patients as well as computer engineering studies related to stroke prediction. Section 3 explores deep learning-based stroke disease prediction systems with real-time brainwave data proposed in the paper, and also discusses prediction methodologies using raw data and frequency properties of brainwaves. Section 4 discusses the real-time stroke disease prediction experiments conducted in this study and the analysis results with deep learning models specialized in time series signal data prediction, while Chapter 5 concludes and proposes future study directions.

2. Related Works

2.1. Related Work on Stroke EEG

Wijaya et al. [16] proposed a method for detecting ischemic stroke disease through various signal process and machine learning techniques. They were able to accurately classify 40 stroke patients and 40 normal people using EEG data from six channels. The data underwent various types of signal processing such as fast Fourier transform (FFT), wavelet transform (WT), short-time Fourier transform (SFFT), and power spectral density (PSD) techniques before multiple frequency domains data were subjected to a multi-layer perceptron (MLP) and decision tree techniques for the detection of ischemic stroke. The proposed system had a test accuracy of 95%. Rahma et al. [17] used an extreme Llearning machine (ELM) to automatically classify Athens Insomnia Scale (AIS) severity based on EEG signals through digital signal processing such as feedforward and WT structures. Their algorithm confirms the usefulness of EEG in signal-based AIS classification with more than 72% test accuracy, sensitivity, and specificity according to the National Institutes of Health Stroke Scale (NIHSS). Li et al. [18] et al. introduced various machine learning methodologies for stroke risk level classification in Chinese stroke screening and presented the experimental results. In particular, the experimental results were analyzed with more than 8 machine learning algorithms, including the logistic regression model. In the experiment using boosting decision tree, the results of recall and precision were 99.94% and 97.33%, respectively. Li et al. suggested an improvement plan for the screening method program that is actually used up to now. ELM was also used by Adhi et al. [19] via detrended fluctuation analysis (DFA) for the automatic detection of ischemic stroke and normal controls. Signal processing was performed on 18 channels, and the subject’s scale index was used as an input to the ELM to classify ischemic stroke. These experiments used 120 hidden neurons and sines as an activation function of EML and obtained 84% accuracy, 87% specificity, and 82% sensitivity. Djamal et al. [20] proposed a method for identifying stroke patients after the occurrence of stroke using a convolutional neural network (CNN). They used wavelets to extract brainwave signal information for use as a feature in machine learning that reflects the patient’s condition after stroke. The extracted features are alpha, beta, theta, delta, and mu waves of brain waves, and the accuracy of the test data was 90% with amplitude and beta features while it was 70% without them.
Studies that analyze EEG characteristics in stroke patients have shown that there are distinct characteristics in three frequency ranges: alpha wave (8~13 Hz), theta wave (4~8 Hz), and delta wave (1~4 Hz). Schneider et al. [21] analyzed brain wave frequency and brain wave topography in 20 mild stroke patients. Their results confirmed a meaningful decrease in alpha wave and increase in delta wave activity in 17 mild stroke patients. In another study, Finnigan et al. [22] reported that abnormal waveforms in the alpha and delta wave were found in stroke patients. Specifically, in stroke patients, the delta wave showed abnormal and slower characteristics than that in normal people, while the alpha wave showed reduced normal and fast activity. This information can be used to detect brain waves in stroke patients using the values of delta, delta and alpha power ratio (DAR), and power ratio index (PRI). Varelas et al. [23] reported that a rhythmic and large amplitude sum of theta and delta power values appeared in stroke patients with additional epilepsy. Ip et al. [24] confirmed the occurrence of changes in theta/delta values in stroke patients, as well as a sharp increase in the values of delta (δ), alpha (α), beta (β) (14~30 Hz), and high gamma (γ) (over 30 Hz) in the right brain hemisphere after stroke. Kim et al. [25] conducted a follow-up and made behavioral observations for three minutes in 12 stroke patients and a normal control group.

2.2. Related Work on AI-based Stroke Classification

Currently, many deep learning-based studies use CT or MRI images to detect stroke [26,27,28,29,30,31,32]. For example, in a study classifying hemorrhagic stroke and ischemic stroke using brain CT images, Gautam et al. [26] achieved a classifier performance of up to 98.77%. Kalchbrenner et al. demonstrated that their proposed 13-layer CNN [27] model showed better performance in comparative experiments with AlexNet [28] and ResNET50 [29]. Chin et al. [30] applied data augmentation to CT images of ischemic stroke patients to expand the number of patch images, used them as input to CNN models to detect ischemic stroke, and obtained more than 90% accuracy. Liu et al. [31] proposed a Res-CNN model that automatically classifies acute ischemic stroke in MRIs. The Res-CNN model solved the performance degradation problem using the residual unit, and it improved the model performance through data expansion. Dourado et al. [32] presented a CT image classification IoT framework applied with a CNN to classify ischemic stroke and hemorrhagic stroke. In that same framework, they further experimented with machine learning methods like Bayesian [33], MLP [34], k-nearest neighbor [35], random forest [36], and support vector machines (SVM) [37] by applying the various machine learning concept [38], and they validated the model with 100% accuracy. They reported that training and testing times of 0.015 s and 0.001 s, respectively, when using the Bayesian Classifier, can be classified as ischemic stroke and hemorrhagic stroke with high accuracy in a short time.
Beyond images, various biological signals have also been used to predict stroke diseases. For example, Yu et al. [39] proposed a pre-detection and prediction method for machine learning and deep learning-based stroke diseases that measure the electrical activities of thighs and calves with EMG biological signal sensors, which can easily be used to acquire data during daily activities. They experimentally verified an accuracy of more than 90% using real-time collected data. Through these experiments, that study demonstrated a novel method that can verify stroke disease with high accuracy based on the pre-symptoms of stroke, body falling, and the degrees of both leg muscles. In another study, Xie et al. [40] proposed CNN-based DenseNet for stroke disease classification and prediction based on ECG data collected using 12 leads, and they obtained 99.99% training accuracy and 85.82% testing accuracy using fine-tuned models for the correlation between stroke and ECG. However, they used other biological signals that are not closely related to the brain. Thus, there is a need for studies using brain waves with AI.
Fawaz et al. [41] conducted a study to expand data using techniques such as conditional general adversarial network (CGAN) [42] to address the lack of data in stroke patients, and to classify EEG in stroke patients through the LSTM encoding process and the frequency mapping phase for loss functions. The proposed model uses frequency properties, and its performance is compared with those of the typical deep learning models of feed forward [43], CNN [27,28], bidirectional LSTM [44], and CNN-bidirectional LSTM [45] as baseline models. The model proposed in that study achieved a high accuracy of 90.5% for classifying stroke patients. In a different study, Guntari et al. [46] classified poststroke EEG signals using RNN and genetic algorithms (GA). The stroke data used in Zhang et al. were extracted using wavelet [47], and the initial 215,040 data points were reduced to 29,160 on 14 channels containing power values; this reduced the figures time from 80 s to 50 s, and an experimental accuracy of 90.0% was obtained. Giri et al. [48] conducted a stroke identification study based on EEG signals and electrooculography (EOG) signals using 1D CNN and batch normalization. The collected data consisted of the relative values, correlation aspects, variance, spectral mean, entropy, kurtosis, and fractal indices of brain waves through feature extraction steps, and an F-Score of 86.1% was ultimately obtained.

3. Deep Learning-Based Stroke Disease Prediction System Using EEG

We propose a novel system for predicting stroke based on deep learning using the raw and attribute values of EEG collected in real time, as presented in Figure 1. The proposed system is composed of (1) a module that collects data in real time; (2) a module that transmits the real-time generated biological signals to the server; (3) a module that analyzes the stored biological data, then extracts and manages frequency attributes; (4) a deep learning-based learning and prediction module; and (5) a biological-signal-based stroke prediction analysis. Next, the initial system extracts and stores a set of pretreatment processes and features based on various biological signal data, such as EEG data and ECG data collected from elderly people. Finally, the stored bio-signal data is used as input for deep learning models to predict and analyze stroke anomalies in older adults in real time.

3.1. Sensor Device and Data Collection Module

We measured and collected various bio-signal data, such as electroencephalography (EEG), electrocardiography (ECG), electromyography (EMG), motion, etc., for the classification and prediction of stroke diseases using bio-signals. Although all bio-signal data were collected in real time, we only used the EEG data to validate the stroke disease prediction and analysis. There are two main methods of measuring brain wave data. In the first method, scalp EEG can be measured painlessly by attaching electrodes to the outside of the skull, i.e., to the scalp. In the second method, electrodes can be surgically inserted into the subject’s cerebral cortex. This method provides a relatively neat signal, but it is highly invasive and involves the risk of incision into the skull. In this paper, the data are acquired using non-invasive scalp EEGs to allow for easier data collection during daily activities and verification accuracy for stroke prediction.

3.2. Data Preprocessing Module

Once the device is set, the EEG data are collected at a sampling rate of 1000 Hz from six channels, and the frequency attribute values are extracted from the raw. The extracted frequency attribute values are categorized into two major values: (1) the power value indicating the extent to which each frequency component has appeared and (2) the relative value indicating the relative proportion of each component in the frequency domain for the entire region. Basic EEG biological tests use power values, which may give different amplitude sizes of EEG waveforms due to individual differences in scalp and skull thickness as well as the contact with the electrode and skin. Therefore, for experiments with multiple subjects, it is desirable to use the relative value of each region for analysis due to the large individual deviation of the power value. The relative values of the frequency domain are extracted for comparison with the results of the experiments using raw data. The frequency attribute value extraction extracts values such as alpha, beta, etc. of the power value by applying FFT to raw EEG data. Additional functions were performed to disaggregate the frequency range based on the extracted values, or a ratio of each value was applied to extract a total of 66 attribute values. In addition, based on the power value, we further extract the relative value representing the relative ratio and design it for use in experiments.
The lines on the graph represent the raw values collected from the six channels shown in Figure 2, while the vertical axis represents frequency in Hz and the horizontal axis represents time. For time, periods of at least five seconds were used to differentiate the walking data. Overall, using only raw EEG data from stroke patients and normal controls to distinguish and determine stroke disease can be difficult for medical professionals. In addition, medical professionals need time to make a proper diagnosis of stroke. In this study, we conduct experiments and validations with the goal of providing meaningful information and ultimately designing and implementing systems that can help medical staff make judgments more quickly, rather than simply attempt to develop a system that can predict stroke early. Figure 3 shows examples of the raw EEG data in (a) stroke patients and (b) normal controls.

3.3. Stroke Prediction Module

The stroke prediction module for the elderly using deep learning-based real-time EEG data proposed in this paper consists of two units, as illustrated in Figure 4. For the offline processing unit, the EEG data are extracted from a database storing the data on various biological signals such as EEG, ECG, and EMG, and a series of processes are followed to learn and tune the parameters based on deep learning. The online processing unit analyzes brainwave data collected in real time based on features learned from the offline processing unit and delivers information about stroke prediction to medical staff in real time. This information can then be communicated to help medical staff predict stroke faster and make accurate diagnoses.
EEG data are time series data that consist of sequential values over time, and it is necessary to consider time information in the learning process. Therefore, in our stroke prediction module we transform and optimize the structure of a deep learning model to obtain a model that can address the characteristics of time series data well and use it in experiments. The deep learning models that were analyzed were CNN [27,28], LSTM [49], bidirectional LSTM [44], etc. CNNs can reduce computation volume through the use of basically shared parameters, and they have the advantage of mitigating overfitting. Moreover, CNNs are increasingly being used to research and develop models by extracting the optimal properties to improve classification and prediction performance. The LSTM model can predict the next stage of time series data by considering both future and historical information of time series data, which can solve the long-term dependency problem. Finally, Bidirectional LSTM is a structure that can overcome the structural limitations of LSTM, where the results of the latter tend to converge based on the last-minute pattern by adding a reverse processing layer to the existing LSTM layer. In this work, we selected four models to fit the time series data classification using the advantages of each model, which are illustrated in Figure 5.

3.3.1. Long-Short Term Memory (LSTM)

The existing RNN is a kind of ANN wherein information persists inside the neural network, and it has been used for image caption generation, automatic translation, etc. [50]. However, RNN has the disadvantage of its learning ability gradually decreasing due to the reduction in its slope when the distance between information grows, i.e., when the length of the input sequence increases. To address these problems, we use the LSTM [49] model, which adds a cell-state to the hidden-state of the RNN. LSTM consists of a forget gate, an input gate, and an output gate. The forget gate f t is a gate for forgetting past information inside a neural network, which preserves the information entirely if the formula result is 1 and discards the information if it is 0.
f t = σ ( W f · [ h t ,   x t ] + b f )
The input gate i t is a gate for remembering the current information inside a neural network, which determines which new information is stored in the cell state. First, Equation (2) determines which value to update, then prepares to add the new candidate values, the C ˜ t vector, to the cell state through Equation (3). The state can be said to be prepared by combining the two sets of information from Equations (2) and (3).
i t = σ ( W i · [ h t 1 ,   x t ] + b i )
C ˜ t = tan h ( W C · [ h t 1 ,   x t ] + b C )
It then updates the past state, C t 1 , to create a new cell state, C t . As shown in Equation (4), the previous state is multiplied by f t , while forgetting what one has chosen to forget at the very first step and adding i t C ˜ t . i t C ˜ t is a value that has been increased or decreased by how much the value was supposed to be updated.
C t = f t C t 1 + i t C ˜ t
Finally, the output gate o t determines how much of the information is represented inside the neural network, where output is filtered based on the cell state. The cell state is input into the tanh layer to receive a value between −1 and 1. This value is multiplied by the output of the previously calculated sigmoid gate, which allows only the desired part to be expressed as output. This process is as shown in Equations (5) and (6).
o t = σ ( W o [ h t 1 ,   x t ] + b o
h t = o t tan h ( C t )

3.3.2. Bidirectional LSTM

Bidirectional RNN (BRNN) is a model used when time series data exhibit significant results in forward inference from the past to the future, as well as reverse inference from future to past [51], i.e., it is used to predict labels for current data through past sequences and future sequences. For this model, there are hidden layers with information on the forward states and hidden layers with information on the backwards states, and these two layers are not interconnected. However, the input value is passed to both hidden layers, and the output layer also receives the value from the two hidden layers to calculate the final output. This is equivalent to Equations (7)–(9), which calculate the activation output h t of the forward hidden layer, the activation output h t of the background hidden layer, and the output y t of the output layer at time t.
h t = σ ( W x h x t + W h h h t 1 + b h
h t , = σ ( W x h x t + W h h h t + 1 + b h
y t = W h y h t + W h y h t + b y
We use bidirectional LSTM (BiLSTM) [44] for the experiments in this paper by applying the LSTM network instead of RNN in bidirectional RNN.

3.3.3. CNN–LSTM

CNN [27,28] is capable of implementing complex nonlinear models, and it is specialized in image and speech recognition. While LSTM models perform well on sophisticated time series data that follow a particular trend, for data that do not exhibit a particular trend and that show severe changes, the prediction value converges upon a specific constant value, thus resulting in poor accuracy. To solve this problem, we combine CNNs that have the advantage of extracting the characteristics of time series data and LSTM models that predict the time series data for the next step by considering both historical and future information regarding time series data [52]. The time series data entered can be more accurately classified and predicted because it extracts the properties of the data through the CNN layer, then passes through the LSTM layer that exploits past and future information.

3.3.4. CNN-Bidirectional LSTM

As mentioned earlier, we combine CNN models at the front of the LSTM model to compensate for the shortcomings of LSTM [49]. We also combine a bidirectional LSTM model with a CNN that can simultaneously make reverse predictions from the future to the past direction and with the underlying forward LSTM model that makes predictions from the past to the future [45].

4. Experiments and Result Analysis

4.1. Data Collection and Description

As explained in Section 3.1, from the various biological signals collected, the experiments were conducted using only EEG data. The measurement and collection of EEG data were conducted at the emergency medical center of Chungnam National University Hospital for senior citizens aged 65 or older from 2017 to 2018. The selected subjects were patients who had received rehabilitation treatment for stroke and who had been diagnosed with stroke within the first month. The experiment involved 48 stroke patients and 75 normal patients from 2017, as well as 13 stroke patients and 137 normal patients from 2018. To achieve an equal comparison, 61 stroke patients and 61 randomly selected control data were ultimately selected for analysis. Five specific daily activity protocols were implemented, including walking, chair sitting and standing, standing, moving objects, and sleeping. After being equipped with the vital-signal collection sensors, all subjects were trained once before the measurement protocol. The first and last of the five measurement protocols were excluded from the experimental data, as these data were likely to reflect the subject’s tension, discomfort, and fatigue. Of the total 273 subjects, 227 cases were evaluated by the medical staff as the NIHSS value for evaluating the severity of stroke. There were 117 males, with a mean age of 74.44, a standard deviation of 6.775, a maximum age of 90, and a minimum age of 65 years. There were 110 females, with an average age of 77.82, a standard deviation of 6.661, a maximum age of 99 years, and a minimum age of 65 years, just like men. Including both males and females, there were 13 elderly patients with NIHSS scores of 0, 153 patients with a score from 1 to 4, 42 patients with a score from 5 to 15, and 8 patients with a score from 21 to 42. In this paper, the patients with an NIHSS score of 5 or higher were specified as stroke patients, and this is a criterion defined by prior references that was agreed upon by the medical staff at Chungnam National University Hospital.
The raw data values acquired from the six channels were extracted using FFT techniques, as described in Section 3.2. After redefining the value of the extracted attributes, a total of 66 power values and relative values were used, as listed in Table 1.

4.2. Performance Evaluation and Indicators for Experiments

All experiments were performed in the same experimental environment: OS, Ubuntu 18.04.4 LTS; CPU, Intel Core i7-9800X; GPU, 2 X NVIDIA Quadro RTX 8000; and RAM, 256 MB. In this paper, we utilize statistical metrics used for disease screening to evaluate the performance of our system. The performance evaluation indicators used are defined below.
  • ① Sensitivity: Percentage of stroke patients who have tested positive.
  • ② Specificity: Percentage of non-stroke patients who have tested negative.
  • ③ False Positive Rate: Percentage of non-stroke patients who have tested positive.
  • ④ False Negative Rate: Percentage of stroke patients who have tested negative.
  • ⑤ Accuracy: Percentage of stroke patients determined as positive and non-patients as negative.
  • ⑥ Precision: Percentage of people who are actually stroke patients among those who have tested positive.
  • ⑦ Recall: Percentage of stroke patients who have previously tested positive.
  • ⑧ F1-Score (Harmonic Mean of Precision and Recall): Percentage of stroke patients who have previously tested positive.
To determine the performance of the classifiers, it is important to have a high accuracy of the model, but the sensitivity and the specificity must also be high, while the false positive and false negative ratios should be low (See Table 2 below). If a non-suitable diagnostic method is used, the sensitivity and the specificity tend to decrease when one side increases, so both should have a high value to achieve a good diagnostic method. When the false positive ratio is actually negative and the result is positive, this can lead to wrong data classification and additional cost. When the false negative ratio is actually positive and the result is negative, this can have a critical impact on the patient’s life, which is used as an important performance evaluation indicator when using medical data.
S e n s i t i v i t y   ( = R e c a l l ) = T P T P + F N
S p e c i f i c i t y = T N F P + T N
F a l s e   P o s i t i v e   R a t e   ( F P R ) = 1 S p e c i f i c i t y
F a l s e   N e g a t i v e   R a t e   ( F N R ) = 1 S e n s i t i v i t y
A c c u r a c y = T P + T N T P + F N + F P + T N
P r e c i s i o n = T P T P + F P
F 1 S c o r e = 2 × P r e c i s i o n × R e c a l l P r e c i s i o n + R e c a l l

4.3. Experimental Results and Analysis

The deep learning-based EEG data stroke classification experiments in this study used LSTM, bidirectional LSTM, CNN-LSTM, and CNN-bidirectional LSTM models, as described in Section 3, and the experiments were conducted by entering three types of data (raw values, power values, and relative values) into each model. Every experiment was performed ten times, and the mean value was shown as the final result.
Table 3, Table 4, Table 5, Table 6, Table 7 and Table 8 list the results of the stroke prediction experiments performed with each model based on the type of data. The experimental results are presented in terms of accuracy, F1-Score, precision, sensitivity, specificity, FNR, and FPR using the performance evaluation metrics described in Section 4.2. Of the total data, 80% were used for learning in the experiments, and the remaining 20% were used for prediction and validation. In this experiment, a data set was constructed according to five-fold validation. We used the average value of the prediction results as a performance index to discover a more generalizable stroke disease prediction model.
The ROC (receiver operating characteristic) curve of the CNN-bidirectional LSTM models in Table 4 is presented in Figure 6. The ROC curve is an index expressing the threshold and performance of binary classification prediction of stroke disease, where the x-axis indicates specificity and the y-axis indicates sensitivity.
When we conducted predictive experiments applying each deep learning model using raw values, the experimental results obtained using the CNN-bidirectional LSTM models were the highest at 94.0%, as presented in Table 3. The low false negative and false positive ratios mean that the rates of classifying stroke patients as normal and normal patients as stroke patients are both very low. CNN-bidirectional LSTM showed satisfactory experimental results for not only accuracy but also performance evaluations such as precision and F1-Score.
Table 5 and Table 6 list the results of the experiments conducted by applying each deep learning model with power values. In this experiment, we show that the CNN-bidirectional LSTM model performed best with a stroke disease prediction accuracy of 81.4%. However, the overall performance was significantly lower than that obtained using raw values. In particular, the FPR (18.5%) and FNR (17.3%) are both high, which are too low to have any clinical meaning and for medical staff to use as a stroke disease judgment.
Finally, after conducting experiments by applying each deep learning model using relative values, we experimentally confirmed that the Bidirectional LSTM model has the highest predictive accuracy at 89.2%. The performance of the bidirectional LSTM was slightly higher when using the relative value than when using the power value, but the overall performance was lower than the experimental results obtained using the raw value. To summarize, CNN-bidirectional LSTM models provide the best performance when raw data values are used. Altogether, the results show that the FPR and FNR alone can predict stroke with high performance. This means that the time needed to detect stroke disease can be shortened without needing to go through the frequency attribute extraction process. For the three data types, the hyper-parameters for each model are listed in Table 9 below.

5. Conclusions

In this paper, we propose a system that enables the early detection and prediction of stroke disease based on deep learning using EEG raw data, power values, and relative values. The analysis of the experimental results of raw data, power values, and relative values showed that using raw data achieved the highest stroke prediction accuracy. In addition, we confirmed that the specificity and sensitivity of the experimental results using raw data are lower than those of other data and models, with very low FPR and FNR. This is an important experimental result indicating that the early detection and onset of stroke disease can be accurately predicted using raw EEG data alone, without requiring the cumbersome process of separating frequency domain properties. The system proposed in this paper can provide useful analytical information for medical staff, patients with stroke with high recurrence potential, or elderly people with high incidence of stroke. It is a significant finding that stroke can be predicted at low cost during daily activities such as walking situations. This study is meaningful, as it can detect the risk of stroke early, before an individual is taken to the emergency room, thus allowing for access to treatment within the golden period. However, to improve the predictive accuracy and performance of real-time predictive models of stroke disease, analysis and predictive models should be studied by integrating health examination data and CT analysis information in a clinical setting.
In future research, we will study accurate predictions and in-depth interpretations of stroke disease using various biological signals such as EMG, ECG, and gait motion, as well as other EEG data. Further, we intend to research and develop systems that provide more reliable and clinically interpretable stroke prediction results by conducting multimodal studies that combine electronic medical recording (EMR) data, such as individual health checkups, with CT or MRI scans and interpretating the information.

Author Contributions

Conceptualization, Y.-A.C. and J.-H.Y.; methodology, Y.-A.C., S.-J.P., J.-A.J., C.-S.P., H.-S.L. and J.-H.Y.; software, Y.-A.C. and J.-H.Y.; validation, Y.-A.C., S.-J.P., J.-A.J., C.-S.P., K.-H.C., H.-S.L. and J.-H.Y.; formal analysis, Y.-A.C., H.-S.L. and J.-H.Y.; investigation, Y.-A.C., S.-J.P., J.-A.J., C.-S.P., K.-H.C., H.-S.L. and J.-H.Y.; resources, Y.-A.C. and J.-H.Y.; data curation, Y.-A.C., K.-H.C. and J.-H.Y.; writing—original draft preparation, Y.-A.C., S.-J.P., J.-A.J., C.-S.P., K.-H.C., H.-S.L. and J.-H.Y.; writing—review and editing, Y.-A.C., H.-S.L. and J.-H.Y.; visualization, Y.-A.C. and J.-H.Y.; supervision, J.-H.Y. and C.-S.P.; project administration, C.-S.P.; funding acquisition, C.-S.P. All authors have read and agreed to the published version of the manuscript.

Funding

This work was supported by a National Research Council of Science & Technology (NST) grant by the Korean Government (MSIP) (No. CRC-15-05-ETRI).

Institutional Review Board Statement

The study was conducted according to the guidelines of the Declaration of Helsinki and approved by the Ethics Committee of Chungnam National University (No.2016-05-022, approved in 5 July 2016).

Informed Consent Statement

Informed consent was obtained from all subjects involved in the study.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. Mendelow, A.D. Stroke: Pathophysiology, diagnosis, and management. Elsevier Health Sci. 2000, 56, 275–286. [Google Scholar]
  2. Global Health Estimates. Geneva: World Health Organization. 2012. Available online: http://www.who.int/healthinfo/global_burden_disease/en (accessed on 1 June 2016).
  3. Feigin, V.L.; Forouzanfar, M.H.; Krishnamurthi, R.; Mensah, G.A.; Connor, M.; Bennett, D.A.; Murray, C. Global and re-gional burden of stroke during 1990–2010: Findings from the Global Burden of Disease Study 2010. Lancet 2014, 383, 245–255. [Google Scholar] [CrossRef]
  4. Roth, G.A.; Mensah, G.A.; Johnson, C.O.; Addolorato, G.; Ammirati, E.; Baddour, L.M.; Barengo, N.C.; Beaton, A.Z.; Benjamin, E.J.; Benziger, C.P.; et al. Global Burden of Cardiovascular Diseases and Risk Factors, 1990–2019: Update from the GBD 2019 Study. J. Am. Coll. Cardiol. 2020, 76, 2982–3021. [Google Scholar] [CrossRef] [PubMed]
  5. Hier, D.B.; Foulkes, M.A.; Swiontoniowski, M.; Sacco, R.L.; Gorelick, P.B.; Mohr, J.P.; Price, T.R.; Wolf, P.A. Stroke recurrence within 2 years after ischemic infarction. Stroke 1991, 22, 155–161. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  6. Kummer, B.R.; Klang, E.; Stein, L.K.; Dhamoon, M.S.; Jetté, N. History of Stroke Is Independently Associated with In-Hospital Death in Patients With COVID-19. Stroke 2020, 51, 3112–3114. [Google Scholar] [CrossRef]
  7. Zhang, L.; Sun, W.; Wang, Y.; Wang, X.; Liu, Y.; Zhao, S.; Long, D.; Chen, L.; Yu, L. Clinical Course and Mortality of Stroke Patients with Coronavirus Disease 2019 in Wuhan, China. Stroke 2020, 51, 2674–2682. [Google Scholar] [CrossRef]
  8. Lee, K. Stroke Diagnosis. J. Korean Med. Assoc. 2002, 45, 1432–1439. [Google Scholar] [CrossRef]
  9. Lee, M.; Ryu, J.; Kim, D. Automated epileptic seizure waveform detection method based on the feature of the mean slope of wavelet coefficient counts using a hidden Markov model and EEG signals. ETRI J. 2020, 42, 217–229. [Google Scholar] [CrossRef]
  10. Acharya, U.R.; Oh, S.L.; Hagiwara, Y.; Tan, J.H.; Adeli, H.; Subha, D.P. Automated EEG-based screening of depression using deep convolutional neural network. Comput. Methods Programs Biomed. 2018, 161, 103–113. [Google Scholar] [CrossRef] [PubMed]
  11. Kim, D.; Kim, L. Detection of Early Stage Alzheimer’s Disease Using EEG Relative Power with Deep Neural Network. In Proceedings of the 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Honolulu, HI, USA, 18–21 July 2018; pp. 352–355. [Google Scholar]
  12. Chambon, S.; Thorey, V.; Arnal, P.; Mignot, E.; Gramfort, A. DOSED: A deep learning approach to detect multiple sleep micro-events in EEG signal. J. Neurosci. Methods 2019, 321, 64–78. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  13. Kwon, Y.H.; Shin, S.B.; Kim, S.D. Electroencephalography based fusion two-dimensional (2D)-convolution neural net-works (CNN) model for emotion recognition system. Sensors 2018, 18, 1383. [Google Scholar] [CrossRef] [Green Version]
  14. O’Shea, A.; Lightbody, G.; Boylan, G.; Temko, A. Neonatal seizure detection from raw multi-channel EEG using a fully convolutional architecture. Neural Netw. 2020, 123, 12–25. [Google Scholar] [CrossRef] [PubMed]
  15. Özdamar, Ö.; Kalayci, T. Detection of Spikes with Artificial Neural Networks Using Raw EEG. Comput. Biomed. Res. 1998, 31, 122–142. [Google Scholar] [CrossRef] [PubMed]
  16. Wijaya, S.K.; Badri, C.; Misbach, J.; Soemardi, T.P.; Sutanno, V. Electroencephalography (EEG) for detecting acute ischemic stroke. In Proceedings of the 2015 4th International Conference on Instrumentation, Communications, Information Technology, and Biomedical Engineering (ICICI-BME), Bandung, Indonesia, 2–3 November 2015; 2015; pp. 42–48. [Google Scholar]
  17. Rahma, O.N.; Wijaya, S.K.; Badri, C. Electroencephalogram analysis with extreme learning machine as a supporting tool for classifying acute ischemic stroke severity. In Proceedings of the 2017 International Seminar on Sensors, Instrumentation, Measurement and Metrology (ISSIMM), Surabaya, Indonesia, 25–26 August 2017; pp. 180–186. [Google Scholar]
  18. Li, X.; Bian, D.; Yu, J.; Li, M.; Zhao, D. Using machine learning models to improve stroke risk level classification methods of China national stroke screening. BMC Med. Inform. Decis. Mak. 2019, 19, 261. [Google Scholar] [CrossRef] [PubMed]
  19. Adhi, H.A.; Wijaya, S.K.; Badri, C.; Rezal, M. Automatic detection of ischemic stroke based on scaling exponent electro-encephalogram using extreme learning machine. J. Phys. Conf. Ser. 2017, 820, 12005–12013. [Google Scholar] [CrossRef]
  20. Djamal, E.C.; Ramadhan, R.I.; Mandasari, M.I.; Djajasasmita, D. Identification of post-stroke EEG signal using wavelet and convolutional neural networks. Bull. Electr. Eng. Inform. 2020, 9, 1890–1898. [Google Scholar] [CrossRef]
  21. Schneider, A.L.; Jordan, K.G. Regional Attenuation WithOut Delta (RAWOD): A distinctive EEG pattern that can aid in the diagnosis and management of severe acute ischemic stroke. Am. J. Electroneurodiagn. Technol. 2005, 45, 102–117. [Google Scholar] [CrossRef]
  22. Finnigan, S.; Wong, A.; Read, S. Defining abnormal slow EEG activity in acute ischaemic stroke: Delta/alpha ratio as an op-timal QEEG index. Clin. Neurophysiol. 2016, 127, 1452–1459. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  23. Varelas, P.N.; Hacein-Bey, L. Ischemic Stroke, Hyperperfusion Syndrome, Cerebral Sinus Thrombosis, and Critical Care Seizures. In Seizures in Critical Care; Springer Science and Business Media LLC, 2017; Volume 14, pp. 155–186. [Google Scholar] [CrossRef]
  24. Ip, Z.; Rabiller, G.; He, J.W.; Yao, Z.; Akamatsu, Y.; Nishijima, Y.; Liu, J.; Yazdan-Shahmorad, A. Cortical stroke affects activity and stability of theta/delta states in remote hippocampal regions*. In Proceedings of the 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Berlin, Germany, 23–27 July 2019; 2019; pp. 5225–5228. [Google Scholar]
  25. Kim, J.; Lee, B.; Lee, H.S.; Shin, K.H.; Kim, M.J.; Son, E. Differences in Brain Waves of Normal Persons and Stroke Patients during Action Observation and Motor Imagery. J. Phys. Ther. Sci. 2014, 26, 215–218. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  26. Gautam, A.; Raman, B. Towards effective classification of brain hemorrhagic and ischemic stroke using CNN. Biomed. Signal. Process. Control 2021, 63, 102178. [Google Scholar] [CrossRef]
  27. Kalchbrenner, N.; Grefenstette, E.; Blunsom, P. A Convolutional Neural Network for Modelling Sentences. In Proceedings of the 52nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), Baltimore, MD, USA, 23–25 June 2014; Volume 1, pp. 655–665. [Google Scholar]
  28. Krizhevsky, A.; Sutskever, I.; Hinton, G.E. Imagenet classification with deep convolutional neural net-works. Commun. ACM 2017, 60, 84–90. [Google Scholar] [CrossRef]
  29. He, K.; Zhang, X.; Ren, S.; Sun, J. Deep residual learning for image recognition. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, 27–30 June 2016; pp. 770–778. [Google Scholar]
  30. Chin, C.-L.; Lin, B.-J.; Wu, G.-R.; Weng, T.-C.; Yang, C.-S.; Su, R.-C.; Pan, Y.-J. An automated early ischemic stroke detection system using CNN deep learning algorithm. In Proceedings of the 2017 IEEE 8th International Conference on Awareness Science and Technology (iCAST), Taichung, Taiwan, 8–10 November 2017; 2017; pp. 368–372. [Google Scholar]
  31. Liu, L.; Chen, S.; Zhang, F.; Wu, F.-X.; Pan, Y.; Wang, J. Deep convolutional neural network for automatically segmenting acute ischemic stroke lesion in multi-modality MRI. Neural Comput. Appl. 2019, 32, 6545–6558. [Google Scholar] [CrossRef]
  32. Dourado, C.M., Jr.; da Silva, S.P.P.; da Nobrega, R.V.M.; Barros, A.C.D.S.; Reboucas Filho, P.P.; de Albuquerque, V.H.C. Deep learning IoT system for online stroke detection in skull computed tomography images. Comput. Netw. 2019, 152, 25–39. [Google Scholar] [CrossRef]
  33. Kabir, M.F.; Hossain, A.; Dahal, K. Enhanced classification accuracy on naive Bayes data mining models. Int. J. Comput. Appl. 2011, 28, 9–16. [Google Scholar] [CrossRef]
  34. Pal, S.K.; Mitra, S. Multilayer Perceptron, Fuzzy Sets, and Classification. IEEE Trans. Neural Netw. 1992, 3, 683–697. [Google Scholar] [CrossRef] [PubMed]
  35. Peterson, L.E. K-nearest neighbor. Scholarpedia 2009, 4, 1883. [Google Scholar] [CrossRef]
  36. Pal, M. Random forest classifier for remote sensing classification. Int. J. Remote Sens. 2005, 26, 217–222. [Google Scholar] [CrossRef]
  37. St, J.P.O.E.D.; Dumais, S.; Osuna, E.; Platt, J.; Scholkopf, B. Support vector machines. IEEE Intell. Syst. 1998, 13, 18–28. [Google Scholar] [CrossRef] [Green Version]
  38. Choi, Y.-A.; Park, S.; Jun, J.-A.; Ho, C.; Pyo, C.-S.; Lee, H.; Yu, J. Machine-Learning-Based Elderly Stroke Monitoring System Using Electroencephalography Vital Signals. Appl. Sci. 2021, 11, 1761. [Google Scholar] [CrossRef]
  39. Yu, J.; Park, S.; Kwon, S.-H.; Ho, C.M.B.; Pyo, C.-S.; Lee, H. AI-based Stroke Disease Prediction System Using Real-Time Electromyography Signals. Appl. Sci. 2020, 10, 6791. [Google Scholar] [CrossRef]
  40. Xie, Y.; Yang, H.; Yuan, X.; He, Q.; Zhang, R.; Zhu, Q.; Chu, Z.; Yang, C.; Qin, P.; Yan, C. Stroke prediction from electrocardiograms by deep neural network. Multimed. Tools Appl. 2021, 80, 17291–17297. [Google Scholar] [CrossRef]
  41. Fawaz, S.; Sim, K.S.; Tan, S.C. Encoding Rich Frequencies for Classification of Stroke Patients EEG Signals. IEEE Access 2020, 8, 135811–135820. [Google Scholar] [CrossRef]
  42. Mirza, M.; Osindero, S. Conditional generative adversarial nets. arXiv 2014, arXiv:1411.1784. 2014. [Google Scholar]
  43. Bebis, G.; Georgiopoulos, M. Feed-forward neural networks. IEEE Potentials 1994, 13, 27–31. [Google Scholar] [CrossRef]
  44. Zhang, S.; Zheng, D.; Hu, X.; Yang, M. Bidirectional long short-term memory networks for relation classification. In Proceedings of the 29th Pacific Asia Conference on Language, Information and Computation, Shanghai, China, 30 October–1 November 2015; pp. 73–78. [Google Scholar]
  45. Jeong, J.H.; Shim, K.H.; Kim, D.J.; Lee, S.W. Brain-controlled robotic arm system based on multi-directional CNN-BiLSTM network using EEG signals. IEEE Trans. Neural Syst. Rehabil. Eng. 2020, 28, 1226–1238. [Google Scholar] [CrossRef] [PubMed]
  46. Guntari, E.W.; Djamal, E.C.; Nugraha, F.; Liem, S.L.L. Classification of Post-Stroke EEG Signal Using Genetic Algorithm and Recurrent Neural Networks. In Proceedings of the 2020 7th International Conference on Electrical Engineering, Computer Sciences and Informatics (EECSI), Yogyakarta, Indonesia, 1–2 October 2020; IEEE: Piscataway, NJ, USA, 2020; pp. 156–161. [Google Scholar]
  47. Zhang, Q.; Benveniste, A. Wavelet networks. IEEE Trans. Neural Netw. 1992, 3, 889–898. [Google Scholar] [CrossRef] [PubMed]
  48. Giri, E.P.; Fanany, M.I.; Arymurthy, A.M.; Wijaya, S.K. Ischemic stroke identification based on EEG and EOG using ID convolutional neural network and batch normalization. In Proceedings of the 2016 International Conference on Advanced Computer Science and Information Systems (ICACSIS), Malang, Indonesia, 15–16 October 2016; 2016; pp. 484–491. [Google Scholar]
  49. Hochreiter, S.; Schmidhuber, J. Long short-term memory. Neural Comput. 1997, 9, 1735–1780. [Google Scholar] [CrossRef] [PubMed]
  50. Zaremba, W.; Sutskever, I.; Vinyals, O. Recurrent Neural Network Regularization. arXiv 2014, arXiv:1409.2329. [Google Scholar]
  51. Schuster, M.; Paliwal, K. Bidirectional recurrent neural networks. IEEE Trans. Signal. Process. 1997, 45, 2673–2681. [Google Scholar] [CrossRef] [Green Version]
  52. Wang, J.; Yu, L.-C.; Lai, K.R.; Zhang, X. Dimensional Sentiment Analysis Using a Regional CNN-LSTM Model. In Proceedings of the 54th Annual Meeting of the Association for Computational Linguistics (Volume 2: Short Papers), Berlin, Germany, 7–12 August 2016; 2016; Volume 2, pp. 225–230. [Google Scholar]
Figure 1. Elderly stroke monitoring system based on deep learning using bio signals (* MVCU: multi vital-signals collector units).
Figure 1. Elderly stroke monitoring system based on deep learning using bio signals (* MVCU: multi vital-signals collector units).
Sensors 21 04269 g001
Figure 2. Six-channel measurement and collection locations of EEG vital-signals.
Figure 2. Six-channel measurement and collection locations of EEG vital-signals.
Sensors 21 04269 g002
Figure 3. Raw EEG signal samples: (a) Raw EEG signals from elderly stroke patients; (b) Raw EEG signal samples from control group.
Figure 3. Raw EEG signal samples: (a) Raw EEG signals from elderly stroke patients; (b) Raw EEG signal samples from control group.
Sensors 21 04269 g003aSensors 21 04269 g003b
Figure 4. Stroke prediction module structure based on deep learning.
Figure 4. Stroke prediction module structure based on deep learning.
Sensors 21 04269 g004
Figure 5. The architecture of the four deep learning models used in the experiment: (a) LSTM; (b) Bidirectional LSTM; (c) CNN–LSTM; (d) CNN-Bidirectional LSTM.
Figure 5. The architecture of the four deep learning models used in the experiment: (a) LSTM; (b) Bidirectional LSTM; (c) CNN–LSTM; (d) CNN-Bidirectional LSTM.
Sensors 21 04269 g005
Figure 6. The ROC curve of the CNN-bidirectional LSTM model using raw EEG bio signals.
Figure 6. The ROC curve of the CNN-bidirectional LSTM model using raw EEG bio signals.
Sensors 21 04269 g006
Table 1. Detailed descriptions of newly defined and extracted EEG attributes.
Table 1. Detailed descriptions of newly defined and extracted EEG attributes.
Frequency BandMeaning and Description
DeltaDelta power (1~4 Hz)
ThetaTheta power (4~8 Hz)
AlphaAlpha power (8~13 Hz)
BetaBeta power (14~30 Hz)
GammaGamma power (30 Hz or more)
Low BetaLow beta power (12~25 Hz)
High BetaHigh beta power (25~30 Hz)
Theta to BetaThe value of the beta ratio in theta (extracting abnormal theta waves)
DARRatio of mean power (delta/alpha)
IDARInverse ratio of DAR (alpha/delta)
PRIPower ratio index (delta+theta to alpha+beta), Low frequency to high frequency
Table 2. Confusion matrix for performance evaluation.
Table 2. Confusion matrix for performance evaluation.
TrueStrokeNormal
Predicted
StrokeTP 1FP2
NormalFN 3TN 4
1 TP (True Positive): Indicator predicting stroke elderly as stroke elderly. 2 FP (False Positive): Indicator predicting stroke elderly as general elderly (normal). 3 FN (False Negative): Indicator predicting general elderly (normal) as stroke elderly. 4 TN (True Negative): Indicator predicting general elderly (normal) as general elderly.
Table 3. DL models’ accuracy, precision, and F1-score based on raw values.
Table 3. DL models’ accuracy, precision, and F1-score based on raw values.
Evaluation
Method
ModelsAccuracyPrecisionF1-Score
Data Sets
Raw dataLSTM70.167.975.4
Bidirectional LSTM91.885.391.7
CNN-LSTM93.796.693.7
CNN-Bidirectional LSTM94.094.694.1
Table 4. DL models’ sensitivity, specificity, FPR, and FNR based on raw values.
Table 4. DL models’ sensitivity, specificity, FPR, and FNR based on raw values.
Evaluation
Method
ModelsSensitivitySpecificityFPR 1FNR 2
Data Sets
Raw dataLSTM90.250.249.89.9
Bidirectional LSTM90.493.56.59.6
CNN-LSTM91.996.13.98.1
CNN-Bidirectional LSTM94.094.36.05.7
1 FPR (False Positive Rate): Indicator of the percentage of general elderly (normal) expected to be stroke elderly. 2 FNP (False Negative Rate): Indicator of the percentage of stroke elderly predicted as general elderly (normal).
Table 5. DL models’ accuracy, precision, and F1-score based on power values.
Table 5. DL models’ accuracy, precision, and F1-score based on power values.
Evaluation
Method
ModelsAccuracyPrecisionF1-Score
Data Sets
Power valueLSTM69.569.568.8
Bidirectional LSTM79.576.480.8
CNN-LSTM74.771.477.3
CNN-Bidirectional LSTM81.480.880.1
Table 6. DL models’ sensitivity, specificity, FPR, and FNR based on power values.
Table 6. DL models’ sensitivity, specificity, FPR, and FNR based on power values.
Evaluation
Method
ModelsSensitivitySpecificityFPRFNR
Data Sets
Power valueLSTM73.864.235.826.2
Bidirectional LSTM88.370.829.211.7
CNN-LSTM86.865.134.913.2
CNN-Bidirectional LSTM82.781.518.517.3
Table 7. DL models’ accuracy, precision, and F1-score based on relative values.
Table 7. DL models’ accuracy, precision, and F1-score based on relative values.
Evaluation
Method
ModelsAccuracyPrecisionF1-Score
Data Sets
Relative valueLSTM81.082.880.7
Bidirectional LSTM89.286.988.8
CNN-LSTM84.082.483.7
CNN-Bidirectional LSTM86.287.385.8
Table 8. DL models’ sensitivity, specificity, FPR, and FNR based on relative values.
Table 8. DL models’ sensitivity, specificity, FPR, and FNR based on relative values.
Evaluation
Method
ModelsSensitivitySpecificityFPRFNR
Data Sets
Relative valueLSTM79.382.517.520.7
Bidirectional LSTM91.687.512.58.4
CNN-LSTM85.287.312.714.8
CNN-Bidirectional LSTM86.083.117.014.0
Table 9. Hyper-parameters of each model.
Table 9. Hyper-parameters of each model.
Data SetsModelsLearning RateBatch SizeEpochOptimizer
RawLSTM0.016450Adam
Bidirectional LSTM0.001128100
CNN- LSTM0.0164200
CNN-Bidirectional LSTM0.00164500
PowerLSTM0.000164300
Bidirectional LSTM0.00132300
CNN- LSTM0.01128500
CNN-Bidirectional LSTM0.00164500
RelativeLSTM0.0001128500
Bidirectional LSTM0.00132300
CNN- LSTM0.00164300
CNN-Bidirectional LSTM0.0164300
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Choi, Y.-A.; Park, S.-J.; Jun, J.-A.; Pyo, C.-S.; Cho, K.-H.; Lee, H.-S.; Yu, J.-H. Deep Learning-Based Stroke Disease Prediction System Using Real-Time Bio Signals. Sensors 2021, 21, 4269. https://doi.org/10.3390/s21134269

AMA Style

Choi Y-A, Park S-J, Jun J-A, Pyo C-S, Cho K-H, Lee H-S, Yu J-H. Deep Learning-Based Stroke Disease Prediction System Using Real-Time Bio Signals. Sensors. 2021; 21(13):4269. https://doi.org/10.3390/s21134269

Chicago/Turabian Style

Choi, Yoon-A, Se-Jin Park, Jong-Arm Jun, Cheol-Sig Pyo, Kang-Hee Cho, Han-Sung Lee, and Jae-Hak Yu. 2021. "Deep Learning-Based Stroke Disease Prediction System Using Real-Time Bio Signals" Sensors 21, no. 13: 4269. https://doi.org/10.3390/s21134269

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop