Feature subset and time segment selection for the classification of EEG data based motor imagery

Abstract The selection of feature subset and time segment is of great significance to the benefit of motor imagery classification. Hence, applying it in classification as well as itself alone has become an increasingly important research field in the brain-computer interface (BCI) systems. Most of the existing literatures only focus on binary-class classification situations in a fixed time segment. However, the modern BCI systems usually have to deal with more motor imagery classes. In this paper, we propose two Parzen window based methods to select the discriminative feature subset and subject-specific time segment. Further, we extend the proposed methods to multi-class issues. Finally, a soft Naive Bayesian classifier is designed to solve not only binary-class but also multi-class motor imagery problems. The proposed methods are validated on two well-known datasets, BCI competition IV dataset 2a and 2b. Experimental results reveal that the proposed methods achieve an improvement of 4.38 % for 2a and 2.54 % for 2b in comparison with the state-of-the-art methods, respectively. Meanwhile, both proposed methods achieve an improvement of 0.04 for 2a and 0.05 for 2b in kappa coefficient, respectively. Besides, the proposed multi-class methods would potentially contribute to the online BCI systems in practice.

[1]  Chiew Tong Lau,et al.  A New Discriminative Common Spatial Pattern Method for Motor Imagery Brain–Computer Interfaces , 2009, IEEE Transactions on Biomedical Engineering.

[2]  H. Chernoff A Measure of Asymptotic Efficiency for Tests of a Hypothesis Based on the sum of Observations , 1952 .

[3]  Tarald O. Kvålseth,et al.  A Coefficient of Agreement for Nominal Scales: An Asymmetric Version of Kappa , 1991 .

[4]  Umberto Castellani,et al.  Infinite Latent Feature Selection: A Probabilistic Latent Graph-Based Ranking Approach , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[5]  Klaus-Robert Müller,et al.  Spatio-spectral filters for improving the classification of single trial EEG , 2005, IEEE Transactions on Biomedical Engineering.

[6]  Rajesh P. N. Rao,et al.  Towards adaptive classification for BCI , 2006, Journal of neural engineering.

