Feature selection using regularized neighbourhood component analysis to enhance the classification performance of motor imagery signals

In motor imagery (MI) based brain-computer interface (BCI) signal analysis, mu and beta rhythms of electroencephalograms (EEGs) are widely investigated due to their high temporal resolution and capability to define the different movement-related mental tasks separately. However, due to the high dimensions and subject-specific behaviour of EEG features, there is a need for a suitable feature selection algorithm that can select the optimal features to give the best classification performance along with increased computational efficiency. The present study proposes a feature selection algorithm based on neighbourhood component analysis (NCA) with modification of the regularization parameter. In the experiment, time, frequency, and phase features of the EEG are extracted using a dual-tree complex wavelet transform (DTCWT). Afterwards, the proposed algorithm selects the most significant EEG features, and using these selected features, a support vector machine (SVM) classifier performs the classification of MI signals. The proposed algorithm has been validated experimentally on two public BCI datasets (BCI Competition II Dataset III and BCI Competition IV Dataset 2b). The classification performance of the algorithm is quantified by the average accuracy and kappa coefficient, whose values are 80.7% and 0.615 respectively. The performance of the proposed algorithm is compared with standard feature selection methods based on Genetic Algorithm (GA), Principal Component Analysis (PCA), and ReliefF and performs better than these methods. Further, the proposed algorithm selects the lowest number of features and results in increased computational efficiency, which makes it a promising feature selection tool for an MI-based BCI system.

[1]  T. Jung,et al.  Translation of EEG Spatial Filters from Resting to Motor Imagery Using Independent Component Analysis , 2012, PloS one.

[2]  Nihat Yilmaz,et al.  Feature Selection Method Based on Artificial Bee Colony Algorithm and Support Vector Machines for Medical Datasets Classification , 2013, TheScientificWorldJournal.

[3]  M. Huzmezan,et al.  A wavelet based de-noising technique for ocular artifact correction of the electroencephalogram , 2002, Proceedings of the Second Joint 24th Annual Conference and the Annual Fall Meeting of the Biomedical Engineering Society] [Engineering in Medicine and Biology.

[4]  Xingyu Wang,et al.  Towards correlation-based time window selection method for motor imagery BCIs , 2018, Neural Networks.

[5]  N. Birbaumer,et al.  Brain–computer interfaces for communication and rehabilitation , 2016, Nature Reviews Neurology.

[6]  Qingsong Ai,et al.  Feature Selection for Motor Imagery EEG Classification Based on Firefly Algorithm and Learning Automata , 2017, Sensors.

[7]  Venkateswarlu Gonuguntla,et al.  BMFLC with neural network and DE for better event classification , 2013, 2013 International Winter Workshop on Brain-Computer Interface (BCI).

[8]  Dan Liu,et al.  An effective feature extraction method by power spectral density of EEG signal for 2-class motor imagery-based BCI , 2018, Medical & Biological Engineering & Computing.

[9]  Moritz Grosse-Wentrup,et al.  Beamforming in Noninvasive Brain–Computer Interfaces , 2009, IEEE Transactions on Biomedical Engineering.

[10]  Soo-Young Lee,et al.  EEG-Based Classification of Implicit Intention During Self-Relevant Sentence Reading , 2016, IEEE Transactions on Cybernetics.

[11]  Jessica Cantillo-Negrete,et al.  Motor Imagery-Based Brain-Computer Interface Coupled to a Robotic Hand Orthosis Aimed for Neurorehabilitation of Stroke Patients , 2018, Journal of healthcare engineering.

[12]  Pratyusha Rakshit,et al.  Feature Selection of Motor Imagery EEG Signals Using Firefly Temporal Difference Q-Learning and Support Vector Machine , 2013, SEMCCO.

[13]  F. L. D. Silva,et al.  Event-related EEG/MEG synchronization and desynchronization: basic principles , 1999, Clinical Neurophysiology.

[14]  Geoffrey E. Hinton,et al.  Neighbourhood Components Analysis , 2004, NIPS.

[15]  Andry Rakotonirainy,et al.  Long Short Term Memory Hyperparameter Optimization for a Neural Network Based Emotion Recognition Framework , 2018, IEEE Access.

[16]  S. N. Resalat,et al.  A Study of Various Feature Extraction Methods on a Motor Imagery Based Brain Computer Interface System , 2016, Basic and clinical neuroscience.

[17]  Michael Unser,et al.  On the Shiftability of Dual-Tree Complex Wavelet Transforms , 2009, IEEE Transactions on Signal Processing.

[18]  Subhojit Ghosh,et al.  Statistical Wavelet Features, PCA, and SVM Based Approach for EEG Signals Classification , 2015 .

[19]  Rebeca Corralejo,et al.  Feature selection using a genetic algorithm in a motor imagery-based Brain Computer Interface , 2011, 2011 Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[20]  Christos Boutsidis,et al.  Unsupervised feature selection for principal components analysis , 2008, KDD.

[21]  Rabab Kreidieh Ward,et al.  Bayesian optimization of BCI parameters , 2016, 2016 IEEE Canadian Conference on Electrical and Computer Engineering (CCECE).

[22]  Fengxi Song,et al.  Feature Selection Using Principal Component Analysis , 2010, 2010 International Conference on System Science, Engineering Design and Manufacturing Informatization.

[23]  Alimed Celecia Ramos,et al.  Feature Selection methods applied to Motor Imagery task classification , 2016, 2016 IEEE Latin American Conference on Computational Intelligence (LA-CCI).

[24]  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).

