Skip to main content

METHODS article

Front. Neurosci., 13 October 2021
Sec. Brain Imaging Methods
This article is part of the Research Topic Machine Learning for Quantitative Neuroimaging Analysis View all 7 articles

Single-Trial EEG Classification via Orthogonal Wavelet Decomposition-Based Feature Extraction

\nFeifei Qi&#x;Feifei Qi1Wenlong Wang&#x;Wenlong Wang2Xiaofeng XieXiaofeng Xie3Zhenghui GuZhenghui Gu2Zhu Liang YuZhu Liang Yu2Fei WangFei Wang4Yuanqing LiYuanqing Li2Wei Wu
Wei Wu2*
  • 1School of Internet Finance and Information Engineering, Guangdong University of Finance, Guangzhou, China
  • 2School of Automation Science and Engineering, South China University of Technology, Guangzhou, China
  • 3Mechanical and Electrical Engineering College, Hainan University, Hainan, China
  • 4School of Software, South China Normal University, Guangzhou, China

Achieving high classification performance is challenging due to non-stationarity and low signal-to-noise ratio (low SNR) characteristics of EEG signals. Spatial filtering is commonly used to improve the SNR yet the individual differences in the underlying temporal or frequency information is often ignored. This paper investigates motor imagery signals via orthogonal wavelet decomposition, by which the raw signals are decomposed into multiple unrelated sub-band components. Furthermore, channel-wise spectral filtering via weighting the sub-band components are implemented jointly with spatial filtering to improve the discriminability of EEG signals, with an l2-norm regularization term embedded in the objective function to address the underlying over-fitting issue. Finally, sparse Bayesian learning with Gaussian prior is applied to the extracted power features, yielding an RVM classifier. The classification performance of SEOWADE is significantly better than those of several competing algorithms (CSP, FBCSP, CSSP, CSSSP, and shallow ConvNet). Moreover, scalp weight maps of the spatial filters optimized by SEOWADE are more neurophysiologically meaningful. In summary, these results demonstrate the effectiveness of SEOWADE in extracting relevant spatio-temporal information for single-trial EEG classification.

1. Introduction

Brain-computer interface (BCI) systems provide an approach for communicating with the external world by brain signals (Lemm et al., 2011). BCI systems based on Electroencephalogram (EEG) is the most common non-invasive modality, as EEG is inexpensive and has high temporal resolution. Motor imagery based BCI is a commonly applied paradigm that can efficiently decode the imagination of movement, and related features are derived from event-related desynchronization/synchronization (ERD/ERS) (Blankertz et al., 2008). The signal processing steps of a BCI system include brain signal acquisition, EEG signal preprocessing, feature extraction and feature classification. The preprocessing step aims at enhancing the relevant information by attenuating the artifacts and noise, e.g., band-pass filtering. The feature extraction stage forms discriminative features for the performed tasks in the spatial domain, temporal domain or frequency domain. The extracted features are then used to train a classification or regression model to decode the user's intent. Hence, advanced signal processing plays a key role in neuroscience research, and feature extraction and classification of EEG signal have always been the two most critical problems encountered in EEG signal analysis.

However, EEG signal processing is very challenging due to the low spatial resolution, high non-stationarity and high intra-subject variability of EEG. The challenges in developing an efficient feature extraction algorithm are to address the above-mentioned issues and form discriminative features. Common spatial patterns (CSP) computes spatial filters, i.e., linear combinations of EEG channels, to enhance class-discriminative band power features contained in the EEG (Blankertz et al., 2008; Lemm et al., 2011). However, the potential problem of CSP is that it does not take into consideration the discriminative information in the temporal or frequency domain. To address this problem, many studies have applied spatial filtering or spatio-temporal filtering for feature extraction, as reviewed below. The common spatio-spectral patterns (CSSP) algorithm (Lemm et al., 2005) is an approach to simultaneously optimize spatial filters and channel-wise temporal filters; however, embedding one temporal delay has less flexility. The common sparse spectral spatial patterns (CSSSP) algorithm (Dornhege et al., 2006) optimizes one high-order finite impulse response filter for all the channels, and a sparse penalty term for the temporal filter is embedded in the objective function of CSSSP. However, no close-form solution is available by CSSSP. DFBCSP (Higashi and Tanaka, 2013) is developed to maximize the power ratio between two motor imagery states (i.e., two classes) to design spatial and temporal filter pairs. Note that the high-dimensional filters are not regularized to ameliorate the potential over-fitting issue. The filter bank common spatial pattern (FBCSP) algorithm (Ang et al., 2008) operates spatial filtering and spectral filtering in a sequential manner, which first filters the EEG signals into several distinct sub-band components and then selects discriminative features from the sub-bands via specific criteria. The variants of FBCSP (Novi et al., 2007; Kavitha et al., 2008; Zhang et al., 2011) differ in the feature selection criterion. BSSFO (Suk and Lee, 2013) formulates a Bayesian inference framework for the features of each sub-band. Along a different line, several algorithms have been proposed to optimize the spatial filters and spectral/temporal filters in separate stages. SPEC-CSP (Tomioka et al., 2006) and ISSPL (Wu et al., 2008) iteratively optimize spatial filters and spectral filters by maximizing the Fisher ratio or margin hyper-plane. However, the objective functions of FBCSP, ISSPL and SPEC-CSP are distinctive for spatial filter and spectral filter optimization; therefore, convergence and optimality cannot be guaranteed. In our previous work, the regularized spatio-temporal filtering and classification (RSTFC) algorithm (Qi et al., 2015) is proposed as a new EEG analysis framework, which shows advantageous classification performance. In recent years, there has been increasing interest in using deep learning with convolutional neural networks to decode imagined tasks from raw EEG signals (Lawhern et al., 2016; Schirrmeister et al., 2017). However, more training samples are needed, as the number of parameters is larger than that of the conventional approaches.

