Analyze EEG signals with extreme learning machine based on PMIS feature selection

AbstractAn electroencephalogram (EEG)-based brain–computer interface (BCI) system can be used to determine the intents for a paralyzed user by analyzing the EEG signals recorded from his scalp. The key technology of BCI systems is feature extraction and pattern recognition algorithm. Many of the BCI research show that nonlinear classification algorithms perform better than the linear, but they are usually far slower than required. Besides, many of these methods employ less significant features and could make the classification process less efficient. In this paper, a method is provided to identify different patterns of EEG signals. A fast nonlinear classification algorithm extreme learning machine (ELM) is provided to identify two different EEG signals. After feature extraction, we adopt a nonlinear feature selection algorithm partial mutual information-based feature selection (PMIS) to eliminate the less significant features. The experimental result shows that (1) ELM can discriminate the two patterns of EEG signals with a relatively satisfactory accuracy and (2) PMIS feature selection can find out the important features effectively and improve the classification accuracy. The final accuracy is 91.5 % against the best existed result of 88.7 % for the tested data set. The classification performance may be further improved by adding complementary features from EEG. This approach may eventually lead to a reliable EEG-based BCI studies.

[1]  Xizhao Wang,et al.  Upper integral network with extreme learning mechanism , 2011, Neurocomputing.

[2]  Holger R. Maier,et al.  Review of Input Variable Selection Methods for Artificial Neural Networks , 2011 .

[3]  Ashish Sharma,et al.  Seasonal to interannual rainfall probabilistic forecasts for improved water supply management: Part 1 — A strategy for system predictor identification , 2000 .

[4]  Justin Werfel,et al.  BCI competition 2003-data set Ia: combining gamma-band power with slow cortical potentials to improve single-trial classification of electroencephalographic signals , 2004, IEEE Transactions on Biomedical Engineering.

[5]  Upmanu Lall,et al.  Seasonal to interannual rainfall probabilistic forecasts for improved water supply management : Part 2 - Predictor identification of quarterly rainfall using ocean-atmosphere information , 2000 .

[6]  Mingtian Zhou,et al.  A two stages sparse SVM training , 2014, Int. J. Mach. Learn. Cybern..

[7]  N. Birbaumer,et al.  BCI2000: a general-purpose brain-computer interface (BCI) system , 2004, IEEE Transactions on Biomedical Engineering.

[8]  Dianhui Wang,et al.  Evolutionary extreme learning machine ensembles with size control , 2013, Neurocomputing.

[9]  H. Berger Über das Elektrenkephalogramm des Menschen , 1933, Archiv für Psychiatrie und Nervenkrankheiten.

[10]  Patrick P. K. Chan,et al.  Bi-firing deep neural networks , 2013, International Journal of Machine Learning and Cybernetics.

[11]  G. Dietsch,et al.  Fourier-Analyse von Elektrencephalogrammen des Menschen , 1932, Pflüger's Archiv für die gesamte Physiologie des Menschen und der Tiere.

[12]  Holger R. Maier,et al.  Non-linear variable selection for artificial neural networks using partial mutual information , 2008, Environ. Model. Softw..

[13]  P. Scheltens,et al.  Research criteria for the diagnosis of Alzheimer's disease: revising the NINCDS–ADRDA criteria , 2007, The Lancet Neurology.

[14]  Wang Xi-zhao,et al.  Architecture selection for networks trained with extreme learning machine using localized generalization error model , 2013 .

[15]  Dianhui Wang,et al.  Extreme learning machines: a survey , 2011, Int. J. Mach. Learn. Cybern..

[16]  Chee Kheong Siew,et al.  Extreme learning machine: Theory and applications , 2006, Neurocomputing.

[17]  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.

[18]  Andrzej Cichocki,et al.  On the Early Diagnosis of Alzheimer’s Disease from EEG Signals: A Mini-Review , 2011 .

[19]  Ashish Sharma,et al.  Seasonal to interannual rainfall probabilistic forecasts for improved water supply management: Part 3 — A nonparametric probabilistic forecast model , 2000 .

[20]  Chunru Dong,et al.  An experimental study on stability and generalization of extreme learning machines , 2015, Int. J. Mach. Learn. Cybern..

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

[22]  Hongming Zhou,et al.  Extreme Learning Machine for Regression and Multiclass Classification , 2012, IEEE Transactions on Systems, Man, and Cybernetics, Part B (Cybernetics).