| Literature DB >> 32940803 |
Aras M Ismael1, Ömer F Alçin2, Karmand Hussein Abdalla3, Abdulkadir Şengür4.
Abstract
In this paper, a novel approach that is based on two-stepped majority voting is proposed for efficient EEG-based emotion classification. Emotion recognition is important for human-machine interactions. Facial features- and body gestures-based approaches have been generally proposed for emotion recognition. Recently, EEG-based approaches become more popular in emotion recognition. In the proposed approach, the raw EEG signals are initially low-pass filtered for noise removal and band-pass filters are used for rhythms extraction. For each rhythm, the best performed EEG channels are determined based on wavelet-based entropy features and fractal dimension-based features. The k-nearest neighbor (KNN) classifier is used in classification. The best five EEG channels are used in majority voting for getting the final predictions for each EEG rhythm. In the second majority voting step, the predictions from all rhythms are used to get a final prediction. The DEAP dataset is used in experiments and classification accuracy, sensitivity and specificity are used for performance evaluation metrics. The experiments are carried out to classify the emotions into two binary classes such as high valence (HV) vs low valence (LV) and high arousal (HA) vs low arousal (LA). The experiments show that 86.3% HV vs LV discrimination accuracy and 85.0% HA vs LA discrimination accuracy is obtained. The obtained results are also compared with some of the existing methods. The comparisons show that the proposed method has potential in the use of EEG-based emotion classification.Entities:
Keywords: EEG rhythms; EEG-based emotion recognition; Fractal dimensions; Majority voting; Wavelet packet entropies
Year: 2020 PMID: 32940803 PMCID: PMC7498529 DOI: 10.1186/s40708-020-00111-3
Source DB: PubMed Journal: Brain Inform ISSN: 2198-4026
Fig. 1The graphical illustration of the proposed work
The pseudo-code of the proposed method
Fig. 2The rhythms of the EEG signal. a Low-pass filtered EEG signal, b Alpha rhythm, c Beta rhythm, d Gamma rhythm, e Theta rhythm, f Delta rhythm
Fig. 3The wavelet packet decomposition of an EEG signal. 2 level decomposition is used using the Daubechies wavelet of order 4
Top 5 EEG channel achievements on alpha rhythm for HV vs LV classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| Fp1 | 73.91 | 41.18 | 60.00 |
| Fc1 | 58.70 | 64.71 | 61.25 |
| T7 | 73.91 | 47.06 | 62.50 |
| O1 | 65.22 | 67.65 | 66.25 |
| T8 | 63.04 | 58.82 | 61.25 |
| Majority voting | 86.95 | 79.41 | 83.75 |
Top 5 EEG channel achievements on beta rhythm for HV vs LV classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| Fc5 | 89.13 | 29.41 | 63.75 |
| O1 | 69.57 | 61.77 | 66.25 |
| Fc6 | 89.13 | 52.94 | 73.75 |
| Fp1 | 89.13 | 35.29 | 66.25 |
| Cz | 73.91 | 47.06 | 62.50 |
| Majority voting | 97.83 | 61.77 | 82.50 |
Top 5 EEG channel achievements on theta rhythm for HV vs LV classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| Fp1 | 67.39 | 50.00 | 60.00 |
| F3 | 71.74 | 58.82 | 66.25 |
| PO4 | 76.09 | 52.94 | 66.25 |
| Fc2 | 80.44 | 55.88 | 70.00 |
| Fp2 | 82.61 | 32.35 | 61.25 |
| Majority voting | 95.65 | 58.82 | 80.00 |
Top 5 EEG channel achievements on delta rhythm for HV vs LV classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| F7 | 71.74 | 44.12 | 60.00 |
| F3 | 65.23 | 70.59 | 67.50 |
| FC1 | 78.26 | 50.00 | 66.25 |
| F4 | 80.44 | 47.06 | 66.25 |
| Fp2 | 80.44 | 50.00 | 67.50 |
| Majority voting | 97.83 | 61.77 | 82.50 |
The majority voting of the rhythm’s achievements for final emotion recognition (HV vs LV)
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| Majority voted alpha | 86.95 | 79.41 | 83.75 |
| Majority voted alpha* | 86.95 | 79.41 | 83.75 |
| Majority voted beta | 97.83 | 61.77 | 82.50 |
| Majority voted theta | 95.65 | 58.82 | 80.00 |
| Majority voted delta | 97.83 | 61.77 | 82.50 |
| Majority voting of rhythms | 100.00 | 67.65 | 86.25 |
Top 5 EEG channel achievements on alpha rhythm for HA vs LA classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| Fp1 | 66.67 | 43.75 | 57.50 |
| Fc1 | 81.25 | 25.00 | 58.75 |
| Fc6 | 79.17 | 34.38 | 61.25 |
| O1 | 77.08 | 34.38 | 60.00 |
| Cp2 | 72.92 | 40.63 | 60.00 |
| Majority voting | 97.92 | 31.25 | 71.25 |
Top 5 EEG channel achievements on beta rhythm for HA vs LA classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| F7 | 60.42 | 56.25 | 58.75 |
| Fc1 | 62.50 | 56.25 | 60.00 |
| Fc6 | 64.58 | 50.00 | 58.75 |
| Cp2 | 58.33 | 56.25 | 57.50 |
| O2 | 95.83 | 12.50 | 62.50 |
| Majority voting | 81.25 | 53.13 | 70.00 |
Top 5 EEG channel achievements on theta rhythm for HA vs LA classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| F3 | 68.75 | 53.13 | 62.500 |
| Fc1 | 75.00 | 40.63 | 61.25 |
| Fp2 | 68.75 | 65.63 | 67.50 |
| F4 | 85.42 | 37.50 | 66.25 |
| Fc2 | 72.92 | 65.63 | 70.00 |
| Majority voting | 89.58 | 62.50 | 78.75 |
Top 5 EEG channel achievements on delta rhythm for HA vs LA classification
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| P3 | 79.17 | 40.63 | 63.75 |
| Oz | 66.67 | 43.75 | 57.50 |
| Fz | 62.50 | 53.13 | 58.75 |
| C4 | 79.17 | 28.13 | 58.75 |
| Cp6 | 93.75 | 21.88 | 65.00 |
| Majority voting | 95.83 | 46.88 | 76.25 |
The majority voting of the rhythm’s achievements for final emotion recognition (HA vs LA)
| Top 5 EEG channel achievements | Sensitivity (%) | Specificity (%) | Accuracy (%) |
|---|---|---|---|
| Majority voted alpha | 97.92 | 31.25 | 71.25 |
| Majority voted beta | 81.25 | 53.13 | 70.00 |
| Majority voted theta | 89.58 | 62.50 | 78.75 |
| Majority voted delta | 95.83 | 46.88 | 76.25 |
| Majority voted theta* | 89.58 | 62.50 | 78.75 |
| Majority voting of rhythms | 93.75 | 71.88 | 85.00 |
Performance comparison of the proposed method with some of the state-of-the-art results
| Method | Accuracy (%) | |
|---|---|---|
| HV vs LV | HA vs LA | |
| Koelstra et al. [ | 57.6 | 62.0 |
| Alazrai et al. [ | 85.8 | 86.6 |
| Huang et al. [ | 66.1 | 82.5 |
| Chandra et al. [ | 65.1 | 65.3 |
| Rozgic et al. [ | 76.9 | 69.1 |
| Abeer et al. (2017) | 82.0 | 82.0 |
| Zhang et al. [ | 75.2 | 81.7 |
| Atkinson et al. [ | 73.1 | 73.0 |
| Tripathi et al. [ | 81.4 | 73.3 |
| Zhuang et al. [ | 69.1 | 71.9 |
| Li et al. [ | 80.7 | 83.7 |
| Yin et al. [ | 83.0 | 84.2 |
| Proposed study | 86.3 | 85.0 |