In addition to spatial filtering, time-frequency representation is highly desirable for EEG feature extraction from time-frequency plots (Qin and He, 2005; Yang et al., 2007). In recent years, wavelet transform has been found to be an effective time-frequency analysis tool for analyzing transient signals and has been applied to seizure detection (Bhattacharyya and Pachori, 2017) and emotion recognition (Mohammadi et al., 2017). However, there is no standard method for selecting the best wavelet and determining the decomposition level for processing EEG signals (Hlawatsch and Boudreaux-Bartels, 1992; Subasi et al., 2006). An algorithm based on an autoregressive model and wavelet packet decomposition is proposed in Zhang et al. (2017); however, the decomposed signals are redundant. CSP is employed for feature extraction on the EEG signal that are reconstructed and de-noised by single level wavelet in Zhang et al. (2010). In Mousavi et al. (2011), wavelet packets are used to decompose EEG signals into multiple levels, and fuzzy logic is combined to select the discriminative packets. EEG signals are decomposed by wavelets (Robinson et al., 2011, 2013) after preprocessing, and then sub-band signals are reconstructed at each level, which are further spatially filtered by the CSP algorithm for feature extraction. An orthogonal wavelet is employed to decompose the EEG signal into several sub-bands in Robinson et al. (2012), and then spatial regularized CSP is performed to extract features using the wavelet coefficients. The wavelet coefficients are taken as input features for probabilistic neural network in Gandhi et al. (2011). In Zhao et al. (2009), a Morlet wavelet transformation is applied for different frequency bins to each row of the composite covariance matrix and store the new rows in a larger matrix, and then optimizes the filters by maximizing the time-frequency ratio. However, the Morlet wavelet is a linear transformation, and some rows might be linearly dependent, leading to a non-full-rank matrix. In summary, there are mainly two potential problems of the existing wavelet-related algorithms: (a) the reconstructed signals are linearly dependent as the used wavelet is not orthogonal, or (b) the reconstructed signals from each level are used to construct features separately; therefore, optimality cannot be guaranteed.

In this paper, an orthogonal wavelet decomposition-based algorithm termed SEOWADE is proposed for single-trial EEG classification of motor imagery signals. The contributions are threefold:

(i) Unrelated sub-band components are obtained for subsequent feature extraction. Specifically, the preprocessed EEG signal is decomposed using orthogonal wavelets, and the decomposed signals are subsequently reconstructed at different levels/scales to yield sub-band components.

(ii) To enhance the discriminativity of the extracted features for classification, the proposed algorithm localize signals in spectral domain and spatial domain by implementing spatio-spectral filtering on reconstructed EEG signals from multiple levels.

(iii) As the choice of wavelet may have a significant impact on the quality of the results regarding the classifier, cross-validation is employed to select the most suitable wavelet function for signal processing. The SEOWADE algorithm is validated using three EEG data sets from past BCI competitions, and it is confirmed that the performance of the proposed algorithm is comparable to or even better than other state-of-the-art algorithms.

The remainder of this paper is structured as follows. Mathematical details of the SEOWADE algorithm is presented in section 2. The experimental results of SEOWADE on the three data sets from past BCI competitions are provided in section 3, where the details of the data sets, analysis pipelines, and classification results are provided. SEOWADE is benchmarked against those of contemporary algorithms: CSP, CSSP, CSSSP, FBCSP, and the algorithm termed shallow ConvNets proposed in Schirrmeister et al. (2017). Finally, section 4 concludes this paper. For ease of reference, the essential mathematical symbols used in this paper are shown in Table 1.

TABLE 1
www.frontiersin.org

Table 1. List of symbols in the paper.

2. Methodology

In this section, the orthogonal wavelet decomposition-based feature extraction method is described. We used orthogonal wavelets to construct independent sub-band components to localize signals spectro-temporally. Subsequently, spectral filtering via linearly transformed the sub-band components and spatial filtering is applied to enhance the discriminativity of EEG signals. In this way, the motor imagery-related patterns are generated by spatio-spectral filtering, and distinct power features defining the performed tasks are extracted. Finally, sparse Bayesian learning with Gaussian prior is applied to the extracted power features, with a RVM classifier obtained for classification.

2.1. Wavelet Decomposition and Reconstruction

Fourier transform has been widely applied to the analysis of non-stationary EEG signals. The advantage of wavelet analysis over short-time Fourier transform is that one can look at the signals at different scales or resolutions: a approximate level and a detailed level. Specifically, the wavelet is a smooth function that oscillates and quickly vanishes, which can localize well in both the frequency domain and the temporal domain (Vetterli and Herley, 1992). A wavelet family ψa,b(t) is a set of elementary functions, which are generated by dilations a's and translations b's of a unique admissible mother wavelet ψ(t): Ψa,b(t)=1|a|ψ(t-ba), where a,b ∈ ℝ, a ≠ 0, and t is the time point. The dilation parameter a, the translation parameter b determine the oscillatory frequency and length, the shifting position of the wavelet, respectively.

2.1.1. Discrete Orthogonal Wavelet Transform

In this subsection, we describe how the discrete orthogonal wavelet transform is implemented. In wavelet transform multi-resolution analysis, a finite impulse response filter pair [high-pass (HP) and low-pass (LP) filters] is specifically designed, the frequency responses of which separate the high-frequency and low-frequency components of the input signal. The HP filter coefficients are associated with the scaling function, and the LP filter is associated with the wavelet function. The outputs of the LP filters are called the approximation (A), and the outputs of the HP filters are called the details (D). In multi-resolution algorithms, any time series can be completely decomposed in terms of the approximation and detail components. Applications of discrete wavelet transformation produce a multi-resolution analysis of signals across time and frequency; however, the resulted wavelet components are redundant, and the computational complexity increases with additional decomposition layers. To address this issue and accommodate non-stationarity frequency analysis, discrete orthogonal wavelet transformation is applied for EEG signal analysis in this paper to obtain an improved tradeoff between temporal resolution and frequency resolution by varying the window length over frequencies.

