Classification of motor imagery electroencephalogram signals by using a divergence based convolutional neural network

被引:10
作者
Dokur, Zumray [1 ]
Olmez, Tamer [1 ]
机构
[1] Istanbul Tech Univ, Dept Elect & Commun Engn, Istanbul, Turkey
关键词
EEG; Motor imagery; Classification; Data augmentation; Convolutional neural network; SPATIAL-PATTERNS; EEG;
D O I
10.1016/j.asoc.2021.107881
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep neural networks (DNNs) are observed to be successful in pattern classification. However, the high classification performances of DNNs are related to their large training sets. Unfortunately, in the literature, the datasets used to classify motor imagery (MI) electroencephalogram (EEG) signals contain a small number of samples. To achieve high performances with small-sized datasets, most of the studies have employed a transformation such as the common spatial patterns (CSP) before the classification process. However, the CSP is dependent on subjects and introduces computational load in real time applications. It is observed in the literature that data augmentation is not applied for increasing the classification performance of EEG signals. In this study, we have investigated the effect of the augmentation process on the classification performance of MI EEG signals instead of using a preceding transformation such as the CSP, and we have demonstrated that the augmentation process is able to compete with the CSP by generating high success rates for the classification of MI EEGs. In addition to the augmentation process, we have modified the DNN structure to increase the classification performance, to decrease the number of nodes in the structure, and to use less number of hyper parameters. A minimum distance network following the last layer of the convolutional neural network (CNN) was used as the classifier instead of a fully connected neural network (FCNN). By augmenting the EEG dataset and focusing solely on CNN's training, the training algorithm of the proposed structure is strengthened without applying any transformation. We tested these improvements on brain-computer interface (BCI) competitions 2005 and 2008 databases with two and four classes, and the positive effects of the augmentation on the average accuracies are demonstrated. (C) 2021 Elsevier B.V. All rights reserved.
引用
收藏
页数:20
相关论文
共 57 条
[1]  
Abbas W, 2018, IEEE ENG MED BIO, P219, DOI 10.1109/EMBC.2018.8512271
[2]  
Alansari M, 2018, 2018 6TH INTERNATIONAL CONFERENCE ON BRAIN-COMPUTER INTERFACE (BCI), P138
[3]  
Aljalal M., 2017, 9 IEEE GCC C EXHIBIT, P1, DOI DOI 10.1109/IEEEGCC.2017.8448212
[4]   A Deep Learning Method for Classification of EEG Data Based on Motor Imagery [J].
An, Xiu ;
Kuang, Deping ;
Guo, Xiaojiao ;
Zhao, Yilu ;
He, Lianghua .
INTELLIGENT COMPUTING IN BIOINFORMATICS, 2014, 8590 :203-210
[5]  
Ang K.K., 2008, BEST STUDY CLASSIFIC
[6]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[7]  
Behri M., 2018, 2018 ADV SCI ENG TEC, DOI [DOI 10.1109/ICASET.2018.8376886, 10.1109/icaset.2018.8376886]
[8]   Convolutional Neural Network Based Approach Towards Motor Imagery Tasks EEG Signals Classification [J].
Chaudhary, Shalu ;
Taran, Sachin ;
Bajaj, Varun ;
Sengur, Abdulkadir .
IEEE SENSORS JOURNAL, 2019, 19 (12) :4494-4500
[9]  
Cheng P, 2018, TENCON IEEE REGION, P1994, DOI 10.1109/TENCON.2018.8650546
[10]  
Chin Z.Y., 2008, BEST STUDY CLASSIFIC