What this is
- This research proposes a -ConvLSTM model utilizing an attention mechanism for classifying EEG signals.
- It addresses challenges in EEG signal classification, including low signal-to-noise ratios and small sample sizes.
- The model combines with a bidirectional ConvLSTM structure to enhance feature extraction and classification accuracy.
Essence
- The -ConvLSTM model achieves average classification accuracies of 73.7% and 92.6% on two datasets, demonstrating its effectiveness in decoding EEG signals.
Key takeaways
- The proposed model effectively extracts significant features from EEG signals, improving classification accuracy. By employing a sliding window for data augmentation, the model increases the training sample size, addressing the challenge of limited labeled data.
- The introduction of reduces the number of parameters while enhancing the extraction of spatial features. The bidirectional ConvLSTM structure captures temporal characteristics, improving the model's robustness against individual variability.
Caveats
- The model's performance may still be influenced by the inherent noise in EEG signals and the variability across different subjects. Further research is needed to validate its effectiveness in more diverse and larger datasets.
Definitions
- Motor Imagery (MI): The mental simulation of movement without actual physical execution, producing specific EEG signal patterns.
- Depthwise Separable Convolution (DSC): A convolutional operation that separates spatial and channel-wise operations to reduce parameters and computational cost.
- Long Short-Term Memory (LSTM): A type of recurrent neural network architecture designed to model sequence data, capable of learning long-term dependencies.
AI simplified
1. Introduction
Brain-computer interface (BCI) technology is a frontier research direction of multidisciplinary integration. It establishes a direct communication pathway between the human brain and general-purpose/specialized computing equipment by means of electroencephalogram (EEG) signals acquisition and decoding technology so as to interact with the outside world [1]. BCI is primarily used in medical treatment for severely ill patients with neurological or muscular disabilities. With the continuous progress of science and technology, the application of the brain-computer interface has been extended to many industries such as education, health care, and entertainment. Brain-computer interface technology is centered on human thinking and ideas, analyzes EEG signals through signal processing algorithm to obtain corresponding control instructions, and finally realizes the control of terminal devices. The control signals of BCI can be a variety of brain functional signals. Among them, the BCI system based on EEG is considered to be the most practical BCI system in the future due to its advantages in ease of use, safety, and system cost. The principle of motor imagery (MI) BCI system is an event-related desynchronization/synchronization phenomenon of EEG signals generated during motor imagery. The generation of Motor Imagery EEG (MI-EEG) signals does not depend on external stimuli and only requires the subjects to imagine the corresponding action. The system design is relatively simple, the experiment is relatively convenient, and the signal transmission rate is fast. It has become one of the main research directions in the field of brain-computer interface. It is the focus and difficulty of relevant brain-computer interface technology to realize the accurate recognition of MI-EEG for outputting control instructions or providing feedback to users.
It is extremely challenging to extract stable and valuable features from motor imagery EEG signals. The traditional EEG feature extraction algorithms mainly include time domain method, frequency domain method, time frequency method, and spatial filtering method. Time-domain method includes power spectrum analysis and phase-locked value [2], frequency-domain method includes autoregression model [3] and digital filter [4], time-frequency method includes wavelet transform [5] and wavelet packet transform [6], and spatial filtering method is mainly common spatial pattern (CSP) algorithm. In current studies, CSP has been proved to be the most effective and widely used among traditional methods. Many studies are based on the extension of the CSP method to improve the classification accuracy of MI tasks. Reference [7] first applied CSP to the classification of motor imagery EEG signals. However, the features extracted using CSP lack frequency domain information. Reference [8] proposed filter bank common spatial pattern (FBCSP), which incorporates frequency domain information by the specific filter for frequency domain selection followed by CSP feature extraction.
EEG signal classification is the most critical step for BCI systems. The traditional classification algorithms mainly include linear discriminant analysis (LDA), support vector machine (SVM), common spatial pattern (CSP), and artificial neural network (ANN). In the emerging deep learning models, feature extraction and feature classification are merged to perform simultaneously in a purely data-driven manner in a single framework, which greatly alleviates the need for manual feature extraction. In recent years, more and more researchers have started to explore the application of deep learning in EEG signal recognition, and many achievements have been achieved. Reference [9] designed and trained various deep convolutional neural networks (CNNs) with different architectures to decode task-related information from the raw EEG signals. By improving the structure of CNNs, at least as good performance as Filter Bank Common Spatial Pattern (FBCSP) algorithm widely used was achieved. In Reference [10], a 4-layer compact convolutional neural network, EEGNet, was designed to classify EEG signals from four different BCI paradigms (P300, error-related negative (ERN), motion-related cortical potential (MRCP), and sensori motor rhythm (SMR)). The experimental results showed that EEGNet achieved good scaling across paradigms with sufficient robustness. This study is of great value for EEG signal recognition cross-tasks and cross-subjects. Reference [11] proposed a deep learning method based on a restricted boltzmann machine (RBM) for motor imagery EEG signals classification. Three RBM models are trained by obtaining the frequency domain representation of EEG signals through fast fourier transform (FFT) and wavelet packet decomposition (WPD). These RBMs are then stacked with additional output layers to form a four-layer neural network called frequency depth belief network (FDBN). In Reference [12, 13], long short-term memory (LSTM) was used to process MI-EEG signals, and the accuracy was 77.30% and 82.75%, respectively. Reference [14] proposed an MI-EEG signal recognition algorithm combining bidirectional LSTM (BiLSTM) and graph convolutional neural network (GCN), using BiLSTM to extract relevant features from original EEG signals, and the connected GCN by cooperating with the topological structure of features improves decoding performance. Reference [15] proposed a classification framework for MI-EEG signals that combines a convolutional neural network (CNN) architecture with a variational autoencoder (VAE) for classification. Reference [16] introduced two deep learning-based frameworks with novel spatiotemporal preserving representations of raw EEG streams to precisely identify human intentions. The two frameworks consist of both convolutional and recurrent neural networks, effectively exploring the preserved spatial and temporal information in either a cascade or a parallel manner. Reference [17] proposed a subject-independent framework based on deep convolutional neural networks. In this framework, it formulated the discriminative feature representation as a combination of the spectral-spatial input embedding the diversity of the EEG signals, as well as a feature representation learned from the CNN through a fusion technique that integrates a variety of discriminative brain signal patterns. Reference [18] proposed a novel multimodal machine learning (ML) based approach is proposed to integrate EEG engineered features for automatic classification of brain states. Reference [19] proposed a novel method based on conditional empirical mode decomposition (CEMD) and a one-dimensional multiscale convolutional neural network (1DMSCNN) is proposed to recognize motor imagery EEG signals. This method can provide a stimulus to the development of human-robot interaction.
In summary, deep learning-based methods have unique advantages over traditional methods. Deep learning can automatically extract features applied to classification from original signals and improve classification accuracy. Feature extraction and feature classification are combined in the one frame to avoid information loss caused by different objective functions of the two stages when feature extraction and feature classification are performed step by step. On the other hand, deep learning expands the scope of applications across paradigms and subjects. However, its application still faces many challenges, and the existing EEG signal recognition algorithms still cannot meet the needs of complex application scenarios of brain-computer interfaces. In addition, it makes accurate recognition more difficult due to the low signal-to-noise ratio and non-stationary and nonlinear characteristics of EEG signal. The limitation of the deep learning algorithm itself is also a bottleneck restricting the development of brain-computer interfaces. The current deep learning-based MI-BCI research mainly faces the following challenges:The EEG signal is very weak, with the amplitude usually less than 200 μV. Motor imagery EEG signal is the spontaneous random signal with nonstationary randomness in time, and slight changes in the external environment or body state may cause unpredictable effects [20]. In addition, the signal is easily mixed with noise signals from the external environment and from such as electrocardiogram (ECG), respiration, electrooculogram (EOG), and EMG artifacts caused by human activities, resulting in a very low signal-to-noise ratio of EEG.There is still a lack of large samples of labeled EEG datasets, which makes the effectiveness of existing deep learning algorithms not fully reflected. The effectiveness of deep learning greatly depends on high-quality labeled data. However, in existing studies, especially in clinical studies, well-labeled and accurate EEG data are still scarce, and the sample size is small. Moreover, it is unrealistic to obtain data sets that can train complex neural network models from a single subject.There is too much individual variability across subjects. Many deep learning algorithms that perform well on a single dataset do not perform as well as expected on other datasets or in other online systems. So far, no one or several algorithms have performed well on all datasets, subjects, or data from the same subject at different time periods.
Based on this, a DSC-ConvLSTM model based on attention mechanism is proposed to extract the effective features of motor imagery EEG signals for classification. To address challenge i, for EEG data with a low signal-to-noise ratio, we designed the model architecture according to the characteristics of multichannel motor imagery EEG signals so that it could effectively extract significant features of EEG signals and improve the accuracy of EEG classification. To address challenge ii, for the small sample size of the well-labeled EEG dataset, we use sliding windows to enhance data before the training network, which is equivalent to increasing the training sample size. To address challenge iii, for the problem of cross-subject variation, we take this into account in the detailed design of the model. In terms of data samples, the data augmentation using the sliding window not only increases the sample size but also makes the network no longer extract global features of the whole trial, making the model diminish learning the individualized characteristics of every subjects' EEG data. Also, in terms of model design, as EEG signals have complex spatio-temporal characteristics, we use a convolutional neural network (CNN) to extract spatial frequency information of EEG, in which depthwise separable convolution (DSC) structure is introduced. On the one hand, the general characteristic of DSC is that it can reduce the number of parameters. On the other hand, for EEG decoding, DSC uses depthwise convolution to extract individual feature mappings in time and pointwise convolution to optimize the combination of feature mappings, and different feature maps can represent the information data of different time scales of EEG signals, which can explicitly decouple the relationships within and across feature mappings and alleviate the problem of individual differences to a certain extent. In view of the fact that EEG data is a kind of temporal data, in addition to using CNN to capture spatial features of EEG signals sequences, we will also use bidirectional Convolution Long Short Time Memory (BiConvLSTM) based on the attention mechanism capturing the temporal evolution of EEG signals sequence. All architectures of the model are designed for the characteristics of multichannel motor imagery EEG signals. In order to verify the validity of the proposed model, we compare it with other novel methods on the decoding accuracies, such as filter bank common spatial patterns (FBCSP) in Reference [8], Deep ConvNets, and Shallow ConvNets in Reference [9], EEGNet-4,2, EEGNet-8,2 in Reference [10]. By analyzing the decoding performance of different models in both the BCI Competition IV Dataset 2a and the High Gamma Dataset. The experimental results show that the DSC-ConvLSTM model based on the attention mechanism proposed in this paper can effectively decode MI-EEG signals with certain stability and robustness, which can effectively mitigate the intertrial and intersubject differences.
The paper is structured as follows:is the introduction,introduces the datasets we use and the details of our proposed model,shows the experimental results and the discussion, andis the conclusion. Section 1 Section 2 Section 3 Section 4
2. Materials and Methods
2.1. Data Description and Processing
The motor imagery BCI (MI-BCI) system is mainly input by EEG signals caused by the rhythmic changes of the cerebral cortex stimulated by motor imagery. The physiological principle of motor imagery EEG is based on event-related synchronization (ERS)/event-related desynchronization (ERD) [21]. When people engage in some specific exercise or mental activity, the coherent area of the cerebral cortex is activated, and thus the metabolism and blood flow in this area are accelerated, resulting in a decrease in the amplitude of the frequency spectrum oscillation of brain waves in some frequency bands, corresponding to a decline in the frequency band energy. This physiological phenomenon is known as ERD. In another situation, when the brain is in a quiet or inert state, specific frequency bands show a significant increase in amplitude, which corresponds to an increase in band energy known as ERS. The ERS/ERD phenomenon generally occurs in the frequency range of 8–12 Hz and 18–26 Hz [22], where most of the sensory-motor cortex potential generated by adults is located. This is the physiological basis for the classification of motor imagery EEG signals. The datasets used in this experiment are BCI Competition IV Dataset 2a [23] and High Gamma Dataset [9].
2.1.1. Dataset 1: BCI Competition IV Dataset 2a
The BCI competition IV Dataset 2a (BCIC IV 2a) consisted of EEG data from nine subjects, and its experimental paradigm consisted of four different motor imagery tasks: left hand, right hand, feet, and tongue. Each subject completed two sessions on different days, and each session consisted of six runs with a short break between each run. Each run consisted of 48 trials (12 for each task). Thus, there are a total of 288 trials per session. EEG signals were recorded by 22 Ag/Al electrodes at a sampling frequency of 250 Hz, bandpass filtered between 0.5 Hz and 100 Hz, and line noise suppressed by a 50 Hz trap filter.
2.1.2. Dataset 2: High Gamma Dataset
Compared with Dataset 1, the High Gamma Dataset (HGD) is a larger dataset with a total of 20 subjects. A trial per subject was divided into 13 runs, for a total of approximately 1000 (992 ± 135.3) trials. The four types of motor imagery tasks were left hand, right hand, feet, and rest (the rest had no specific movement but had the same visual cues as other tasks in the paradigm design). Except for the last two runs, the training set consisted of approximately 880 trials and the test set consisted of approximately 160 trials. The data were recorded by 128 electrodes at a sampling frequency of 5 kHz. And the noninvasive detection of high-frequency EEG components in the sensorimotor cortex due to ERS/ERD phenomenon was optimized [24]. In the experiments, the HGD dataset was resampled to 250 Hz to ensure that it was identical to the BCIC IV 2a dataset so that the network parameter settings could be adapted to both datasets.
Due to the small sample size of the EEG dataset, we use the sliding window for data augmentation prior to network training instead of using the entire trial directly as input to the network. This allows the trials to be divided into multiple smaller trials, increasing training samples. The schematic diagram of data augmentation is shown in Figure 1. In this study, the time range of samples we took was from −0.5 s before the start of each trial to 4s after the start of each trial which achieved the best results we had ever experimented. For each sliding window, the samples within each sliding window segment are used as a new example. For example, given an original trial Xi=RC·T of subject i, where C denotes the number of channels and T denotes the time step. The time length of the sliding window is T′, then after multiple window slides, each trial will get multiple new examples Xi′=RC·T′. The time step of each sliding window is 2 s, the start point of the first sliding window is 0.5 s before the start of each trial, and the start point of the last sliding window is 2 s after the start of each trial, that is, the end point is 4 s after the start of each trial. Therefore, the whole range is from 0.5 s before the start of each trial to 4 s after the start, which is the same as our original setting. We believe that it has a higher recognition accuracy when windows are in the vicinity of 0.5 s to 2.5 s after the start of each trial. Because the subjects are most focused, just starting to concentrate, and do not appear to have motor imagery fatigue. According to the above Settings, the starting point of the sliding window ranges from −0.5 s to 2 s, a total of 2.5 s. The signal sampling frequency was 250 Hz, and 625 new samples were generated for each trial. That is, the number of our training samples increased by 625 times. These new examples are trained separately by the network, and the average of the predictions will be used as the final prediction of the overall trial, and the loss of the overall trial is calculated by comparing with the target value. This data preprocessing approach not only increases the number of training samples but also make the network no longer extract global features and individualized features of every trial.
2.2. Overall Network Structure
Because EEG signals are nonstationary and stochastic, and the sample size is small but contains abundant information. The signal acquisition process is susceptible to environmental and individual variability, so it is difficult to effectively guarantee the robustness of the classification system. In this paper, a DSC-ConvLSTM network structure is proposed according to the characteristics of EEG signals, as Figure 2 shows. The final convolutional classification layer further fuses the feature information extracted by DSC-ConvLSTM and maps the extracted feature matrix into the sample marker space to achieve end-to-end classification and recognition of EEG signals.
EEG signals have complex temporal and spatial characteristics, so the overall model is based on the basic structure of CNN-LSTM. The deep convolutional neural network can learn the spatial frequency information of EEG signals by continuous training. In addition, considering the multichannel characteristics of EEG signals, the deep separable convolutional structure (DSC) is used in the CNN module. The CNN module ignores the information of the temporal domain, while the LSTM module can capture the temporal evolution of EEG signal sequences. The gate mechanism of LSTM enables the trained model to adapt to long-term dependence, which is conducive to the serialization feature extraction of EEG signals. In addition, considering the contextual temporal characteristics and underlying spatial features of EEG signals, we improve the single LSTM unit and propose a new ConvLSTM structure to classify EEG signals. ConvLSTM changes the existing structure of LSTM unit, where the state-to-state conversion of classical LSTM adopts a fully connected form, while ConvLSTM adopts a convolutional form. In addition, the attention mechanism is introduced to select the information which is more critical to the current task objective from the numerous information to improve the decoding accuracy.
2.3. Depthwise Separable Convolution
The DSC module network structure and parameter settings are shown in Table 1, which consists of 4 blocks. The first block is special, designed to better process EEG signals as input. The other three blocks are all standard convolution-max-pooling.
In Block1, the convolution module is divided into two layers: time-domain convolution and spatial filtering. The EEG signal input into the network is a two-dimensional array, whose width is the number of time steps and its length is the number of electrode channels. And unlike images that have three RGB color channels, each electrode of EEG signals is one channel. In the first layer time-domain convolution is designed as F1 2D convolution filters with kernel size of (1, 64), and this layer outputs F1 feature maps containing EEG signals with different bandpass frequencies. Then in the second layer spatial filter uses a Depthwise Convolution, which is designed as D∗F1 2D Convolution filters with the kernel size (C, 1) to extract the spatial domain features of EEG signals in the electrode channel dimension. The advantage of Depthwise Convolution is that it reduces the number of network training parameters. Different from the conventional convolution operation, one convolution kernel of Depthwise Convolution is responsible for one channel, and one channel is only convolved by one convolution kernel, so there are as many filters as there are channels. After the new channel feature maps are obtained, the standard 1∗1 cross-channel convolution operation is then performed on this batch of new channel feature maps. Depthwise Convolution operation reduces the number of parameters compared to a standard convolution operation and has better results because it learns each channel of the EEG signal (each channel corresponds to a different filter instead of all channels corresponding to the same filter). The depth parameter D controls the number of spatial filters that need to be learned for each feature mapping, and here we set D to 1. This two-layer convolution pattern is somewhat inspired by the Filter-Bank Common Spatial Pattern (FBCSP) algorithm [25].
In Block2-Block4, there are basically similar modules. In Block2, for example, the separable convolution is used, whose core idea is to decompose a complete convolution operation into two steps: Depthwise Convolution and Pointwise Convolution [26]. Here, the separable convolution consists of F2 Depthwise Convolution with kernel size (1, 16) and D∗F2 Pointwise Convolution with kernel size (1, 1). Its main advantages: One is to reduce the number of network parameters; the other is to decouple the relationship within and across feature mappings explicitly. Depthwise Convolution is used to summarize each feature map, respectively, and finally, Pointwise Convolution is used to optimize and combine the output. This is very suitable for EEG signal classification characteristics, where different feature mappings can represent data from different time scales of EEG signals.
The activation function is exponential linear unit (ELU) [27], which is more sensitive to feature types to improve the classification performance compared to Square and ReLU that generate average power. Before the ELU unit, batch normalization (BN) is applied on each dimension of feature mapping and the entire model is regularized using dropout [28], where the dorpout rate is set to 0.5 in the validation of the dataset between individual subjects to prevent overfitting of the network on small data samples. At the same time, the linear transformation is also forced to separate into a combination of time-domain convolution and spatial filtering, implicitly regularizing the entire model. The pooling method is chosen as max-pooling, because for the most common image applications, max-pooling preserves the texture features, and mean-pooling preserves the overall data features. At the same time, different features of the EEG signal may have different manifestations in high-frequency and low-frequency components. Compared to mean-pooling, max-pooling extracts the features with the largest and strongest responses and inputs them to the next module. Only the last block uses mean-pooling to ensure the integrity of EEG data information. Softmax is used for the final classification layer.
2.4. ConvLSTM with Attention
LSTM is an improvement of recurrent neural network (RNN), which retains the ability of RNN to model sequence data accurately. In order to solve the problem of gradient disappearance or explosion during the training process of RNN, LSTM added memory unit C, input gate I, forgetting gate F, and output gate O, the input gate obtains new input from the outside and processes new data. The forgetting gate determines when to selectively forget a particular output result, so as to select the best time lag for the input sequence. The output gate computes all the results and generates the output for the LSTM unit. The main difference between ConvLSTM and LSTM is that standard LSTM adopts full connection form in input-to-state and state-to-state connection, and decoding does not take into account spatial structure characteristics, while ConvLSTM adopts the form of convolution. The difference in details between the two is shown in Figure 3. For the ConvLSTM structure, all the inputs input xt, cell output Ct, hidden layer state Ht, and ft, it, ot are 3D Tensor, where two dimensions are spatial features. However, the input dimension of the traditional LSTM module is one-dimensional, which can be regarded as the two dimensions of spatial features are 1, which is not applicable to spatial sequence data. Therefore, the convolution form of input and state connection can fully extract the spatial features of EEG. However, since the previous DSC module has already focused on significant spatial regions with valuable spatiotemporal information, we expect the ConvLSTM module to only focus on spatiotemporal feature fusion. To further optimize the model, global mean pooling is performed for input features and hidden states to reduce the spatial dimensionality. In addition, we did several experiments: (a) removing the convolutional structure in the three gates, and the convolutional structure was only introduced into the input-to-state transformation; (b) leaving the convolutional structure in the three gates; (c) leaving the convolutional structure only in the input gate/forgetting gate/output gate. The result is optimal in all experiments when the convolution structure is only introduced into the input-to-state transition, while the gate still remains the original fully connected mechanism. The added convolutional structure enables LSTM to perform spatiotemporal feature fusion, while the gate mechanism retains its advantage in long-term temporal fusion. The internal network structure of ConvLSTM is shown in Figure 4. Equations (1)–(7) show the process of ConvLSTM.where σ is the sigmoid layer, the symbol “∗” denotes the convolution operator, and “·” denotes the Hadamard product. The process is as follows: The first step is to decide what information to discard from the cell states, which is done by the forget gate. As shown in equation (3), the gate reads ht−1 and xt, and outputs a value between 0 and 1 to each number in the cell state. 1 means “keep completely” and 0 means “discard completely.” The next step is to determine what new information is stored in the cell state. As shown in equations (4) and (5), the sigmoid layer, called the “input gate layer,” determines what values we will update and the updated values are it. Then, a tanh layer creates a new vector of candidate values C˜t that will be added to the state. Finally, we need to determine what value to output. This output will be based on the state of our cell. As shown in equation (6), a sigmoid layer is run to determine which part of the cell state will be output. As shown in equation (7), we take the cell state and process it through tanh (to get a value between −1 and 1) and multiply it with the output of the sigmoid gate, and we end up outputting only the part that we determined to be output.
In addition, the attention mechanism is introduced into the model. The structure of the attention mechanism is shown in Figure 5. The mechanism learns the importance of each element from the sequence and then merges elements according to their importance. In the process of improving the model by using the attention mechanism, we tried to introduce attention mechanism at different points in the network, such as introducing an attention mechanism on the input and improving the forgetting gate/input gate/output gate by utilizing the attention mechanism on the channel. We find that it is better to introduce attention mechanism in the input, and the specific attention mechanism embedded in the three gates does not facilitate feature fusion, but it only brings additional memory and computing consumption. The process of attention mechanism can be described as follows:(8)Zt=tanhWw∗Ht+bw,(9)at=patttHt=expZtΓZw∑texpZtΓZw,(10)yt=∑tat•Ht.
The output yt of the attention mechanism module is the weighted sum of at for the output Ht of the ConvLSTM module. Zt is the output of the last classification layer, at is the output of the probability distribution of the softmax layer, where Ww, Zw and bw are the weights and bias of training, and the most significant features are selected from the output of the ConvLSTM module through the weights at, because at represents the contribution to the decoding task.
Alternatively, we use bidirectional ConvLSTM, where two ConvLSTM units with different directions are fused and communicate with the context as a complete module. The bidirectional ConvLSTM consists of two unidirectional ConvLSTM networks stacked forward and backward, where ConvLSTM⟶denotes the prediction of the forward input signal sequence from the beginning to the end, calculated forward from 1 to t to obtain and save the output of the forward hidden layer at each moment and ConvLSTM← denotes prediction of the backward input signal sequence from the end to the beginning, calculated backward from moment t to 1 to obtain and save the output of the backward hidden layer at each moment. The two ConvLSTM units perform information fusion in the hidden layer to predict the input at the current moment t. The output is determined jointly by the two ConvLSTM networks. The calculation is performed by error backpropagation in the BiConvLSTM module. The error backpropagation in BiConvLSTM module is similar to the BP algorithm, which continuously searches for better points along the negative gradient direction of parameters to be optimized until model converges. The backpropagation of the error term also includes two directions: One is the backpropagation along time, that is, the error term at each moment is calculated from the current moment t; One is to propagate the error term up one layer in BiConvLSTM module. The algorithm process of error back propagation is as follows: Firstly, the output value of each neuron was calculated forward, and there were five variable ft, it, C˜t, ot and ht. We give the forward propagation formula of ConvLSTM as equations (3)–(7). Then the error term of the neuron is calculated backwards. The error propagates step by step by gradients δ of hidden state neurons. The hidden states are ft and C˜t. Here, two gradient variables δh(t) and δC(t) are defined, so the gradient of the final sequence index position is δh(Γ) and δC(t). The gradient of δh(Γ) is determined by the output gradient error of this layer, while the reverse gradient error of δC(Γ) is composed of the gradient error of δC(t + 1) in the previous layer and the gradient error coming back from δh(t) in this layer. Therefore, only δC(t) is used in the backpropagation, and the variable δh(t) is only used to help us calculate at a certain layer and does not participate in the backpropagation. With δh(Γ) and δC(Γ), it is easy to calculate the various weight parameters Wxf, Whf, Wxi, Whi, Wxc, Whc, Wxo, Who, bf, bi, bc, bo.
3. Results and Discussion
This session mainly introduces the training and testing of the model. The model is trained using the Adam optimizer to make the model converge better. And the model is trained by minimizing the classification cross-entropy loss function. We use 4-Fold cross-validation, where two sets are training sets, one set is a validation set, and the last set is a test set. As in baseline, we use a time window of −0.5 s~4 s for all datasets. The hyperparameters of the model, including batch size, learning rate, convolutional kernel size, etc., were continuously tuned and optimized. Figures 6 and 7 show the iteration curves during the training process, demonstrating the iterative changes in accuracy and LOSS. Accuracy converges of the training set close to 1, and the LOSS is close to 0. It shows that the model achieves good convergence. The computational complexity of our model is shown in Table 2 (It should be noted that the number of parameters in the table is calculated based on the BCIC IV 2a dataset). The classification accuracy results are shown in Tables 3 and 4. A visual comparison of the classification accuracy of our method with other baseline methods is shown in Figures 8 and 9.
As can be seen from Tables 3 and 4, our method achieves better results in decoding accuracy on both datasets. Our model achieves an average classification accuracy of 73.7% and 92.6% on BCIC IV 2a and HGD datasets, respectively. The decoding performance is improved on both datasets, which demonstrates the generality of the method we proposed. Moreover, the results are validated by 4-Fold cross-validation, which proves the model in this paper reduces the effect of EEG signal variability among different subjects to a certain extent and improves the overall robustness and stability of the model, which is relevant for the application of motor imagery BCI devices.
To further understand the importance of each element of the model for the deep learning structure, we have done an ablation study, as shown in Table 5. The most initial structure is the CNN-LSTM model. The accuracy of the original model in the two datasets is 58.6% and 78.9%, respectively. The design structure in the table is the improvement of each step on the initial structure, and the improvement process is from top to bottom. It clearly shows the purpose of our improved structure and its effect on improving accuracy.
4. Conclusion
To extract and classify motor imagery EEG signal features more accurately, the DSC-ConvLSTM model based on the attention mechanism is proposed in this paper. Due to the small sample size of existing EEG datasets, but the effectiveness of deep learning greatly still depends on high-quality labeled data. In this paper, a sliding window is firstly used for data enhancement, which not only meets the requirements of deep learning models for large sample datasets but also makes the network no longer extract global characteristics of the whole trial, making the model diminish learning the individualized characteristics of every subjects' EEG data, which can effectively eliminate the influence of individual specificity and instability of EEG signals on model training. In the aspect of model construction, the convolutional neural network is used to extract the temporal and spatial frequency features of EEG signals, in which a deep separable convolutional structure is used, which not only reduces the number of network parameters and training time but also facilitates the direct feature extraction of EEG signals. Then the bidirectional ConvLSTM based on the attention mechanism is used to extract the temporal characteristics of EEG signals. ConvLSTM improves the existing LSTM cell structure, and the state-to-state conversion of ConvLSTM adopts the form of convolution. And considering the compromise between the number of parameters and performance, global mean pooling is performed for input features and hidden states to reduce the spatial dimension, and convolution is only introduced into the input-to-state transitions for spatiotemporal feature fusion, while the weights for other unnecessary transitions between other gates do not take the convolutional form. In addition, we introduce an attention mechanism only in the input to improve the classification accuracy. The results show that our model achieves better results on different datasets and makes further progress in effectiveness and robustness, which can provide finer granularity, more content, more stable and real-time system for the corresponding brain-computer interface domain, thus extending the brain-computer interface functions and improving the efficiency of brain-computer interface systems.
In the near future, we will continue keeping the focus on the research on improving the classification accuracy of EEG signals. As for the individual difference of the same network model, we will try to learn more about the algorithms of feature extraction and feature classification to improve the classification accuracy across different subjects. We believe that transfer learning can be a breakthrough to solve this kind of problem.