The concept of multi-resolution and successive approximation of orthogonal wavelet transformation can be explained as follows. Let Ai, i ∈ ℤ be the space of band limited functions, and let Di, i ∈ ℤ be the orthogonal complement of Ai in Ai−1. These spaces are related as in the equations (Hazarika et al., 1997): AiAi−1, Ai−1 = AiDi. The orthogonal wavelet bases are constructed such that they span Ai and Di. For example, at level i = 1, the functions that approximate signals of space A0 in A1 represent a perfect half-band LP filter h1 and those in D1 represent a perfect half-band HP filter h0. At each decomposition level, orthogonal wavelet decomposition involves filtering with h1 and h0 followed by sub-sampling by 2. From a signal processing approach, the orthogonal wavelet transform can be defined as applying filters and samplers on discrete time sequences to perform a coarse half-resolution approximation of the original time sequences (Vetterli and Herley, 1992). The original signals can be reconstructed from these sub-band components using the reverse process, i.e., up-sampling by 2 and filtering using time reversed filter sequences h1 and h0.

2.1.2. Wavelet Filtering and Embedding

Given a T-length EEG signal x = [x(0), x(1), ⋯ , x(T − 1)], we show how the sub-band components of x by wavelet filtering and embedding are obtained as follows. The finite impulse response of the N-length filter h0 is in the form of h0 = [h0(0), h0(1), ⋯ , h0(N − 1)], while the N-length filter h1 satisfies the following expression:

h1(n)=(1)n·h0(N1n), n{0,1,,N1}.    (1)

The assumption is that the even-shifted version of the impulse response h0 [rows of the circulate matrix H0 given in (2)] forms an orthogonal set that spans the subspace D1:

H0=(          h0(N1) h0(N2) h0(N3)  h0(0) 0 0 0  0 0 h0(N1)  h0(2) h0(1) h0(0) 0          )    (2)

Similarly, the even-shifted version of h1 forms H1 which has the same structure as H0 and spans subspace A1. Note that the subspaces spanned by H0 and H1 are non-overlapped:

H0·H0*=I, H1·H1*=I, H0·H1*=H1·H0*=0,    (3)

where B* denotes the complex conjugate of matrix B, and I is the identity matrix.

The EEG signals x projected onto subspaces D1 and A1 followed by sub-sampling by 2, denoted by u1 and v1 can be represented by:

u1=H0x, v1=H1x,    (4)

where ⊤ denotes the transpose operator. Then, u1 consists of the detailed coefficients, and v1 consists of the approximation coefficients. The reconstruction of signal is achieved as:

u1=H0*H0x, v1=H1*H1x.    (5)

Figure 1A shows the processing steps of decomposition and reconstruction by orthogonal wavelet transformation at level 1.

FIGURE 1
www.frontiersin.org

Figure 1. (A) EEG signal decomposition and reconstruction by orthogonal wavelet. Here h0(n) and h1(n) present the finite impulse response of the half-band LP filter and half-band HP filter, and h0(n) and h1(n) present the reversed filters of h0(n) and h1(n), respectively. (B) EEG signal decomposition and reconstruction by orthogonal wavelet filtering to obtain the wavelet filtered and embedded EEG data, with the level of wavelet decomposition equals to three.

For multilevel orthogonal wavelet decomposition, the aforementioned process repeats to decompose vl, l ∈ {1, 2, ⋯ , L} at each level. The wavelet decomposition using coefficients from all L levels first gives L + 1 wavelet coefficients, and then retains coefficients from only one components for construction, until all the coefficients are used for construction, with L + 1 reconstructed signals obtained: vL, uL, uL1, ⋯ , u1. Specifically, Figure 1B demonstrates the process of obtaining the sub-band components with the impulse response of decomposition and reconstruction filters. The sub-band components via wavelet filtering and embedding can be presented by the reconstructed signals of the EEG signal x are denoted as a embedding matrix:

P=(u1u2 uLvL)(L+1)×T.    (6)

In Figure 2, an example of the 100th trial of the training data set from subject al implemented by DB7 is presented, with the level of wavelet decomposition is three.

FIGURE 2
www.frontiersin.org

Figure 2. The 100th trial of the training data set from subject al implemented by DB7 where the level of wavelet decomposition is three: (A) waveform in time domain after 7–40 Hz band-pass filtering; (B) amplitude waveform in frequency domain; (C) amplitude waveforms of the four sub-band components in the time domain; (D) amplitude waveforms in the frequency domain of the four sub-band components.

2.2. SEOWADE

Though the CSP algorithm is a highly successful method that has gained a surge of popularity and interest, the performance suffers from a non-discriminative brain rhythm that has an overlapping frequency range with the most discriminative brain rhythm. On the other hand, the frequency band on which the CSP operates is either selected manually or unspecifically set to a broad band filter, which is likely to degrade the performance by using an inappropriate frequency band. In the proposed algorithm, the sub-band components are weighted channel-wisely for spectral filtering because the discriminative frequency bands are channel-distinct, and simultaneously spatial filtering is implemented to enhance the discriminability of EEG signals. Note that the previous paper (Robinson et al., 2013) considered only the relative power distribution between each wavelet component, while our algorithm considers the total power of all the orthogonal wavelet components after spatio-spectral filtering, which is considered to capture the discriminative features of the EEG signal.

2.2.1. Combined Spatio-Spectral Filtering

The channel-wise spectral filters and the spatial filter can be re-parameterized by a single vector, which is presented below. Let X=[x1,x2,,xC]C×T denote a single-trial EEG signal, where C and T denote the number of channels and sample points, respectively. Each xc1×T, c ∈ {1, 2, ⋯ , C} denotes a T-length time sequence given by xc = [xc(0), xc(1), ⋯ , xc(T − 1)]. Suppose the L+1-length spectral filter for the sub-band components of the c-th channel is ac=[ac(0),ac(1),,ac(L)], c ∈ {1, 2, ⋯ , C}, and the C-length spatial filter is s = [s(1), s(2), ⋯ , s(C)]. The C channel-specific spectral filters ac's composes a matrix A ∈ ℝ(L+1) × C as: A = [a1, a2, ⋯ , aC]. Then, the combined operations of spatial filtering and channel-wise spectral filtering on X can be collectively expressed as:

Z=wX~,    (7)

where wRM×1 with M = (L + 1) × C denotes the combined spatio-spectral filter:

w=vec((sA))=[s(1)·a1,s(2)·a2,,s(C)·aC]   =[w1,w2,,wC],    (8)

where X~ denotes the embedding spatio-spectral matrix that appends the reconstructed wavelet signals of all the channels:

