| Literature DB >> 32328157 |
Wei Zhao1, Wenbing Zhao2, Wenfeng Wang3, Xiaolu Jiang1, Xiaodong Zhang4, Yonghong Peng5, Baocan Zhang1, Guokai Zhang6.
Abstract
The detection of recorded epileptic seizure activity in electroencephalogram (EEG) segments is crucial for the classification of seizures. Manual recognition is a time-consuming and laborious process that places a heavy burden on neurologists, and hence, the automatic identification of epilepsy has become an important issue. Traditional EEG recognition models largely depend on artificial experience and are of weak generalization ability. To break these limitations, we propose a novel one-dimensional deep neural network for robust detection of seizures, which composes of three convolutional blocks and three fully connected layers. Thereinto, each convolutional block consists of five types of layers: convolutional layer, batch normalization layer, nonlinear activation layer, dropout layer, and max-pooling layer. Model performance is evaluated on the University of Bonn dataset, which achieves the accuracy of 97.63%∼99.52% in the two-class classification problem, 96.73%∼98.06% in the three-class EEG classification problem, and 93.55% in classifying the complicated five-class problem.Entities:
Year: 2020 PMID: 32328157 PMCID: PMC7166278 DOI: 10.1155/2020/9689821
Source DB: PubMed Journal: Comput Math Methods Med ISSN: 1748-670X Impact factor: 2.238
Figure 1Sample EEG signals in this study.
Figure 2The proposed one-dimensional convolutional neural network architecture.
Details of the CNN structure used in this research.
| Block | Type | Number of neurons | Kernel size for each | Stride |
|---|---|---|---|---|
| (Output layer) | Output feature map | |||
| Conv 1 | Convolution | 139 × 20 | 40 | 1 |
| BN | 139 × 20 | — | — | |
| ReLU | 139 × 20 | — | — | |
| Dropout | 139 × 20 | — | — | |
| Max-pooling | 70 × 20 | 2 | 2 | |
|
| ||||
| Conv 2 | Convolution | 51 × 40 | 20 | 1 |
| BN | 51 × 40 | — | — | |
| ReLU | 51 × 40 | — | — | |
| Dropout | 51 × 40 | — | — | |
| Max-pooling | 26 × 40 | 2 | 2 | |
|
| ||||
| Conv 3 | Convolution | 17 × 80 | 10 | 1 |
| BN | 17 × 80 | — | — | |
| ReLU | 17 × 80 | — | — | |
| Dropout | 17 × 80 | — | — | |
| Max-pooling | 9 × 80 | 2 | 2 | |
|
| ||||
| FC 1 | FC | 64 | — | — |
| ReLU | 64 | — | — | |
| Dropout | 64 | — | — | |
|
| ||||
| FC 2 | FC | 32 | — | — |
| ReLU | 32 | — | — | |
| Dropout | 32 | — | — | |
| FC 3 | FC | 2 or 3 or 5 | — | — |
Configurations of 8 models using 10-fold cross-validation for the A vs. B vs. C vs. D vs. E cases.
| Block | Parameter | M1 | M2 | M3 | M4 | M5 | M6 | M7 | M8 |
|---|---|---|---|---|---|---|---|---|---|
| Conv 1 | Number of kernels | 20 | 20 | 20 | 20 | 20 | 20 | 20 | 20 |
| Size of receptive field | 5 | 5 | 5 | 5 | 40 | 40 | 40 | 40 | |
| Dropout rate | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | |
|
| |||||||||
| Conv 2 | Number of kernels | 40 | 40 | 40 | 40 | 40 | 40 | 40 | 40 |
| Size of receptive field | 3 | 3 | 3 | 3 | 20 | 20 | 20 | 20 | |
| Dropout rate | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | |
|
| |||||||||
| Conv 3 | Number of kernels | 80 | 80 | 80 | 80 | 80 | 80 | 80 | 80 |
| Size of receptive field | 3 | 3 | 3 | 3 | 10 | 10 | 10 | 10 | |
| Dropout rate | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | 0.2 | |
|
| |||||||||
| FC 1 | Number of neurons | 32 | 32 | 64 | 64 | 32 | 32 | 64 | 64 |
| Dropout rate | 0.2 | 0.5 | 0.2 | 0.5 | 0.2 | 0.5 | 0.2 | 0.5 | |
|
| |||||||||
| FC 2 | Number of neurons | 16 | 16 | 32 | 32 | 16 | 16 | 32 | 32 |
| Dropout rate | 0.2 | 0.5 | 0.2 | 0.5 | 0.2 | 0.5 | 0.2 | 0.5 | |
|
| |||||||||
| A vs. B vs. C vs. D vs. E | Acc | 90.47 | 88.61 | 91.89 | 91.20 | 93.37 | 91.62 |
| 92.92 |
| Sen | 75.98 | 70.86 | 79.66 | 77.79 | 83.33 | 78.85 | 83.73 | 82.30 | |
| Spe | 94.00 | 92.72 | 94.92 | 94.45 | 95.83 | 94.71 | 95.93 | 95.57 | |
Confusion matrix for the three-class problem (B vs. D vs. E.) across 10-folds.
| Predicted | Acc | Sen | Spe | Pre | F1 | ||||
|---|---|---|---|---|---|---|---|---|---|
| Normal | Preictal | Seizure | |||||||
| Original | Normal | 2263 | 36 | 1 | 98.32 | 98.39 | 98.28 | 96.63 | 97.50 |
| Preictal | 49 | 2220 | 31 | 97.54 | 96.52 | 98.04 | 96.10 | 96.31 | |
| Seizure | 30 | 54 | 2216 | 98.32 | 96.35 | 99.30 | 98.58 | 97.45 | |
| Overall | — | — |
| — | 98.06 | 97.09 | 98.54 | 97.10 | 97.09 |
Accuracies (%) of 10-fold cross-validation using model M7.
| Data sets combination | K1 | K2 | K3 | K4 | K5 | K6 | K7 | K8 | K9 | K10 | Mean |
|---|---|---|---|---|---|---|---|---|---|---|---|
| A vs. E | 100 | 99.57 | 99.57 | 99.35 | 99.35 | 99.57 | 99.13 | 99.57 | 99.35 | 99.78 | 99.52 |
| B vs. E | 99.78 | 99.13 | 99.57 | 98.91 | 99.13 | 99.35 | 98.70 | 98.70 | 98.70 | 99.13 | 99.11 |
| C vs. E | 99.35 | 98.04 | 98.04 | 96.96 | 98.26 | 97.39 | 97.39 | 97.83 | 98.48 | 98.48 | 98.02 |
| D vs. E | 97.61 | 98.04 | 98.26 | 98.04 | 97.17 | 98.04 | 96.52 | 97.17 | 96.52 | 98.91 | 97.63 |
| AB vs. E | 99.57 | 99.13 | 99.57 | 99.57 | 99.57 | 99.13 | 99.57 | 99.13 | 99.57 | 98.99 | 99.38 |
| AC vs. E | 99.28 | 98.70 | 99.13 | 98.84 | 99.13 | 98.70 | 98.99 | 99.57 | 99.42 | 98.55 | 99.03 |
| AD vs. E | 98.12 | 97.83 | 98.41 | 98.70 | 98.41 | 98.41 | 98.55 | 98.55 | 99.13 | 98.84 | 98.50 |
| BC vs. E | 98.70 | 98.41 | 97.68 | 98.55 | 98.55 | 98.99 | 98.84 | 99.28 | 99.57 | 98.26 | 98.68 |
| BD vs. E | 97.39 | 97.10 | 97.54 | 98.84 | 98.26 | 97.54 | 98.41 | 97.97 | 97.83 | 97.39 | 97.83 |
| CD vs. E | 97.68 | 97.54 | 98.41 | 97.83 | 98.41 | 97.25 | 98.84 | 98.41 | 97.97 | 97.97 | 98.03 |
| ABC vs. E | 99.24 | 98.26 | 99.24 | 98.91 | 98.80 | 99.02 | 98.91 | 99.24 | 98.91 | 98.37 | 98.89 |
| ABD vs. E | 98.80 | 98.37 | 98.80 | 98.26 | 98.80 | 99.35 | 98.48 | 97.93 | 98.15 | 98.26 | 98.52 |
| BCD vs. E | 98.26 | 97.61 | 98.59 | 98.26 | 98.59 | 99.24 | 98.04 | 98.70 | 97.93 | 98.37 | 98.36 |
| ABCD vs. E | 98.96 | 99.22 | 98.70 | 98.52 | 98.35 | 99.22 | 98.78 | 98.61 | 99.13 | 98.09 | 98.76 |
| A vs. C vs. E | 96.04 | 97.05 | 97.00 | 97.39 | 94.98 | 97.58 | 97.00 | 96.09 | 96.81 | 97.39 | 96.73 |
| A vs. D vs. E | 97.63 | 97.10 | 97.54 | 95.94 | 97.00 | 96.67 | 97.39 | 97.87 | 96.81 | 96.43 | 97.04 |
| B vs. C vs. E | 97.63 | 97.97 | 98.12 | 97.68 | 98.36 | 97.20 | 97.87 | 99.03 | 97.68 | 97.58 | 97.91 |
| B vs. D vs. E | 98.35 | 98.30 | 98.07 | 97.49 | 98.26 | 97.97 | 97.20 | 98.45 | 98.45 | 98.06 | 98.06 |
| AB vs. CD vs. E | 96.70 | 97.10 | 97.74 | 96.43 | 96.72 | 97.97 | 94.96 | 97.91 | 96.96 | 97.25 | 96.97 |
| A vs. B vs. C vs. D vs. E | 92.99 | 94.37 | 94.00 | 93.41 | 93.36 | 92.73 | 93.74 | 93.25 | 93.74 | 93.91 | 93.55 |
Comparison between the proposed method and other methods using the same dataset.
| Data sets combination | Methodology | Study | Acc (%) | Our Acc (%) |
|---|---|---|---|---|
| A vs. E | TFA + ANN | Tzallas et al. [ | 100 | 99.52 |
| DWT + Kmeans + MLPNN | Orhan et al. [ | 100 | ||
| 1-D-LBP + FT/BN | Kaya et al. [ | 99.50 | ||
| DWT + NB/KNN | Sharmila and Geethanjali [ | 100 | ||
| TQWT + KNNE + SVM | Bhattacharyya et al. [ | 100 | ||
| LMD + GA-SVM | Zhang and Chen [ | 100 | ||
| CNN + M-V | Ullah et al. [ | 100 | ||
| CWT + CNN | Turk and Ozerdem [ | 99.50 | ||
|
| ||||
| B vs. E | DWT + NB/KNN | Sharmila and Geethanjali [ | 99.25 | 99.11 |
| TQWT + KNNE + SVM | Bhattacharyya et al. [ | 100 | ||
| CNN + M-V | Ullah et al. [ | 99.6 | ||
| CWT + CNN | Turk and Ozerdem[ | 99.50 | ||
|
| ||||
| C vs. E | DWT + NB/KNN | Sharmila and Geethanjali [ | 99.62 | 98.02 |
| TQWT + KNNE + SVM | Bhattacharyya et al. [ | 99.50 | ||
| CNN + M-V | Ullah et al. [ | 99.1 | ||
| CWT + CNN | Turk and Ozerdem [ | 98.50 | ||
|
| ||||
| D vs. E | 1-D-LBP + FT/BN | Kaya et al. [ | 95.50 | 97.63 |
| DWT + NB/KNN | Sharmila and Geethanjali [ | 95.62 | ||
| TQWT + KNNE + SVM | Bhattacharyya et al. [ | 98 | ||
| LMD + GA-SVM | Zhang and Chen [ | 98.10 | ||
| CNN + M-V | Ullah et al. [ | 99.4 | ||
| CWT + CNN | Turk and Ozerdem [ | 98.50 | ||
|
| ||||
| AB vs. E | DWT + NB/KNN | Sharmila and Geethanjali [ | 99.16 | 99.38 |
| CNN + M-V | Ullah et al. [ | 99.8 | ||
|
| ||||
| CD vs. E | 1-D-LBP + FT/BN | Kaya et al. [ | 97.00 | 98.03 |
| DWT + NB/KNN | Sharmila and Geethanjali [ | 98.75 | ||
| CNN + M-V | Ullah et al. [ | 99.7 | ||
|
| ||||
| ABCD vs. E | DWT + Kmeans + MLPNN | Orhan et al. [ | 99.60 | 98.76 |
| DWT + NB/KNN | Sharmila and Geethanjali [ | 97.1 | ||
| TQWT + KNNE + SVM | Bhattacharyya et al. [ | 99 | ||
| LMD + GA-SVM | Zhang and Chen [ | 98.87 | ||
| CNN + M-V | Ullah et al. [ | 99.7 | ||
|
| ||||
| B vs. D vs. E | CNN | Acharya et al. [ | 88.7 | 98.06 |
| CWT + CNN | Turk and Ozerdem [ | 98.00 | ||
|
| ||||
| AB vs. CD vs. E | DWT + Kmeans + MLPNN | Orhan et al. [ | 95.60 | 96.97 |
| TQWT + KNNE + SVM | Bhattacharyya et al. [ | 98.60 | ||
| LMD + GA-SVM | Zhang and Chen [ | 98.40 | ||
| CNN + M-V | Ullah et al. [ | 99.1 | ||
|
| ||||
| A vs. B vs. C vs. D vs. E | TFA + ANN | Tzallas et al. [ | 89 | 93.55 |
| MEMD + ANN | Zahra et al. [ | 87.2 | ||
| CWT + CNN | Turk and Ozerdem [ | 93.60 | ||