| Literature DB >> 35812226 |
Jingxia Chen1, Chongdan Min1, Changhao Wang1, Zhezhe Tang1, Yang Liu1, Xiuwen Hu1.
Abstract
In electroencephalograph (EEG) emotion recognition research, obtaining high-level emotional features with more discriminative information has become the key to improving the classification performance. This study proposes a new end-to-end emotion recognition method based on brain connectivity (BC) features and domain adaptive residual convolutional network (short for BC-DA-RCNN), which could effectively extract the spatial connectivity information related to the emotional state of the human brain and introduce domain adaptation to achieve accurate emotion recognition within and across the subject's EEG signals. The BC information is represented by the global brain network connectivity matrix. The DA-RCNN is used to extract high-level emotional features between different dimensions of EEG signals, reduce the domain offset between different subjects, and strengthen the common features between different subjects. The experimental results on the large public DEAP data set show that the accuracy of the subject-dependent and subject-independent binary emotion classification in valence reaches 95.15 and 88.28%, respectively, which outperforms all the benchmark methods. The proposed method is proven to have lower complexity, better generalization ability, and domain robustness that help to lay a solid foundation for the development of high-performance affective brain-computer interface applications.Entities:
Keywords: EEG; brain connectivity; domain adaptative; emotion recognition; residual convolution
Year: 2022 PMID: 35812226 PMCID: PMC9257260 DOI: 10.3389/fnins.2022.878146
Source DB: PubMed Journal: Front Neurosci ISSN: 1662-453X Impact factor: 5.152
FIGURE 1EEG signal acquisition and connectivity feature representation.
FIGURE 2Brain data-driven approach.
FIGURE 3Residual learning process.
FIGURE 4The framework of the residual convolutional neural network and the structure diagram of the residual block. (A) The framework of RCNN model. (B) The structure of residual block.
The hyper parameters of the proposed BC-DA-RCNN model.
| Layer type | Size | Stride | Output shape | |
| 1 | Input | 32 × 32 | – | None |
| 2 | Convolution layer | 32 filters size of (3 × 3 or 5 × 5 or 7 × 7) | 1 | 32 × 32 × 32 |
| 3 | Residual block 1 | 32 filters size of (3 × 3 or 5 × 5 or 7 × 7) | 1 | 32 × 32 × 32 |
| 4 | Residual block 2 | 64 filters size of (3 × 3 or 5 × 5 or 7 × 7) | 1 | 32 × 32 × 64 |
| 5 | Residual block 3 | 128 filters size of (3 × 3 or 5 × 5 or 7 × 7) | 1 | 32 × 32 × 128 |
| 6 | Dense layer 1 | 1,024 units with dropout rate: 0.2 | – | 1024 |
| 7 | Dense layer 2 | 512 units with dropout rate: 0.2 | – | 512 |
| 8 | SoftMax | – | – | 4 or 2 |
FIGURE 5Structure of the optimized DA-RCNN model.
EEG data and label format.
| Scenarios | Feature type | Window length | EEG data shape | Label shape |
| Subject-dependent | PSD | 3-s | 32(channel) × 384(points) × 4600(epochs) | 1 × 4,600 |
| PLV/PCC/TE/WCC | 3-s | 32(channel) × 32(channel) × 4600(epochs) | 1 × 4,600 | |
| Subject-independent | PSD | 3-s | 32(channel) × 384(points) × 147200(epochs) | 1 × 147,200 |
| PLV/PCC/TE/WCC | 3-s | 32(channel) × 32(channel) × 147200(epochs) | 1 × 147,200 |
Hyper parameters of the benchmark models on PLV feature.
| Benchmark models | Input data size | Implementation details |
| SVM | [32 × 32, sample_size] | kernel = “rbf”, gamma = 8, |
| BT | [32 × 32, sample_size] | Method = bag, nLearn: 100, weak learner: Tree, Type: classification |
| CNN | [batch_size, feature_size]: [60, 32 × 32] | Hidden_layers = 2, hidden_size = 64, batch_size = 60, learning_rate = 0.005, dropout = 0.2, epochs = 120 |
| LSTM | [batch_size, seq_len, channels]: [80, 32, 32] | Hidden_layers = 2, hidden_size = 64, batch_size = 120, learning_rate = 0.004, dropout = 0.2, epochs = 80, num_directions = 2 |
| DBN | [batch_size, feature_size]: [60, 32 × 32] | Hidden_layers = 3, hidden_size = 64, batch_size = 60, learning_rate = 0.004, dropout = 0.2, epochs = 140 |
| BiLSTM | [batch_size, seq_len, channels]: [60, 32, 32] | Hidden_layers = 2, hidden_size = 64, learning_rate = 0.008, dropout = 0.2, num_directions = 2, epochs = 200 |
The classification results of the DA-RCNN model configured with different hyperparameters.
| Blocks | Kernels_size | Arousal (%) | Valence (%) | ||||
| Sp | Sn | Acc | Sp | Sn | Acc | ||
| 1 | 3 × 3 | 69.54 | 80.70 | 75.12 | 72.42 | 79.10 | 75.76 |
| 5 × 5 | 72.57 | 85.19 | 78.88 | 72.59 | 85.27 | 78.93 | |
| 2 | 3 × 3 | 82.9 | 90.30 | 86.60 | 83.62 | 89.16 | 86.39 |
| 5 × 5 | 80.36 | 93.70 | 87.03 | 81.07 | 93.25 | 87.16 | |
| 3 | 3 × 3 | 88.24 | 97.66 | 92.95 | 89.12 | 97.38 | 93.25 |
| 5 × 5 |
|
|
|
|
|
| |
| 4 | 3 × 3 | 86.82 | 91.20 | 89.01 | 85.51 | 90.57 | 88.04 |
| 5 × 5 | 85.53 | 90.55 | 88.04 | 85.65 | 91.81 | 88.73 | |
| 5 | 3 × 3 | 82.88 | 88.47 | 85.73 | 82.78 | 89.06 | 85.92 |
| 5 × 5 | 81.98 | 89.00 | 85.49 | 83.70 | 87.62 | 85.66 | |
Bold values represent the best comparative results.
Overall performance of the proposed DA-RCNN model on different features.
| Models | Features | Subject-dependent | Subject-independent | |||
| Valence acc (%) | Arousal acc (%) | Valence acc (%) | Arousal acc (%) | |||
| DA-RCNN | w = 3 × 3 | PSD | 87.12 | 86.90 | 79.20 | 78.40 |
| w = 5 × 5 | 87.92 | 87.65 | 80.46 | 79.50 | ||
| w = 7 × 7 | 85.40 | 84.10 | 78.50 | 78.95 | ||
| w = 3 × 3 | PCC | 89.06 | 89.14 | 84.09 | 83.51 | |
| w = 5 × 5 | 92.37 | 92.05 | 85.42 | 84.05 | ||
| w = 7 × 7 | 90.50 | 89.42 | 82.17 | 81.36 | ||
| w = 3 × 3 | PLV | 93.25 | 92.95 | 86.05 | 85.15 | |
| w = 5 × 5 |
|
|
|
| ||
| w = 7 × 7 | 92.37 | 92.05 | 85.73 | 84.98 | ||
| w = 3 × 3 | TE | 88.14 | 88.73 | 80.08 | 80.14 | |
| w = 5 × 5 | 89.06 | 89.73 | 81.50 | 81.39 | ||
| w = 7 × 7 | 87.17 | 87.82 | 79.44 | 79.47 | ||
| w = 3 × 3 | WCC | 88.09 | 89.32 | 81.18 | 81.09 | |
| w = 5 × 5 | 89.40 | 89.90 | 83.74 | 82.20 | ||
| w = 7 × 7 | 89.72 | 87.61 | 82.33 | 80.82 | ||
Bold values represent the best comparative results.
The emotion classification accuracy (%) comparison of various methods.
| Models | Features | Subject-dependent | Subject-independent | ||
| Valence acc ( | Arousal acc ( | Valence acc ( | Arousal acc ( | ||
| BT | PSD | 78.65 (0.0061) | 78.18 (0.0059) | 70.67 (0.0052) | 71.33 (0.0048) |
| PLV | 80.13 (0.0027) | 80.50 (0.0031) | 73.98 (0.0040) | 73.06 (0.0043) | |
| SVM | PSD | 79.75 (0.0043) | 78.90 (0.0040) | 70.92 (0.0015) | 71.20 (0.0016) |
| PLV | 80.62 (0.0014) | 80.15 (0.0021) | 75.14 (0.0020) | 74.90 (0.0029) | |
| DBN | PSD | 81.50 (0.0006) | 81.30 (0.0011) | 75.37 (0.0049) | 75.45 (0.0052) |
| PLV | 83.12 (0.0007) | 82.76 (0.0008) | 77.10 (0.0050) | 77.67 (0.0032) | |
| LSTM | PSD | 82.61 (0.0033) | 81.95 (0.0041) | 76.80 (0.0014) | 77.14 (0.0010) |
| PLV | 85.66 (0.0009) | 85.90 (0.0013) | 80.85 (0.0010) | 80.25 (0.0003) | |
| CNN | PSD | 83.03 (0.0043) | 83.25 (0.0039) | 78.83 (0.0032) | 78.95 (0.0039) |
| PLV | 87.60 (0.0058) | 86.50 (0.0062) | 81.10 (0.0060) | 80.60 (0.0049) | |
| BILSTM | PSD | 85.12 (0.0010) | 84.19 (0.0012) | 79.75 (0.0073) | 80.12 (0.0065) |
| PLV | 90.71 (0.0006) | 90.27 (0.0007) | 83.89 (0.0045) | 82.75 (0.0050) | |
| DA-RCNN (w = 5 × 5) | PSD | 88.66 | 87.65 | 82.95 | 79.50 |
|
|
|
|
|
| |
Bold values represent the best comparative results.
FIGURE 6The training process curve in the subject-dependent experiment in the valence.
FIGURE 7The training process curve in the subject-independent experiment in the valence.
FIGURE 8The training process curve in the subject-dependent experiment in arousal.
FIGURE 9The training process curve in the subject-independent experiment in arousal.
The experimental results of three variants of the proposed model.
| Methods | Subject-dependent | Subject-independent | ||
| Valence acc (%) | Arousal acc (%) | Valence acc (%) | Arousal acc (%) | |
| RCNN | 85.50 | 85.11 | 80.98 | 80.70 |
| DA-RCNN | 88.66 | 87.03 | 82.95 | 82.55 |
| BC-RCNN | 90.59 | 90.05 | 85.60 | 85.13 |
| BC-DA-RCNN |
| 94.84 |
|
|
Bold values represent the best comparative results.
The classification results using different connectivity mode.
| Experiment type | Connectivity mode | PCC (Accuracy %) | PLV (Accuracy %) | ||
| w = 3 × 3 | w = 5 × 5 | w = 3 × 3 | w = 5 × 5 | ||
| Subject-dependent | Dist-mode | 92.05 | 92.37 | 93.25 |
|
| Global-mode | 90.56 | 91.28 | 93.15 |
| |
| Local-mode | 84.41 | 84.79 | 86.95 |
| |
| Subject-independent | Dist-mode | 86.42 | 86.05 | 87.60 |
|
| Global-mode | 85.64 | 86.70 | 86.64 |
| |
| Local-mode | 81.07 | 82.19 | 81.27 |
| |
Bold values represent the best comparative results.
FIGURE 10Visualization of PCC connectivity features. (A,B) Respectively shows the average PCC connectivity pattern under the positive and negative emotion.
FIGURE 11Visualization of PLV connectivity features. (A,B) Respectively shows the average PLV connectivity pattern under the positive and negative emotion.