X~=(P1P2PC),    (9)

and PcR(L + 1) × T denotes the component matrix of the c-th channel, c ∈ {1, 2, ⋯ , C}, as described in (6).

According to (7), the implementation of spatial filtering and channel-wise spectral filtering on EEG signal X can be represented by linear combinations of X~ by w, where w is re-parameterized by a spatial filter and channel-wise spectral filters.

2.2.2. Filter Optimization

The optimization objective of our algorithm is to determine spatio-spectral filter w by solving the following maximizing variance ratio problem as in CSP (Blankertz et al., 2008):

maxwJ1(w)maxwwR~1wwR~2w,    (10)

where R~1 and R~2 are augmented covariance matrices under two states (labeled with 1 and 2) estimated by X~'s with trace normalization applied:

R~i=1NikX~kiX~kitr[X~kiX~ki], i{1,2},    (11)

where Ni is the number of training samples for the i-th state, X~ki is the k-th training sample for the i-th state, i ∈ {1, 2}.

To ameliorate the potential over-fitting problem, we perform a l2-norm regularization, which penalizes non-smooth combined spatio-spectral filters, as shown in the following Rayleigh quotient maximization problem:

w: =maxwJ2(w)maxwwR~1ww(R~2+ρ·I)w,    (12)

where ρ is the regularization parameter. Then, w can be decomposed into a spatial filter s and channel-wise spectral filter ac's, c ∈ {1, 2, ⋯ , C}. In this work, the c-th coefficient of s is s(c) = sgn(wc(1))·||wc||2, and ac = wc/sc.

Typically, one can retain only a small number of projection vectors, which contain most of the discriminative information for each state. We term the number of filters per class as m. Multiple spatio-spectral filters can be obtained by performing the generalization eigenvalue problem: R~1w=λR~2w. The resulting eigenvalues λ = J2(w) associated with each spatio-spectral filter is presumed to indicate the discriminability of the spatio-spectrally filtered EEG signals between the two class. By exchanging R~1 and R~2, m filters for another state can be obtained. In total, 2m spatio-spectral filters are retained for subsequent analysis.

2.2.3. Classification

Suppose W is a combination of the 2m filters: W=[w1,w2,,w2m-1,w2m]M×2m. Spatio-filtering can be implemented by projecting the wavelet filtered and embedded EEG data onto W as: Z=WX~2m×T. A classifier can be trained on the feature vectors obtained by normalizing and log-transforming the variances of Z as:

f=log(diag[ZZ]tr[ZZ])R2m×1,    (13)

where f denotes the feature vector. The log transformation serves to approximate the normal distribution of the feature vector, which is the basic assumption of some classifiers.

Typically, the prediction of class label ŷ is defined over the features f as follows, where u ∈ ℝ2m×1 and u0 denote the vector of parameters and the bias term, respectively, and Φ is the basis function:

y^(f;u)=uΦ(f)+u0.    (14)

The objective of a classifier is to estimate the values for those parameters. Sparse Bayesian learning is designed to manage the computational and statistical complexity in a principled way, which defines a hyper-parameterized prior over the parameter with the following form:

p(u|α)=k=12mp(uk|αk)k=12mαk12exp(αk2uk2).    (15)

This type of prior ultimately favor a sparse model that also fits the features well. The relevance vector machine (RVM) is a Bayesian sparse kernel technique for regression and classification, which is implemented as the classifier with Gaussian prior in this paper.

In summary, the involved analysis steps of SEOWADE can be described as follows, with the analysis flowchart presented in Figure 3:

• The EEG data are acquired and preprocessed;

• The preprocessed signals are decomposed by orthogonal wavelets at different scales, which are then reconstructed using wavelet reconstruction, with sub-band components are obtained;

• With the wavelet filtered and embedded data, the spatio-spectral filters are optimized by (12);

• We normalize and log-transform the discriminative features constructed by the spatio-spectrally filtered signals as in (13);

• The RVM classifier with a Gaussian prior is trained with the processed features, and the performance of the algorithm in terms of classification performance is determined.

FIGURE 3
www.frontiersin.org

Figure 3. Analysis pipeline of SEOWADE algorithm, with each EEG figure represent a single-trial EEG signal with dimension C × T. Note that the dimensions of a single-trial EEG signal after preprocessing, wavelet filtering and embedding, channel-wise spectral filtering and spatial filtering are C × T, M × T, C × T, and 2m × T, then the log-variance features with dimension 2m × 1 per trial are obtained for classification.

3. Data Analysis and Results

In this section, the binary classification performance of SEOWADE algorithm applied to three real EEG data sets is shown. In particular, the performance of SEOWADE is compared with that of the following algorithms: CSP, FBCSP (Ang et al., 2008), CSSP (Lemm et al., 2005), CSSSP (Dornhege et al., 2006), and shallow ConvNets proposed in Schirrmeister et al. (2017). Several intuitive examples are illustrated to demonstrate the relative advantage of SEOWADE over the compared algorithms. The MATLAB code of SEOWADE is available upon requests from the authors to allow for the reproducibility.

3.1. Data Description

In this work, three publicly available motor imagery data sets from past BCI competitions are used for performance evaluation. The data sets consist of EEG data from 17 subjects during motor imagination.

(1) Data Set A (Data Set IVa from BCI Competition III; Dornhege et al., 2004): This data set is recorded from five subjects (al, aa, av, aw, and ay), with a sampling rate of 100 Hz and 118 recorded channels. The experiment is a cue-based motor imagery paradigm, with the subjects performing right-hand and foot motor imagery tasks (labeled 1 and 2, respectively). A total of 280 trials are available for each subject, among which 224 (80%), 168 (60%), 84 (30%), 56 (20%), and 28 (10%) trials compose the training data set for al, aa, av, aw, and ay, and the remaining trials compose the testing data set.