[7]  R. Leeb,et al.  BCI Competition 2008 { Graz data set B , 2008 .

[8]  Yiming Yang,et al.  A Comparative Study on Feature Selection in Text Categorization , 1997, ICML.

[9]  Lloyd A. Smith,et al.  Practical feature subset selection for machine learning , 1998 .

[10]  Tom Fawcett,et al.  An introduction to ROC analysis , 2006, Pattern Recognit. Lett..

[11]  Klaus-Robert Müller,et al.  The non-invasive Berlin Brain–Computer Interface: Fast acquisition of effective performance in untrained subjects , 2007, NeuroImage.

[12]  M Congedo,et al.  A review of classification algorithms for EEG-based brain–computer interfaces , 2007, Journal of neural engineering.

[13]  Thomas M. Cover,et al.  Elements of Information Theory , 2005 .

[14]  Kai Zhao,et al.  Multiclass Motor Imagery Recognition of Single Joint in Upper Limb Based on NSGA- II OVO TWSVM , 2018, Comput. Intell. Neurosci..

[15]  ST Johnston,et al.  Paradox of pattern separation and adult neurogenesis: A dual role for new neurons balancing memory resolution and robustness , 2016, Neurobiology of Learning and Memory.

[16]  Rabab K Ward,et al.  A survey of signal processing algorithms in brain–computer interfaces based on electrical brain signals , 2007, Journal of neural engineering.

[17]  Pat Langley,et al.  Selection of Relevant Features and Examples in Machine Learning , 1997, Artif. Intell..

[18]  Charles W. Therrien,et al.  Discrete Random Signals and Statistical Signal Processing , 1992 .

[19]  Ronald L. Rivest,et al.  Training a 3-node neural network is NP-complete , 1988, COLT '88.

[20]  Chih-Jen Lin,et al.  A comparison of methods for multiclass support vector machines , 2002, IEEE Trans. Neural Networks.

[21]  Richard D. Deveaux,et al.  Applied Smoothing Techniques for Data Analysis , 1999, Technometrics.

[22]  Ron Kohavi,et al.  Wrappers for Feature Subset Selection , 1997, Artif. Intell..

[23]  G. Pfurtscheller,et al.  Brain-Computer Interfaces for Communication and Control. , 2011, Communications of the ACM.

[24]  G.F. Inbar,et al.  Feature selection for the classification of movements from single movement-related potentials , 2002, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[25]  Paul S. Bradley,et al.  Feature Selection via Concave Minimization and Support Vector Machines , 1998, ICML.

[26]  Dario Farina,et al.  Comparison of feature selection and classification methods for a brain–computer interface driven by non-motor imagery , 2010, Medical & Biological Engineering & Computing.

[27]  Cyrus Shahabi,et al.  Feature subset selection and feature ranking for multivariate time series , 2005, IEEE Transactions on Knowledge and Data Engineering.

[28]  Jie Wang,et al.  Toward optimal feature and time segment selection by divergence method for EEG signals classification , 2018, Comput. Biol. Medicine.

[29]  Pascal Vincent,et al.  Representation Learning: A Review and New Perspectives , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[30]  Klaus-Robert Müller,et al.  Combined Optimization of Spatial and Temporal Filters for Improving Brain-Computer Interfacing , 2006, IEEE Transactions on Biomedical Engineering.

[31]  Klaus-Robert Müller,et al.  The Berlin Brain-Computer Interface: Accurate performance from first-session in BCI-naive subjects , 2008, IEEE Transactions on Biomedical Engineering.

[32]  Cuntai Guan,et al.  Regularizing Common Spatial Patterns to Improve BCI Designs: Unified Theory and New Algorithms , 2011, IEEE Transactions on Biomedical Engineering.

[33]  Jie Wang,et al.  Feature extraction by common spatial pattern in frequency domain for motor imagery tasks classification , 2017, 2017 29th Chinese Control And Decision Conference (CCDC).

[34]  K.-R. Muller,et al.  Optimizing Spatial filters for Robust EEG Single-Trial Analysis , 2008, IEEE Signal Processing Magazine.

[35]  Cuntai Guan,et al.  Mutual information-based selection of optimal spatial-temporal patterns for single-trial EEG-based BCIs , 2012, Pattern Recognit..

[36]  Bo Tang,et al.  Toward Optimal Feature Selection in Naive Bayes for Text Categorization , 2016, IEEE Transactions on Knowledge and Data Engineering.

[37]  Cuntai Guan,et al.  Optimizing the Channel Selection and Classification Accuracy in EEG-Based BCI , 2011, IEEE Transactions on Biomedical Engineering.

[38]  Chong-Ho Choi,et al.  Input Feature Selection by Mutual Information Based on Parzen Window , 2002, IEEE Trans. Pattern Anal. Mach. Intell..

[39]  Konstantinos N. Plataniotis,et al.  Separable Common Spatio-Spectral Patterns for Motor Imagery BCI Systems , 2016, IEEE Transactions on Biomedical Engineering.

[40]  Huan Liu,et al.  Feature Selection for High-Dimensional Data: A Fast Correlation-Based Filter Solution , 2003, ICML.

[41]  Gunnar Blohm,et al.  Multi-Class Motor Imagery EEG Decoding for Brain-Computer Interfaces , 2012, Front. Neurosci..

[42]  Perica Strbac,et al.  Toward optimal feature selection using ranking methods and classification algorithms , 2011 .

[43]  Daphne Koller,et al.  Toward Optimal Feature Selection , 1996, ICML.

[44]  Jonathan R Wolpaw,et al.  Brain–computer interfaces as new brain output pathways , 2007, The Journal of physiology.

[45]  Klaus-Robert Müller,et al.  Adaptive Methods in BCI Research - An Introductory Tutorial , 2009 .

[46]  Mineichi Kudo,et al.  Entropy Criterion for Classifier-Independent Feature Selection , 2005, KES.

[47]  J. Wolpaw Chapter 64 Brain-computer interfaces (BCIs) for communication and control: a mini-review , 2004 .

[48]  G. Pfurtscheller,et al.  Optimal spatial filtering of single trial EEG during imagined hand movement. , 2000, IEEE transactions on rehabilitation engineering : a publication of the IEEE Engineering in Medicine and Biology Society.

[49]  Cuntai Guan,et al.  Filter Bank Common Spatial Pattern (FBCSP) in Brain-Computer Interface , 2008, 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence).

[50]  Isabelle Guyon,et al.  An Introduction to Variable and Feature Selection , 2003, J. Mach. Learn. Res..

[51]  Ugur Halici,et al.  A novel deep learning approach for classification of EEG motor imagery signals , 2017, Journal of neural engineering.

[52]  José del R. Millán,et al.  Evaluation Criteria for BCI Research , 2007 .

[53]  Chih-Jen Lin,et al.  Errata to "A comparison of methods for multiclass support vector machines" , 2002, IEEE Trans. Neural Networks.

[54]  E. Parzen On Estimation of a Probability Density Function and Mode , 1962 .

[55]  M Congedo,et al.  A review of classification algorithms for EEG-based brain–computer interfaces: a 10 year update , 2018, Journal of neural engineering.

[56]  Vernon J. Lawhern,et al.  Detecting alpha spindle events in EEG time series using adaptive autoregressive models , 2013, BMC Neuroscience.