[25]  Mahyar Hamedi,et al.  Electroencephalographic Motor Imagery Brain Connectivity Analysis for BCI: A Review , 2016, Neural Computation.

[26]  Xiaopei Wu,et al.  Envelope detection based on online ICA algorithm and its application to motor imagery classification , 2013, 2013 6th International IEEE/EMBS Conference on Neural Engineering (NER).

[27]  Xingyu Wang,et al.  Sparse Bayesian Classification of EEG for Brain–Computer Interface , 2016, IEEE Transactions on Neural Networks and Learning Systems.

[28]  Ad Aertsen,et al.  Review of the BCI Competition IV , 2012, Front. Neurosci..

[29]  Ayman Atia,et al.  Brain computer interfacing: Applications and challenges , 2015 .

[30]  Darrell Whitley,et al.  A genetic algorithm tutorial , 1994, Statistics and Computing.

[31]  Wei Yang,et al.  Neighborhood Component Feature Selection for High-Dimensional Data , 2012, J. Comput..

[32]  Xingyu Wang,et al.  Temporally Constrained Sparse Group Spatial Patterns for Motor Imagery BCI , 2019, IEEE Transactions on Cybernetics.

[33]  Gert Pfurtscheller,et al.  Motor imagery and direct brain-computer communication , 2001, Proc. IEEE.

[34]  Jorma Laaksonen,et al.  Regularized Neighborhood Component Analysis , 2007, SCIA.

[35]  Kwee-Bo Sim,et al.  Analysis the effect of PCA for feature reduction in non-stationary EEG based motor imagery of BCI system , 2014 .

[36]  Alain Rakotomamonjy,et al.  Ensemble of SVMs for Improving Brain Computer Interface P300 Speller Performances , 2005, ICANN.

[37]  Jinsong Leng,et al.  A genetic Algorithm-Based feature selection , 2014 .

[38]  Klaus-Robert Müller,et al.  The BCI competition 2003: progress and perspectives in detection and discrimination of EEG single trials , 2004, IEEE Transactions on Biomedical Engineering.

[39]  Xingyu Wang,et al.  Spatial-Temporal Discriminant Analysis for ERP-Based Brain-Computer Interface , 2013, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[40]  Shiliang Sun,et al.  A review of optimization methodologies in support vector machines , 2011, Neurocomputing.

[41]  Yu Zhang,et al.  EEG classification using sparse Bayesian extreme learning machine for brain–computer interface , 2018, Neural Computing and Applications.

[42]  Xingyu Wang,et al.  Discriminative Feature Extraction via Multivariate Linear Regression for SSVEP-Based BCI , 2016, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[43]  J. Wolpaw,et al.  Clinical Applications of Brain-Computer Interfaces: Current State and Future Prospects , 2009, IEEE Reviews in Biomedical Engineering.

[44]  Igor Kononenko,et al.  Estimating Attributes: Analysis and Extensions of RELIEF , 1994, ECML.

[45]  Pratyusha Rakshit,et al.  Artificial Bee Colony Based Feature Selection for Motor Imagery EEG Data , 2012, BIC-TA.