(2) Data Set B (Data Set IIIa from BCI Competition III; Schlögl et al., 2005): This data set is recorded three subjects (K3, K6, L1), with a sampling rate of 250 Hz and 60 recorded channels. These subjects perform left-hand, right-hand, foot and tongue motor imagery tasks (labeled 1, 2, 3, and 4, respectively). Both the training and the testing data sets contain 45 trials per class for subject K3, and 30 trials for subjects K6 and L1. We split the data set of each subject to generate C42=6 data subsets, as our aim is to evaluate the binary classification performances of these algorithms. Therefore, a total of 18 data subsets are obtained, and for convenience, we use “subject's name-class labels” to denote the data subset.

(3) Data Set C (Data Set IIa from BCI Competition IV; Naeem et al., 2006): This data set comprises 22-channel EEG signals with a sampling rate of 250 Hz. Nine subjects (A01-A09) perform left-hand, right-hand, foot and tongue motor imagery tasks (labeled 1, 2, 3, and 4, respectively). Both the training and the testing data sets contain 72 trials per class for each subject. Akin to Data Set B, 6 × 9 = 54 data subsets are generated for binary classification.

A total of 5 + 18 + 54 = 77 data subsets are obtained for evaluation. The proposed SEOWADE algorithm and competing algorithms are run on each data subset for binary classification. The channel layouts for the three data sets are presented in Figure 4.

FIGURE 4
www.frontiersin.org

Figure 4. Electrode layouts for (A) Data Set A with 118 channels, (B) Data Set B with 60 channels, (C) Data Set C with 22 channels.

3.2. Analysis Pipeline

The analysis pipeline of the considered algorithms for binary classification is illustrated as follows.

(1) Preprocessing.

The following preprocessing steps are applied to the above mentioned 77 data subsets.

• All the channels are used for data analysis, i.e., channels covering both hemispheres are included in the analysis to obtain spatio-spectral information.

• A sixth-order Butterworth filter with a pass-band range of 7-40 Hz is used to filter the EEG signals to filter out the components unrelated to sensorimotor rhythms.

• The time window is set to 0.5-3.5 s for the three data sets, where 0 denotes the time of cue ends.

For FBCSP, the EEG signals are band-pass filtered into eight non-overlapping sub-band components (7–11, 11–15, 15–19, 19–23, 23–27, 27–31, 31–35, 35–40 Hz) using a sixth-order Butterworth filter.

(2) Feature Extraction.

For each data subset, the training data sets after preprocessing are fed into SEOWADE and other competing algorithms to optimize the spatial filters, spatio-temporal filters or spatio-spectral filters. The normalized and log-transformed variances of the spatially, spatio-temporally or spatio-spectrally filtered signals are then defined as the features for all the algorithms. Following previous studies, 3 features per state are constructed from the filters corresponding to the 3 largest generalized eigenvalues. That is, 2m = 6 features are obtained.

(3) Classification.

RVM with Gaussian prior is applied as the classifier for CSP, FBCSP, CSSP, CSSSP and SEOWADE. The classification performance on the test data set is measured in terms of mean classification accuracy and standard deviation.

3.3. Hyper-Parameters Settings

Several hyper-parameters need to be pre-determined for SEOWADE by 10-fold cross-validation during the analysis of CSSP, CSSSP, and SEOWADE. The hyper-parameters and the candidate sets for the algorithms are summarized in Table 2. Note that when L = 0 and ρ = 0 in SEOWADE, or τ = 0 in CSSP, these algorithms are reduced to CSP.

TABLE 2
www.frontiersin.org

Table 2. Candidate sets of the hyper-parameters for each algorithm.

N-fold cross-validation, sometimes called rotation estimation (Devijver and Kittler, 1982), is the statistical practice of partitioning a sample of data into N subsets such that the analysis is initially performed on N − 1 subsets, while the other subset is retained for subsequent use in conforming and validating the initial analysis. The initial subsets of data are called the training set, and the other subset is called the validation or testing set. The cross-validation process is then repeated N times (the folds), with each of the N subsets used exactly once as the validation data set. The results from the N folds can then be averaged to produce a single estimation. For each hyper-parameter, the one that achieves the highest average classification accuracy across the 10 repetitions is determined as its value.

3.4. Classification Performances

The classification performance of SEOWADE is compared with those of other compared algorithms in this subsection. The testing classification accuracies for all the algorithms compared with SEOWADE are shown in Figure 5, and the points above the diagonal line indicate that the classification accuracy of SEOWADE is higher than that of the compared algorithm on the x-axis. Moreover, the mean testing classification accuracies on the three data sets and on all the data sets are summarized in Table 3, in which the entries highlighted in boldface indicate the best performance among the competing algorithms. Repeated measures analysis of variance (ANOVA) shows that SEOWADE and the other five algorithms significantly differ in the classification performance [F(5,5 × 76) = 7.9617, and p = 3.76 × 10−7). Moreover, based on the Bonferroni-corrected Wilcoxon signed-rank tests, the significance of the results between the two compared algorithms at the 5, 0.5, and 0.1% levels is indicated by *, **, *** beneath the corresponding sub-figure of Figure 5. It can be confirmed that a considerable improvement in classification performance can be achieved by SEOWADE over the compared algorithms, which shows the effectiveness of SEOWADE.

FIGURE 5
www.frontiersin.org

Figure 5. Classification accuracies (%) of the algorithms (CSP, FBCSP, CSSP, CSSSP, and shallow ConvNets) compared with SEOWADE for the 77 binary data subsets, among which the five-pointed stars, circles and diamonds in each sub-figure denote the corresponding data subsets from Data Set A, Data Set B and Data Set C, respectively. Moreover, based on the Bonferronicorrected Wilcoxon signed-rank tests, the significance of the results between the two compared algorithms at the 5, 0.5, and 0.1% levels is indicated by *, **, *** beneath the corresponding sub-figure of this figure.

TABLE 3
www.frontiersin.org

Table 3. Mean classification accuracies and standard deviation (%) of CSP, FBCSP, CSSP, CSSSP, shallow ConvNets and SEOWADE on the three data sets (the numbers of subjects are M = 5, 18, 54, respectively), with RVM (with Gaussian prior) is implemented as the classifier.

3.5. Presentations of Extracted Filters

