| Literature DB >> 35049650 |
Ghadir Ali Altuwaijri1,2, Ghulam Muhammad1,3.
Abstract
Automatic high-level feature extraction has become a possibility with the advancement of deep learning, and it has been used to optimize efficiency. Recently, classification methods for Convolutional Neural Network (CNN)-based electroencephalography (EEG) motor imagery have been proposed, and have achieved reasonably high classification accuracy. These approaches, however, use the CNN single convolution scale, whereas the best convolution scale varies from subject to subject. This limits the precision of classification. This paper proposes multibranch CNN models to address this issue by effectively extracting the spatial and temporal features from raw EEG data, where the branches correspond to different filter kernel sizes. The proposed method's promising performance is demonstrated by experimental results on two public datasets, the BCI Competition IV 2a dataset and the High Gamma Dataset (HGD). The results of the technique show a 9.61% improvement in the classification accuracy of multibranch EEGNet (MBEEGNet) from the fixed one-branch EEGNet model, and 2.95% from the variable EEGNet model. In addition, the multibranch ShallowConvNet (MBShallowConvNet) improved the accuracy of a single-scale network by 6.84%. The proposed models outperformed other state-of-the-art EEG motor imagery classification methods.Entities:
Keywords: Convolutional Neural Networks (CNN); brain computer interfaces (BCI); deep learning (DL); electroencephalography (EEG); motor imagery (MI)
Mesh:
Year: 2022 PMID: 35049650 PMCID: PMC8773854 DOI: 10.3390/bios12010022
Source DB: PubMed Journal: Biosensors (Basel) ISSN: 2079-6374
Figure 1Typical MI EEG-based signal processing and classification system.
Summary of related work.
| Related Work | Methods | Database | Acc% | Comment |
|---|---|---|---|---|
| Tang et al. [ | 5-layer CNN | Private, with two subjects and two classes | 86.41% ± 0.77 | It is one of the first papers that used a deep learning model to classify EEG-based MI. The method was tested on a private database. |
| Dose et al. [ | Shallow CNN | Physionet EEG Motor Movement/MI Dataset | 2classes 80.38% | As the number of classes increased, the accuracy dropped. |
| Sakhavi et al. [ | FBCSP, C2CM | BCI competition IV-2a dataset | 74.46% (0.659 kappa) | The authors used the DL model as a classifier only after they extracted features using a handcrafted approach. |
| Xu et al. [ | Wavelet transform time-frequency images, two-layer CNN | Dataset III from BCI competition II and dataset 2a from BCI competition IV | 92.75% | This paper also used CNN as a classifier, and extracted the features from a combination of time-frequency images using wavelet transforms. |
| Zhao et al. [ | Multi-branch 3D CNN | BCI competition IV-2a dataset | 75.02% (0.644 kappa) | The 3D filter has more complexity, which makes it difficult to implement in real-time applications. |
| Amin et al. [ | Multi-layer CNN-based fusion models: | BCI competition IV-2a dataset and HGD | 75.7–95.4% | Good accuracy using fixed parameters. |
| M. Riyad et al. [ | Incep-EEGNet | BCI competition IV-2a | 74.07% | They preprocessed the data (resample the signals at 128 Hz, and filter with a bandpass filter between 1 Hz and 32 Hz); also used cropping as data augmentation, and they trained the model with different learning rates in a large number of epochs. |
| T. M. Ingolfsson et al. [ | EEG-TCNET | BCI competition IV-2a | 77.35% | Good paper with good accuracy using fixed and variable parameters. |
| Y. Li et al. [ | CP-MixedNet | BCI competition IV-2a dataset and HGD | 74.6% | It is a good model that has a multiscale in a part of it, but has a large number of parameters (836 K). |
| X. Liu et al. [ | Parallel spatial-temporal self-attention CNN | BCI competition IV-2a dataset and HGD | 78.51% | A good paper that used self-attention in two parts. |
| Y. Li et al. [ | TS-SEFFNet | BCI competition IV-2a dataset and HGD | 74.71% | It is a big model that has a large number of parameters (282 K). |
Figure 2Timing pattern of the BCI Competition IV-2a.
Figure 3Architecture of the EEGNet model.
Figure 4Architecture of the ShallowConvNet model where C = channels; T = time sample; KE = kernel size; F1 = number of filters; and C1,2,3,4 = classes.
Figure 5Architectural of MBEEGNet model (F1 = number of filters in temporal convolution, KE = kernel size in temporal convolution, Pe = dropout probability).
Figure 6Architecture of MBShallowConvNet model.
Global hyper-parameters used for all subjects in MBEEGNet.
| Branch | Hyperparameter | Value |
|---|---|---|
| First branch | Kernel size | 16 |
| Number of temporal filters | 4 | |
| Dropout rate | 0 | |
| Second branch | Kernel size | 32 |
| Number of temporal filters | 8 | |
| Dropout rate | 0.1 | |
| Third branch | Kernel size | 64 |
| Number of temporal filters | 16 | |
| Dropout rate | 0.2 |
Classification accuracy (%) and κ-scores on the MI BCI IV-2a dataset.
| ←Subject | EEGNet [ | EEG-TCNet [ | Incep-EEGNet [ | Variable EEGNet [ | Our Proposed MBEEGNet | ShallowConvNet [ | Our Proposed MBShallow CovNet | |||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Acc. | κ | Acc. | κ | Acc. | κ | Acc. | κ | Acc. | κ | Acc. | κ | Acc. | κ | |
| S1 | 84.34 | 0.79 | 85.77 | 0.81 | 78.47 | 0.71 | 86.48 | 0.82 |
|
| 79.51 | 0.73 |
|
|
| S2 | 54.06 | 0.39 | 65.02 | 0.53 | 52.78 | 0.37 | 61.84 | 0.49 |
|
| 56.25 | 0.42 |
|
|
| S3 | 87.54 | 0.83 | 94.51 | 0.93 | 89.93 | 0.87 | 93.41 | 0.91 |
|
| 88.89 | 0.85 |
|
|
| S4 | 63.59 | 0.51 | 64.91 | 0.53 | 66.67 | 0.56 | 73.25 | 0.64 |
|
| 80.90 | 0.75 |
|
|
| S5 | 67.39 | 0.57 | 75.36 | 0.67 | 61.11 | 0.48 | 76.81 | 0.69 |
|
| 57.29 | 0.43 |
|
|
| S6 | 54.88 | 0.39 | 61.40 | 0.49 | 60.42 | 0.47 | 59.07 | 0.45 |
|
| 53.28 | 0.38 |
|
|
| S7 | 88.80 | 0.85 | 87.36 | 0.83 | 90.63 | 0.88 | 90.25 | 0.87 |
|
|
|
| 88.02 | 0.84 |
| S8 | 76.75 | 0.69 | 83.76 | 0.78 | 82.29 | 0.76 | 87.45 | 0.83 |
|
| 81.25 | 0.75 |
|
|
| S9 | 74.24 | 0.65 | 78.03 | 0.71 |
|
| 82.95 | 0.77 | 83.69 | 0.78 | 79.17 | 0.72 |
|
|
| Mean | 72.40 | 0.63 | 77.35 | 0.70 | 74.07 | 0.65 | 79.06 | 0.72 |
|
| 74.31 | 0.66 |
|
|
| S. D. | 13.27 | 0.18 | 11.57 | 0.15 | 14.06 | 0.19 | 12.28 | 0.16 |
|
| 14.54 | 0.19 |
|
|
In this table, the bold values indicate the best results, Acc. is the accuracy and S. D. is the standard deviation.
Figure 7Average classification accuracy on the MI BCI IV-2a dataset.
Precision, recall, and F1 Score on the MI BCI IV-2a dataset using MBEEGNet.
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | Average | ||
|---|---|---|---|---|---|---|---|---|---|---|---|
|
| LH | 90.36 | 52.84 | 95.52 | 83.00 | 70.86 | 59.54 | 96.61 | 93.81 | 90.45 | 81.44 |
| RH | 96.81 | 55.83 | 100 | 72.00 | 83.75 | 60.18 | 86.00 | 82.16 | 78.00 | 79.41 | |
| F | 87.09 | 77.61 | 90.00 | 80.84 | 70.79 | 78.23 | 97.00 | 87.26 | 75.30 | 82.68 | |
| Tou. | 84.08 | 86.00 | 92.81 | 83.67 | 82.33 | 66.40 | 86.65 | 87.56 | 91.00 | 84.50 | |
| Avg. | 89.58 | 68.07 | 94.58 | 79.88 | 76.93 | 66.09 | 91.57 | 87.70 | 83.69 | 82.01 | |
|
| LH | 92.69 | 61.13 | 95.71 | 77.57 | 91.26 | 56.73 | 83.26 | 92.25 | 85.55 | 81.79 |
| RH | 88.58 | 58.03 | 95.78 | 71.64 | 77.78 | 64.17 | 93.38 | 94.58 | 70.08 | 79.34 | |
| F | 87.88 | 88.74 | 92.59 | 90.81 | 74.27 | 67.71 | 93.54 | 81.46 | 87.62 | 84.96 | |
| Tou. | 89.36 | 66.15 | 94.13 | 80.85 | 68.91 | 77.46 | 98.30 | 83.97 | 93.81 | 83.66 | |
| Avg. | 89.63 | 68.51 | 94.55 | 80.22 | 78.05 | 66.52 | 92.12 | 88.06 | 84.27 | 82.44 | |
|
| Avg. | 89.61 | 68.29 | 94.57 | 80.05 | 77.49 | 66.30 | 91.84 | 87.88 | 83.98 | 82.22 |
In this table, LH: left hand, RH: right hand, F: feet, Tou.: tongue.
Precision, recall, and F1 Score on the MI BCI IV-2a dataset using MBShallowConvNet.
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | AVG. | ||
|---|---|---|---|---|---|---|---|---|---|---|---|
|
| LH | 83.58 | 61.18 | 94.19 | 96.00 | 77.46 | 66.60 | 100 | 93.09 | 96.61 | 85.41 |
| RH | 78.61 | 62.37 | 94.53 | 77.08 | 77.31 | 66.34 | 88.00 | 96.81 | 76.06 | 79.68 | |
| F | 90.18 | 74.30 | 92.54 | 77.92 | 81.92 | 67.73 | 85.91 | 73.15 | 79.76 | 80.39 | |
| Tou. | 78.00 | 82.16 | 93.91 | 79.40 | 74.52 | 58.41 | 78.16 | 84.58 | 82.75 | 79.10 | |
| Avg. | 82.59 | 70.00 | 93.79 | 82.60 | 77.80 | 64.77 | 88.02 | 86.91 | 83.80 | 81.14 | |
|
| LH | 81.55 | 67.93 | 95.53 | 85.64 | 85.27 | 66.73 | 79.62 | 86.27 | 88.66 | 81.91 |
| RH | 89.77 | 61.94 | 98.45 | 81.57 | 81.05 | 66.93 | 92.15 | 89.32 | 73.53 | 81.63 | |
| F | 75.63 | 93.55 | 91.44 | 81.00 | 71.49 | 63.67 | 87.75 | 82.67 | 79.44 | 80.74 | |
| Tou. | 85.90 | 63.32 | 90.12 | 81.70 | 75.20 | 61.70 | 96.53 | 88.82 | 94.86 | 82.02 | |
| Avg. | 83.21 | 71.68 | 93.89 | 82.47 | 78.25 | 64.76 | 89.01 | 86.77 | 84.12 | 81.58 | |
|
| Avg. | 82.90 | 70.83 | 93.84 | 82.54 | 78.03 | 64.76 | 88.51 | 86.84 | 83.96 | 81.36 |
In this table, LH: left hand, RH: right hand, F: feet, Tou.: tongue.
Accuracy for proposed models at different parameter combinations.
| Methods | Hyperparameters | Activation Function | Average Accuracy (%) |
|---|---|---|---|
| MBEEGNet | B1:F1 = 8, KE = 32, Pe = 0.2 | Relu | 77.03 |
| B1:F1 = 8, KE = 32, Pe = 0.2 | elu | 80.30 | |
| B1:F1 = 4, KE = 16, Pe = 0 | Relu | 78.63 | |
| B1:F1 = 4, KE = 16, Pe = 0 | elu | 82.01 | |
| MBShallowConvNet | KE1 = 10, KE2 = 20, KE3 = 30 | - | 80.36 |
| KE1 = 15, KE2 = 25, KE3 = 35 | - | 78.63 | |
| KE1 = 5, KE2 = 15, KE3 = 20 | - | 81.15 |
In this table, B1, B2, B3 mean branch 1, 2, 3, respectively.
Figure 8The confusion matrixes of MBEEGNet on the MI BCI IV-2a dataset.
Figure 9The confusion matrixes of MBShallowConvNet on the MI BCI IV-2a dataset.
Comparison of the number of parameters and mean accuracy.
| Methods | Mean Accuracy (%) | Number of Parameters |
|---|---|---|
| DeepConvNet [ | 71.99 | 284 × 103 |
| EEGNet [ | 72.40 | 2.63 × 103 |
| ShallowConvNet [ | 74.31 | 47.31 × 103 |
| TS-SEFFNet [ | 74.71 | 282 × 103 |
| CP-MixedNet [ | 74.60 | 836 × 103 |
| EEG-TCNet [ | 77.35 | 4.27 × 103 |
| Variable EEGNet [ | 79.06 | 15.6 × 103 |
| Our proposed (MBEEGNet) | 82.01 | 8.908 × 103 |
| Our proposed (MBShallowConvNet) | 81.15 | 147.22 × 103 |
Classification accuracy (%) on the HGD.
|
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | Mean | Std. Dev. |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
|
| 94.37 | 92.50 | 100 | 96.25 | 96.87 | 98.12 | 93.07 | 96.87 | 98.12 | 91.25 | 80.00 | 96.25 | 95.60 | 79.37 | 93.47 | 6.30 |
|
| 95.02 | 95.02 | 100 | 99.40 | 98.17 | 98.80 | 93.13 | 95.52 | 98.18 | 92.14 | 89.43 | 96.02 | 94.45 | 88.88 | 95.30 | 3.50 |
|
| 96.87 | 93.75 | 99.37 | 98.12 | 98.12 | 93.12 | 92.45 | 96.87 | 98.12 | 90.62 | 76.25 | 95.00 | 94.96 | 91.25 | 93.92 | 5.79 |
|
| 98.25 | 96.23 | 98.80 | 98.18 | 97.65 | 96.90 | 93.80 | 97.00 | 97.52 | 92.50 | 80.78 | 96.25 | 95.62 | 92.04 | 95.11 | 4.62 |
|
| 81.88 | 91.88 | 93.13 | 92.50 | 90.63 | 93.13 | 84.28 | 90.80 | 96.88 | 85.00 | 88.13 | 91.25 | 89.94 | 83.75 | 89.51 | 4.32 |
|
| 90.69 | 93.53 | 98.53 | 96.88 | 92.90 | 93.53 | 92.40 | 91.78 | 96.88 | 89.88 | 92.78 | 95.40 | 93.03 | 87.34 | 93.25 | 2.97 |
* Reproduced.
Figure 10Average classification accuracy on the HGD; * means reproduced.
Precision, recall, F1 score, and κ-score on the HGD using MBEEGNet.
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | AVG. | ||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
|
| LH | 92.81 | 95.19 | 100 | 100 | 95.19 | 97.61 | 92.54 | 97.10 | 97.61 | 95.00 | 79.56 | 100 | 92.09 | 84.34 | 94.22 |
| RH | 94.81 | 97.49 | 100 | 100 | 100 | 100 | 90.27 | 85.00 | 95.10 | 93.00 | 88.56 | 100 | 88.09 | 83.08 | 93.96 | |
| F | 97.39 | 94.72 | 100 | 100 | 97.49 | 100 | 95.00 | 100 | 100 | 93.91 | 95.19 | 87.00 | 100 | 88.09 | 96.34 | |
| Tou. | 95.10 | 92.72 | 100 | 97.61 | 100 | 97.61 | 94.71 | 100 | 100 | 86.61 | 94.38 | 97.10 | 97.61 | 100 | 96.67 | |
| Avg. | 95.03 | 95.03 | 100 | 99.40 | 98.17 | 98.80 | 93.13 | 95.52 | 98.18 | 92.13 | 89.42 | 96.02 | 94.45 | 88.88 | 95.30 | |
|
| LH | 97.28 | 97.44 | 100 | 100 | 100 | 100 | 92.72 | 86.61 | 100 | 95.38 | 93.71 | 100 | 90.64 | 81.47 | 95.38 |
| RH | 92.77 | 97.59 | 100 | 100 | 97.66 | 97.66 | 94.74 | 96.70 | 97.54 | 100 | 81.65 | 100 | 91.76 | 84.35 | 95.17 | |
| F | 93.00 | 90.73 | 100 | 97.66 | 97.58 | 97.66 | 94.72 | 100 | 95.33 | 81.95 | 97.04 | 96.77 | 95.42 | 92.15 | 95.00 | |
| Tou. | 97.34 | 94.61 | 100 | 100 | 97.56 | 100 | 90.56 | 100 | 100 | 93.38 | 87.04 | 88.18 | 100 | 97.66 | 96.17 | |
| Avg. | 95.10 | 95.09 | 100 | 99.41 | 98.20 | 98.83 | 93.18 | 95.83 | 98.22 | 92.68 | 89.86 | 96.24 | 94.46 | 88.91 | 95.43 | |
|
| Avg. | 95.06 | 95.06 | 100 | 99.41 | 98.18 | 98.82 | 93.16 | 95.68 | 98.20 | 92.40 | 89.64 | 96.13 | 94.45 | 88.89 | 95.36 |
|
| Avg. | 93.37 | 93.36 | 100 | 99.40 | 97.56 | 98.40 | 90.84 | 94.03 | 97.57 | 89.52 | 85.90 | 94.70 | 92.60 | 85.18 | 93.73 |
Where LH: left hand, RH: right hand, F: feet, Tou.: tongue.
Precision, recall, F1 score, and κ-score on the HGD using MBShallowConvNet.
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | AVG. | ||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
|
| LH | 100 | 97.61 | 100 | 97.61 | 93.09 | 97.39 | 86.00 | 97.29 | 95.19 | 92.81 | 73.15 | 97.39 | 97.39 | 91.91 | 94.06 |
| RH | 93.00 | 97.39 | 100 | 97.49 | 100 | 100 | 94.28 | 90.73 | 97.49 | 92.72 | 65.59 | 97.61 | 92.54 | 83.75 | 93.04 | |
| F | 100 | 94.81 | 100 | 100 | 100 | 95 | 97.49 | 100 | 97.39 | 91.91 | 90.27 | 95.00 | 97.39 | 92.54 | 96.56 | |
| Tou. | 100 | 95.10 | 95.19 | 97.61 | 97.49 | 95.19 | 97.49 | 100 | 100 | 92.54 | 94.00 | 95.00 | 95.19 | 100 | 96.77 | |
| Avg. | 98.25 | 96.23 | 98.80 | 98.18 | 97.65 | 96.90 | 93.81 | 97.00 | 97.52 | 92.50 | 80.75 | 96.25 | 95.63 | 92.05 | 95.11 | |
|
| LH | 97.75 | 100 | 100 | 100 | 100 | 95.10 | 93.78 | 91.25 | 100 | 97.48 | 72.35 | 95.19 | 95.10 | 86.55 | 94.61 |
| RH | 100 | 95.10 | 100 | 97.59 | 97.75 | 97.47 | 87.04 | 97.12 | 97.59 | 95.09 | 75.23 | 100 | 92.45 | 88.79 | 94.37 | |
| F | 100 | 92.68 | 95.42 | 95.33 | 95.42 | 95.19 | 97.49 | 100 | 95.19 | 86.14 | 92.98 | 94.91 | 95.19 | 93 | 94.92 | |
| Tou. | 95.51 | 97.34 | 100 | 100 | 97.68 | 100 | 97.49 | 100 | 97.47 | 91.99 | 82.10 | 95.00 | 100 | 100 | 96.75 | |
| Avg. | 98.32 | 96.28 | 98.85 | 98.23 | 97.71 | 96.94 | 93.95 | 97.09 | 97.56 | 92.68 | 80.66 | 96.27 | 95.68 | 92.09 | 95.17 | |
|
| Avg. | 98.28 | 96.25 | 98.83 | 98.20 | 97.68 | 96.92 | 93.88 | 97.05 | 97.54 | 92.59 | 80.71 | 96.26 | 95.66 | 92.07 | 95.14 |
|
| Avg. | 97.67 | 94.97 | 98.40 | 97.57 | 96.86 | 95.86 | 91.74 | 96.00 | 96.69 | 89.99 | 74.37 | 95.00 | 94.16 | 89.39 | 93.48 |
Where LH: left hand, RH: right hand, F: feet, Tou.: tongue.