A feature extraction and classification algorithm for motor imagery EEG signals based on decision tree and CSP-SVM

被引:0
作者
Luo, Yuan [1 ]
He, Xiaoyi [1 ]
Ren, Ke [1 ]
机构
[1] Chongqing Univ Posts & Telecommun, Chongqing Municipal Level Key Lab Photoelect Info, Chongqing 400065, Peoples R China
来源
OPTICS IN HEALTH CARE AND BIOMEDICAL OPTICS XI | 2021年 / 11900卷
关键词
Motor imagery; EEG; decision tree; common spatial pattern; support vector machine;
D O I
10.1117/12.2601842
中图分类号
Q5 [生物化学];
学科分类号
071010 ; 081704 ;
摘要
In order to solve the problems of low recognition accuracy for motor imagery EEG signals, this paper presents a feature extraction and classification algorithm based on decision tree and CSP-SVM. Firstly, we select the fixed frequency of signals ranging from 8 to 30 Hz. Secondly, multiple spatial filters are constructed by using the one versus the rest common spatial pattern (OVR-CSP) and extract the feature vectors. Support vector machine (SVM) is employed to classify the feature vectors so that the best spatial filter is selected. We build the first branch of decision tree with the spatial filter selected and SVM. Then, OVR-CSP and SVM are used to build the branches of the decision tree repeatedly. Finally, 2005 BCI competition IIIa data set is used to validate the effect of the proposed algorithm. The results show that the highest accuracy of the proposed algorithm can reach 94.27% which proves the effectiveness of the algorithm.
引用
收藏
页数:8
相关论文
共 21 条
  • [1] Filter bank common spatial pattern algorithm on BCI competition IV Datasets 2a and 2b
    Ang, Kai Keng
    Chin, Zheng Yang
    Wang, Chuanchu
    Guan, Cuntai
    Zhang, Haihong
    [J]. FRONTIERS IN NEUROSCIENCE, 2012, 6
  • [2] Chatterjee Rajdeep, 2016, 2016 2nd International Conference on Computational Intelligence and Networks (CINE), P84, DOI 10.1109/CINE.2016.22
  • [3] A novel machine learning based feature selection for motor imagery EEG signal classification in Internet of medical things environment
    Chatterjee, Rajdeep
    Maitra, Tanmoy
    Islam, S. K. Hafizul
    Hassan, Mohammad Mehedi
    Alamri, Atif
    Fortino, Giancarlo
    [J]. FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2019, 98 : 419 - 434
  • [4] A novel hybrid kernel function relevance vector machine for multi-task motor imagery EEG classification
    Dong, Enzeng
    Zhou, Kairui
    Tong, Jigang
    Du, Shengzhi
    [J]. BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 60
  • [5] Comparative study of motor imagery classification based on BP-NN and SVM
    Jia, Hongru
    Wang, Shuai
    Zheng, Dezhi
    Qu, Xiaolei
    Fan, Shangchun
    [J]. JOURNAL OF ENGINEERING-JOE, 2019, 2019 (23): : 8646 - 8649
  • [6] Motor imagery task classification using transformation based features
    Khorshidtalab, Aida
    Salami, Momoh J. E.
    Akmeliawati, Rini
    [J]. BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2017, 33 : 213 - 219
  • [7] An effective feature extraction method by power spectral density of EEG signal for 2-class motor imagery-based BCI
    Kim, Chungsong
    Sun, Jinwei
    Liu, Dan
    Wang, Qisong
    Paek, Sunggyun
    [J]. MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2018, 56 (09) : 1645 - 1658
  • [8] Riemannian Distances for Signal Classification by Power Spectral Density
    Li, Yili
    Wong, Kon Max
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN SIGNAL PROCESSING, 2013, 7 (04) : 655 - 669
  • [9] Mahmood A, 2017, IEEE ENG MED BIO, P1034, DOI 10.1109/EMBC.2017.8037003
  • [10] Augmented Complex Common Spatial Patterns for Classification of Noncircular EEG From Motor Imagery Tasks
    Park, Cheolsoo
    Took, Clive Cheong
    Mandic, Danilo P.
    [J]. IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2014, 22 (01) : 1 - 10