To determine the physiological significance of the algorithm and the source of informative features in the brain, we study the extracted spatial filters obtained from the proposed SEOWADE algorithm and other compared algorithms. The results are shown by three representative examples, namely, subjects ay, K6-23, and A09-24 from the three data sets, with the performed motor imagery tasks right-hand vs. foot, right-hand vs. foot, and right-hand vs. tongue, respectively. Figure 6 shows the topological scalp distribution maps of the spatial filters optimized by CSP, CSP, CSSP, CSSSP, and SEOWADE, with the weight maps related to the leading eigenvalue for each class.

FIGURE 6
www.frontiersin.org

Figure 6. Scalp weight maps of the optimized spatial filters (corresponding to the leading eigenvalues of each class) optimized by CSP, CSSP, CSSSP, and SEOWADE for the three representative subjects, and the absolute value r's of correlation coefficients between the spatial weight maps from different algorithms: (A) ay, with the performed motor imagery tasks are right-hand vs. foot; (B) K6-23, with the performed motor imagery tasks are right-hand vs. foot; and (C) A09-24, with the performed motor imagery tasks are right-hand vs. tongue.

Overall, the scalp weight distributions from SEOWADE are neurophysiologically relevant to the motor imagery tasks, with strong weights over the corresponding regions in the contralateral motor cortex. In addition, we compare the spatial filters of different algorithms by calculating the absolute value r's of correlation coefficients between their spatial maps, which is shown in the rightmost column of Figure 6. It can be seen that some spatial filter maps obtained by different algorithms are highly correlated, yet not identical.

3.6. Computational Costs

SEOWADE can be optimized efficiently. Suppose the number of the decomposition level for SEOWADE is L, and the number of channels is C, below we assess the computational complexity of each algorithm in the feature extraction step. The computational complexity is O(((N + 1)C)2) for SEOWADE, and O(C2), O((2C)2) for CSP and CSSP respectively. As for FBCSP, the complexity is approximately 8 times as large as that of CSP, since CSP is called for each of the 8 sub-band components. As for CSSSP, the spatial filters and channel-common temporal filters are optimized iteratively, and the complexity is O(C3) for spatial filter optimization and O(K1N) for temporal filter optimization in each iteration, respectively, where K1 is the number of iterations for temporal filter optimization with gradient descent. Let K2 denote the number of iterations to reach convergence for CSSSP. The total complexity for CSSSP is O(K2(C3+K1N)).

To highlight, in Table 4, we provide the runtimes for CSP, FBCSP, CSSP, CSSSP, shallow Convnet (abbreviated as sConvnet here), and SEOWADE (with DB7 implemented as the orthogonal wavelet) on the representative examples. sConv is coded in Pytorch, and other algorithms are run on a Windows PC with a 3.70-GHz Inter Core (TM) i7-8700K CPU and 16-GB RAM, in MATLAB© R2018b. It can be seen that the runtime of SEOWADE is less than those of CSSP and sConvnet, albeit more than those of CSP and CSSP.

TABLE 4
www.frontiersin.org

Table 4. Runtimes (millisecond) of CSP, FBCSP, CSSP, CSSSP, shallow Convnet (abbreviated as sConvnet here), and SEOWADE (with DB7 implemented as the orthogonal wavelet) applied to the representative examples ay, k6-23, and A09-24.

4. Discussion

Note that common average reference, removal of eyes movements and blinks are not performed for SEOWADE and other competing algorithms. The reason is that these preprocessing steps amount to spatial filtering on EEG, which is to a large extent redundant considering that the spatial filtering algorithms considered in the paper are already optimal for within-subject classification under their respective optimization criteria.

The high classification performance of SEOWADE on the three open motor imagery data sets demonstrates the effectiveness of SEOWADE. However, a relatively small-sized training set may lead to biased models with poor generalization performance. To address this issue, our work can be improved along the following two lines:

(i) Introducing inter-subject transfer learning. One idea is borrowing useful information from other subjects or other sessions to facilitate the current model learning process. In this case, how to utilize the information from other subjects to regularize the current model of the target subject will be the focus of our future work.

(ii) Generating adversarial examples. Alternatively, the number of training sample can be enlarged by generating adversarial examples, i.e., through perturbing the real sample via Generative Adversarial Networks (GAN). With more training samples, the obtained model is expected to be more stable and more robust with better generalization ability.

5. Conclusions

In this work, an SEOWADE algorithm based on orthogonal wavelet decomposition is proposed for motor imagery EEG signal classification. By SEOWADE, spatial filter and channel-specific spectral filters can be optimized simultaneously under a single optimization problem, with a l2-norm regularization term embedded to ameliorate over-fitting issue. Feature vectors are extracted via normalize and log-transform the spatio-spectral filtered signals, and then RVM with Gaussian prior is employed for classification. The proposed algorithm can effectively localize signals both in spatial and temporal domain, and thus can provide discriminative features for classification.

One motivation of BCI research is the application to motor imagery signal classification. Three motor imagery data sets from past BCI competitions are used to evaluate the performance of the proposed algorithm. Compared with the classic algorithms CSP, FBCSP, CSSP, CSSSP and shallow ConvNets, the testing classification performance of SEOWADE is significantly better at the 5% level. The extracted spatial patterns of SEOWADE are mainly located in the contralateral cortex, which is neurophysiologically relevant.

Data Availability Statement

The original contributions presented in the study are included in the article/supplementary material, further inquiries can be directed to the corresponding author/s.

Author Contributions

FQ, WWa, and WWu were responsible for the data analysis and contributed to the writing of the manuscript and interpretation of the data. XX and FW contributed to the data analysis and interpretation of the data. ZG, ZY, and YL contributed to the interpretation of the data. All authors contributed to the article and approved the submitted version.

Funding

This work was supported in part by the National Natural Science Foundation of China (No. 61906048, No. 61876063, and No. 61906019), Guangdong Basic and Applied Basic Research Foundation (No. 2020A1515010350, No. 2019A1515011870, and No. 2021A1515011853), the Foundation of Guangdong (No. 2020TSZK005), the Natural Science Foundation of Hainan (No. 2019RC165), and the Young Talents Science and Technology Innovation Project of Hainan Association for Science and Technology (No. QCXM202011).

Conflict of Interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

Publisher's Note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

Acknowledgments

