EEG data classification using wavelet features selected by Wilcoxon statistics

Abstract This paper introduces a method to classify EEG signals using features extracted by an integration of wavelet transform and the nonparametric Wilcoxon test. Orthogonal Haar wavelet coefficients are ranked based on the Wilcoxon test’s statistics. The most prominent discriminant wavelets are assembled to form a feature set that serves as inputs to the naïve Bayes classifier. Two benchmark datasets, named Ia and Ib, downloaded from the brain–computer interface (BCI) competition II are employed for the experiments. Classification performance is evaluated using accuracy, mutual information, Gini coefficient and F-measure. Widely used classifiers, including feedforward neural network, support vector machine, k-nearest neighbours, ensemble learning Adaboost and adaptive neuro-fuzzy inference system, are also implemented for comparisons. The proposed combination of Haar wavelet features and naïve Bayes classifier considerably dominates the competitive classification approaches and outperforms the best performance on the Ia and Ib datasets reported in the BCI competition II. Application of naïve Bayes also provides a low computational cost approach that promotes the implementation of a potential real-time BCI system.

[1]  E. Lehmann,et al.  Nonparametrics: Statistical Methods Based on Ranks , 1976 .

[2]  Kaushik K. Majumdar,et al.  Single-Trial EEG Classification Using Logistic Regression Based on Ensemble Synchronization , 2014, IEEE Journal of Biomedical and Health Informatics.

[3]  Yasuharu Koike,et al.  A real-time BCI with a small number of channels based on CSP , 2011, Neural Computing and Applications.

[4]  Abdulhamit Subasi,et al.  EEG signal classification using PCA, ICA, LDA and support vector machines , 2010, Expert Syst. Appl..

[5]  Witold Pedrycz,et al.  Fuzzy wavelet packet based feature extraction method and its application to biomedical signal classification , 2005, IEEE Transactions on Biomedical Engineering.

[6]  Yan Li,et al.  Modified CC-LR algorithm with three diverse feature sets for motor imagery tasks classification in EEG based brain-computer interface , 2014, Comput. Methods Programs Biomed..

[7]  Daniel Rivero,et al.  Automatic epileptic seizure detection in EEGs based on line length feature and artificial neural networks , 2010, Journal of Neuroscience Methods.

[8]  Daniel Rivero,et al.  Automatic feature extraction using genetic programming: An application to epileptic EEG classification , 2011, Expert Syst. Appl..

[9]  Jian Pei,et al.  A rank sum test method for informative gene discovery , 2004, KDD.

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

[11]  Ahmad Ayatollahi,et al.  An efficient neural network based method for medical image segmentation , 2014, Comput. Biol. Medicine.

[12]  Ferat Sahin,et al.  New classification techniques for electroencephalogram (EEG) signals and a real-time EEG control of a robot , 2011, Neural Computing and Applications.

[13]  Wei-Yen Hsu,et al.  EEG-based motor imagery classification using enhanced active segment selection and adaptive classifier , 2011, Comput. Biol. Medicine.

[14]  Vladimir Bostanov,et al.  BCI competition 2003-data sets Ib and IIb: feature extraction from event-related brain potentials with the continuous wavelet transform and the t-value scalogram , 2004, IEEE Transactions on Biomedical Engineering.

[15]  Yüksel Özbay,et al.  Integration of type-2 fuzzy clustering and wavelet transform in a neural network based ECG classifier , 2011, Expert Syst. Appl..

[16]  Saeid Nahavandi,et al.  Classification of healthcare data using genetic fuzzy logic system and wavelets , 2015, Expert Syst. Appl..

[17]  W. Kruskal,et al.  Use of Ranks in One-Criterion Variance Analysis , 1952 .

[18]  R. Quian Quiroga,et al.  Unsupervised Spike Detection and Sorting with Wavelets and Superparamagnetic Clustering , 2004, Neural Computation.

[19]  Yan Li,et al.  Clustering technique-based least square support vector machine for EEG signal classification , 2011, Comput. Methods Programs Biomed..

[20]  Reza Rostami,et al.  Classifying depression patients and normal subjects using machine learning techniques , 2011, 2011 19th Iranian Conference on Electrical Engineering.

[21]  Maryam Vatankhah,et al.  Perceptual pain classification using ANFIS adapted RBF kernel support vector machine for therapeutic usage , 2013, Appl. Soft Comput..

[22]  Bijaya K. Panigrahi,et al.  A comparative study of wavelet families for EEG signal classification , 2011, Neurocomputing.

[23]  K.-R. Muller,et al.  BCI meeting 2005-workshop on BCI signal processing: feature extraction and translation , 2006, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[24]  Shiliang Sun,et al.  The stochastic approximation method for adaptive Bayesian classifiers: towards online brain–computer interfaces , 2011, Neural Computing and Applications.

[25]  Yu Cao,et al.  Motor imagery classification based on joint regression model and spectral power , 2013, Neural Computing and Applications.

[26]  Duoqian Miao,et al.  Best basis-based wavelet packet entropy feature extraction and hierarchical EEG classification for epileptic detection , 2011, Expert Syst. Appl..

[27]  Abdulhamit Subasi,et al.  Classification of EEG signals using neural network and logistic regression , 2005, Comput. Methods Programs Biomed..

[28]  Nasour Bagheri,et al.  Multiple classifier system for EEG signal classification with application to brain–computer interfaces , 2012, Neural Computing and Applications.

[29]  Reza Boostani,et al.  A new approach for EEG signal classification of schizophrenic and control participants , 2011, Expert Syst. Appl..

[30]  Chao Li,et al.  Enhancement of Medical Image Details via Wavelet Homomorphic Filtering Transform , 2014, J. Intell. Syst..

[31]  Radford M. Neal.  Pattern Recognition and Machine Learning , 2007, Technometrics.

[32]  Mitchell J. Mergenthaler.  Nonparametrics: Statistical Methods Based on Ranks , 1979 .

[33]  A. S. Rodionov,et al.  Comparison of linear, nonlinear and feature selection methods for EEG signal classification , 2004, International Conference on Actual Problems of Electron Devices Engineering, 2004. APEDE 2004..

[34]  Christopher M. Bishop,et al.  Pattern Recognition and Machine Learning (Information Science and Statistics) , 2006 .

[35]  Yan Guozheng,et al.  EEG feature extraction based on wavelet packet decomposition for brain computer interface , 2008 .

[36]  Yangsong Zhang,et al.  Z-Score Linear Discriminant Analysis for EEG Based Brain-Computer Interfaces , 2013, PloS one.

[37]  U. Rajendra Acharya,et al.  EEG Signal Analysis: A Survey , 2010, Journal of Medical Systems.

[38]  U. Rajendra Acharya,et al.  Author's Personal Copy Biomedical Signal Processing and Control Automated Diagnosis of Epileptic Eeg Using Entropies , 2022 .

[39]  Ram Bilas Pachori,et al.  Classification of ictal and seizure-free EEG signals using fractional linear prediction , 2014, Biomed. Signal Process. Control..

[40]  Motoaki Kawanabe,et al.  Toward Unsupervised Adaptation of LDA for Brain–Computer Interfaces , 2011, IEEE Transactions on Biomedical Engineering.

[41]  H. Flor,et al.  A spelling device for the paralysed , 1999, Nature.