The authors would like to thank Dr. Tianyou Yu and Dr. Zhenfu Wen for helpful comments, and are grateful to Klaus-Robert Müller, Benjamin Blankertz, Gabriel Curio, Clemens Brunner, Robert Leeb, Gert Pfurscheller, and Alois Schlögl for providing the EEG data sets used in this paper.

References

Ang, K. K., Chin, Z. Y., Zhang, H., and Guan, C. (2008). “Filter bank common spatial pattern (FBCSP) in brain-computer interface,” in IJCNN (Hong Kong), 2390–2397.

Google Scholar

Bhattacharyya, A., and Pachori, R.-B. (2017). A multivariate approach for patient-specific EEG seizure detection using empirical wavelet transform. IEEE Trans. Biomed. Eng. 64, 2003–2015. doi: 10.1109/TBME.2017.2650259

PubMed Abstract | CrossRef Full Text | Google Scholar

Blankertz, B., Tomioka, R., Lemm, S., Kawanabe, M., and Müller, K.-R. (2008). Optimizing spatial filters for robust EEG single-trial analysis. IEEE Signal Process. Mag. 25, 41–56. doi: 10.1109/MSP.2008.4408441

CrossRef Full Text | Google Scholar

Devijver, P. A., and Kittler, J. (1982). Pattern Recognition: A Statistical Approach. London: Prentice-Hall.

Dornhege, G., Blankertz, B., Curio, G., and Müller, K.-R. (2004). Boosting bit rates in noninvasive EEG single-trial classifications by feature combination and multiclass paradigms. IEEE Trans. Biomed. Eng. 51, 993–1002. doi: 10.1109/TBME.2004.827088

PubMed Abstract | CrossRef Full Text | Google Scholar

Dornhege, G., Blankertz, B., Krauledat, M., Losch, F., Curio, G., and Müller, K.-R. (2006). Combined optimization of spatial and temporal filters for improving brain-computer interfacing. IEEE Trans. Biomed. Eng. 53, 2274–2281. doi: 10.1109/TBME.2006.883649

PubMed Abstract | CrossRef Full Text | Google Scholar

Gandhi, T., Panigrahi, B. K., and Anand, S. (2011). A comparative study of wavelet families for EEG signal classification. Neurocomputing 74, 3051–3057. doi: 10.1016/j.neucom.2011.04.029

PubMed Abstract | CrossRef Full Text | Google Scholar

Hazarika, N., Chen, J. Z., Tsoi, A. C., and Sergejew, A. (1997). Classification of EEG signals using the wavelet transform. Signal Process. 59, 89–92. doi: 10.1016/S0165-1684(97)00038-8

CrossRef Full Text

Higashi, H., and Tanaka, T. (2013). Simultaneous design of FIR filter banks and spatial patterns for EEG signal classification. IEEE Trans. Biomed. Eng. 60, 1100–1110. doi: 10.1109/TBME.2012.2215960

PubMed Abstract | CrossRef Full Text | Google Scholar

Hlawatsch, F., and Boudreaux-Bartels, G. F. (1992). Linear and quadratic time-frequency signal representations. IEEE Signal Process. Mag. 8, 21–67. doi: 10.1109/79.127284

CrossRef Full Text | Google Scholar

Kavitha, P. T., Guan, C., Lau, C. T., and Vinod, A. P. (2008). “An adaptive filter bank for MI based brain-computer interface,” in 30th Annual International IEEE EMBS Conference (British Columbia), 1104–1107.

Google Scholar

Lawhern, V.-J., Solon, A.-J., Waytowich, N.-R., Gordon, S.-M., Hung, C.-P., and Lance, B.-J. (2016). EEGNet: a compact convolutional network for EEG-based brain-computer interfaces. J. Neural Eng. 15, 056013.1–056013.17. doi: 10.1088/1741-2552/aace8c

PubMed Abstract | CrossRef Full Text | Google Scholar

Lemm, S., Blankertz, B., Curio, G., and Müller, K.-R. (2005). Spatio-spectral filters for improved classification of single trial EEG. IEEE Trans. Biomed. Eng. 52, 1541–1548. doi: 10.1109/TBME.2005.851521

PubMed Abstract | CrossRef Full Text | Google Scholar

Lemm, S., Blankertz, B., Dickhaus, T., and Müller, K.-R. (2011). Introduction to machine learning for brain imaging. Neuroimage 56, 387–399. doi: 10.1016/j.neuroimage.2010.11.004

PubMed Abstract | CrossRef Full Text | Google Scholar

Mohammadi, Z., Frounchi, J., and Amiri, M. (2017). Wavelet-based emotion recognition system using EEG signal. Neural Comput. Appl. 28, 1985–1990. doi: 10.1007/s00521-015-2149-8

CrossRef Full Text | Google Scholar

Mousavi, E. A., Maller, J. J., Fitzgerald, P. B., and Lithgow, B. J. (2011). Wavelet common spatial pattern in asynchronous offline brain computer interfaces. Biomed. Signal Process. Control 6, 121–128. doi: 10.1016/j.bspc.2010.08.003

CrossRef Full Text | Google Scholar

Naeem, M., Brunner, C., Leeb, R., Graimann, B., and Pfurtscheller, G. (2006). Seperability of four-class motor imagery data using independent components analysis. J. Neural Eng. 3, 208–216. doi: 10.1088/1741-2560/3/3/003

PubMed Abstract | CrossRef Full Text | Google Scholar

Novi, Q., Guan, C., Dat, T. H., and Xue, P. (2007). “Sub-band common spatial pattern (SBCSP) for brain-computer interface,” in Proceedings of the 3rd International IEEE/EMBS Conference on Neural Engineering (Hawaii), 204–207. doi: 10.1109/CNE.2007.369647

PubMed Abstract | CrossRef Full Text | Google Scholar

Qi, F., Li, Y., and Wu, W. (2015). RSTFC: a novel algorithm for spatio-temporal filtering and classification of single-trial EEG. IEEE Trans. Neural Netw. Learn. Syst. 26, 3070–3082. doi: 10.1109/TNNLS.2015.2402694

PubMed Abstract | CrossRef Full Text | Google Scholar

Qin, L., and He, B. (2005). A wavelet-based time-frequency analysis approach for classification of motor imagery for brain-computer interface applications. J. Neural Eng. 2, 65–72. doi: 10.1088/1741-2560/2/4/001

PubMed Abstract | CrossRef Full Text | Google Scholar

Robinson, N., Vinod, A.-P., Guan, C., Ang, K.-K., and Peng, T.-K. (2012). “A modified Wavelet-Common Spatial Pattern method for decoding hand movement directions in brain computer interfaces,” in International Joint Conference on Neural Networks (Brisbane, QLD), 1–5.

Google Scholar

Robinson, N., Vinod, A. P., Ang, K. K., Tee, K. P., and Guan, C. (2013). EEG-based classification of fast and slow hand movements using wavelet-CSP algorithm. IEEE Trans. Biomed. Eng. 60, 2123–2132. doi: 10.1109/TBME.2013.2248153

PubMed Abstract | CrossRef Full Text | Google Scholar

Robinson, N., Vinod, A. P., Guan, C., Ang, K. K., and Peng, T. K. (2011). “A wavelet-CSP method to classify hand movement directions in EEG based BCI system,” in 2011 8th International Conference on Information, Communications and Signal Processing (Singapore), 1–5. doi: 10.1109/ICICS.2011.6174210

CrossRef Full Text | Google Scholar

Schirrmeister, R.-T., Springenberg, J.-T., Fiederer, L., Glasstetter, M., Eggensperger, K., Tangermann, M., et al. (2017). Deep learning with convolutional neural networks for EEG decoding and visualization. Hum. Brain Mapp. 38, 5391–5420. doi: 10.1002/hbm.23730

PubMed Abstract | CrossRef Full Text | Google Scholar

Schlögl, A., Lee, F., Bischof, H., and Pfurtscheller, G. (2005). Characterization of four-class motor imagery EEG data for the BCI-Competition 2005. J. Neural Eng. 2, L14–L22. doi: 10.1088/1741-2560/2/4/L02

PubMed Abstract | CrossRef Full Text | Google Scholar

Subasi, A., Yilmaz, M., and Ozcalik, H. R. (2006). Classification of EMG signals using wavelet neural network. J. Neuroence Methods 156, 360–367. doi: 10.1016/j.jneumeth.2006.03.004

PubMed Abstract | CrossRef Full Text | Google Scholar

Suk, H., and Lee, S. (2013). A novel Bayesian framework for discriminative feature extraction in brain-computer interfaces. IEEE Trans. Pattern Anal. Mach. Intell. 35, 286–299. doi: 10.1109/TPAMI.2012.69

PubMed Abstract | CrossRef Full Text | Google Scholar

Tomioka, R., Dornhege, G., Nolte, G., Blankertz, B., Aihara, K., and Müller, K.-R. (2006). Spectrally Weighted Common Spatial Pattern Algorithm for Single Trial EEG Classification. Technical report. Deptment of Mathematical Engineering, The University of Tokyo. doi: 10.1007/11861898_42

CrossRef Full Text | Google Scholar

Vetterli, M., and Herley, C. (1992). Wavelets and filter banks: theory and design. IEEE Trans. Signal Process. 40, 2207–2232. doi: 10.1109/78.157221

CrossRef Full Text | Google Scholar

Wu, W., Gao, X., Hong, B., and Gao, S. (2008). Classifying single-trial EEG during motor imagery by iterative spatio-spectral patterns learning (ISSPL). IEEE Trans. Biomed. Eng. 55, 1733–1743. doi: 10.1109/TBME.2008.919125

PubMed Abstract | CrossRef Full Text | Google Scholar

Yang, B. H., Yan, G. Z., Yan, R. G., and Wu, T. (2007). Adaptive subject-based feature extraction in brain-computer interfaces using wavelet packet best basis decomposition. Med. Eng. Phys. 29, 48–53. doi: 10.1016/j.medengphy.2006.01.009

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhang, H., Yang, C. Z., Ang, K. K., Guan, C., and Wang, C. (2011). Optimum spatio-spectral filtering network for brain-computer interface. IEEE Trans. Neural Netw. 22, 52–63. doi: 10.1109/TNN.2010.2084099

PubMed Abstract | CrossRef Full Text | Google Scholar

Zhang, L., Liu, G., and Wu, Y. (2010). “Wavelet and common spatial pattern for EEG signal feature extraction and classification,” in 2010 International Conference on Computer, Mechatronics, Control and Electronic Engineering (CMCE) (Changchun), 243–246.

Google Scholar

Zhang, Y., Liu, B. X J, and Huang, D. (2017). Classification of EEG signals based on autoregressive model and wavelet packet decomposition. Neural Process. Lett. 45, 365–378. doi: 10.1007/s11063-016-9530-1

CrossRef Full Text | Google Scholar

Zhao, Q., Zhang, L., and Cichocki, A. (2009). EEG-based asynchronous BCI control of a car in 3D virtual reality environments. Chin. Sci. Bull. 54, 78–87. doi: 10.1007/s11434-008-0547-3

CrossRef Full Text | Google Scholar

Keywords: brain-computer interface, orthogonal wavelet decomposition, spatio-spectral filtering, l2-norm regularization, relevance vector machine, sparse Bayesian learning

Citation: Qi F, Wang W, Xie X, Gu Z, Yu ZL, Wang F, Li Y and Wu W (2021) Single-Trial EEG Classification via Orthogonal Wavelet Decomposition-Based Feature Extraction. Front. Neurosci. 15:715855. doi: 10.3389/fnins.2021.715855

Received: 27 May 2021; Accepted: 13 September 2021;
Published: 13 October 2021.

Edited by:

Dakai Jin, PAII Inc., United States

Reviewed by:

Elvira Pirondini, University of Pittsburgh, United States
Veeky Baths, Birla Institute of Technology and Science, India
Fengze Liu, Johns Hopkins University, United States

Copyright © 2021 Qi, Wang, Xie, Gu, Yu, Wang, Li and Wu. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

*Correspondence: Wei Wu, YXV3ZWl3dSYjeDAwMDQwO3NjdXQuZWR1LmNu

These authors have contributed equally to this work

Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.