Enhanced EEG Emotion Recognition Using MIMO-Based Denoising and Band-Wise Attention Graph Neural Network
Abstract
1. Introduction
- 1.
- Application of an MIMO noise estimation and reduction technique: We apply a multiple-input, multiple-output (MIMO) noise estimation and reduction technique to effectively remove non-stationary noise artifacts. Specifically, multichannel minima-controlled recursive averaging (M-MCRA) is applied for noise estimation, and generalized eigenvalue decomposition (GEVD)-based subspace filtering is applied for noise reduction.
- 2.
- Attention-based BFE-Net band aggregation: A single-head self-attention module is proposed to explicitly model the inter-dependency between frequency bands. Through this, the model adaptively assigns weights to frequency bands that have make considerable contributions to the determination of the current emotional state.
- 3.
- Achievement and verification of SOTA performance: In subject-independent experiments using the benchmark SEED (SJTU Emotion EEG Dataset) [19] and SEED-IV datasets, the proposed model achieved performance surpassing that of representative GNN models, including the existing DGCNN [6], RGNN [7], SOGNN [8], and BFE-Net [12] models.
2. Related Works
2.1. Theoretical Background of MIMO Signal Processing
2.2. BFE-Net
3. Proposed Methods
3.1. MIMO Noise Estimation and Reduction
3.1.1. M-MCRA-Based Non-Stationary Noise Covariance Estimation
3.1.2. Effective Subspace Decomposition Using GEVD
3.1.3. Adaptive Gain Control and Signal Reconstruction
- High SNR range (≥20 dB): Set to operate similarly to a Wiener filter, preventing large signal distortion.
- Low SNR range (<−5 dB): Set to 100 or more (experimentally, ) to suppress the generation of musical noise and strongly remove background noise.
- Intermediate range: Smoothly vary through linear interpolation to ensure filtering continuity.
3.2. Self-Attention-Based Band Aggregation
3.2.1. Input Embedding and Projection
3.2.2. Inter-Band Attention Map
3.2.3. Feature Refinement and Fusion
4. Experiments
4.1. Experimental Setup
4.1.1. Datasets
4.1.2. Implementation Details
- STFT and Model Parameters: In the STFT process for noise reduction, a window length of 64 ms and an overlap of 50% were applied.
- Data Dimensions: The model’s input data were configured as a four-dimensional tensor of . Here, N denotes the total number of samples (trials), consisting of a total of 675 (15 subjects × 3 sessions × 15 trials) for the SEED dataset and 1080 (15 subjects × 3 sessions × 24 trials) for the SEED-IV dataset. The number of EEG channels is 62, and 5 represents the number of frequency bands (). For the time dimension (T), zero-padding was applied to unify variable trial lengths, fixing it to 265 points for the SEED dataset and 64 points for the SEED-IV dataset.
- Hyperparameters and Environment: All models in this study were implemented using the PyTorch framework (v2.5.1), and training was performed on an NVIDIA RTX A6000 GPU (NVIDIA Corporation, Santa Clara, CA, USA). The Adam optimizer was used as the optimization algorithm, with the learning rate set to , batch size to 64, and total epochs to 200. Cross-entropy loss was used as the loss function, but label smoothing of was applied to enhance the model’s generalization performance. Additionally, the dropout rate was set to to prevent overfitting, and leave-one-subject-out (LOSO) cross-validation was performed, that is, the process of using data from 1 out of 15 subjects as the test set and data from the remaining 14 as the training set was repeated to verify the model’s generalization performance.
- Comparative Denoising Setup: To verify the superiority of the proposed MIMO denoising, we compared it with FastICA, a widely used blind source separation algorithm. FastICA was implemented using the scikit-learn library (v1.6.1), with the number of components set to 62 (equal to the number of channels). Artifact-related independent components (ICs) were identified based on spectral characteristics (power ratio between high/low frequency bands), and the top 10% of ICs with the highest artifact scores were removed to reconstruct the clean EEG signals.
4.2. Experimental Results
4.2.1. Comparison with State-of-the-Art Models
4.2.2. Ablation Study
4.2.3. Qualitative Analysis and Visualization
4.3. Computational Complexity Analysis
5. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Hutchison, A.; Gerstein, L. Emotion recognition, emotion expression, and cultural display rules: Implications for counseling. J. Asia Pac. Couns. 2017, 7, 19–35. [Google Scholar] [CrossRef]
- Lin, W.; Li, C. Review of studies on emotion recognition and judgment based on physiological signals. Appl. Sci. 2023, 13, 2573. [Google Scholar] [CrossRef]
- Duan, R.N.; Zhu, J.Y.; Lu, B.L. Differential entropy feature for EEG-based emotion classification. In Proceedings of the 2013 6th International IEEE/EMBS Conference on Neural Engineering (NER), San Diego, CA, USA, 6–8 November 2013; IEEE: New York, NY, USA, 2013; pp. 81–84. [Google Scholar]
- Sharma, R.; Meena, H.K. Emerging trends in EEG signal processing: A systematic review. SN Comput. Sci. 2024, 5, 415. [Google Scholar] [CrossRef]
- Bessadok, A.; Mahjoub, M.A.; Rekik, I. Graph neural networks in network neuroscience. IEEE Trans. Pattern Anal. Mach. Intell. 2022, 45, 5833–5848. [Google Scholar] [CrossRef] [PubMed]
- Song, T.; Zheng, W.; Song, P.; Cui, Z. EEG emotion recognition using dynamical graph convolutional neural networks. IEEE Trans. Affect. Comput. 2018, 11, 532–541. [Google Scholar] [CrossRef]
- Zhong, P.; Wang, D.; Miao, C. EEG-based emotion recognition using regularized graph neural networks. IEEE Trans. Affect. Comput. 2020, 13, 1290–1301. [Google Scholar] [CrossRef]
- Li, J.; Li, S.; Pan, J.; Wang, F. Cross-subject EEG emotion recognition with self-organized graph neural network. Front. Neurosci. 2021, 15, 611653. [Google Scholar] [CrossRef]
- Chen, W.; Liao, Y.; Dai, R.; Dong, Y.; Huang, L. EEG-based emotion recognition using graph convolutional neural network with dual attention mechanism. Front. Comput. Neurosci. 2024, 18, 1416494. [Google Scholar] [CrossRef]
- Lu, W.; Tan, T.P.; Ma, H. Bi-branch vision transformer network for EEG emotion recognition. IEEE Access 2023, 11, 36233–36243. [Google Scholar] [CrossRef]
- Li, Y.; Chen, J.; Li, F.; Fu, B.; Wu, H.; Ji, Y.; Zhou, Y.; Niu, Y.; Shi, G.; Zheng, W. GMSS: Graph-based multi-task self-supervised learning for EEG emotion recognition. IEEE Trans. Affect. Comput. 2022, 14, 2512–2525. [Google Scholar] [CrossRef]
- Zhang, J.; Hao, Y.; Wen, X.; Zhang, C.; Deng, H.; Zhao, J.; Cao, R. Subject-independent emotion recognition based on EEG frequency band features and self-adaptive graph construction. Brain Sci. 2024, 14, 271. [Google Scholar] [CrossRef]
- Shen, F.; Peng, Y.; Kong, W.; Dai, G. Multi-scale frequency bands ensemble learning for EEG-based emotion recognition. Sensors 2021, 21, 1262. [Google Scholar] [CrossRef]
- Yong, X.; Ward, R.K.; Birch, G.E. Facial EMG contamination of EEG signals: Characteristics and effects of spatial filtering. In Proceedings of the 2008 3rd International Symposium on Communications, Control and Signal Processing, St. Julian’s, Malta, 12–14 March 2008; IEEE: New York, NY, USA, 2008; pp. 729–734. [Google Scholar]
- Chang, C.Y.; Hsu, S.H.; Pion-Tonachini, L.; Jung, T.P. Evaluation of artifact subspace reconstruction for automatic artifact components removal in multi-channel EEG recordings. IEEE Trans. Biomed. Eng. 2019, 67, 1114–1121. [Google Scholar] [CrossRef] [PubMed]
- Makeig, S.; Bell, A.; Jung, T.P.; Sejnowski, T.J. Independent component analysis of electroencephalographic data. Adv. Neural Inf. Process. Syst. 1995, 8, 145–151. [Google Scholar]
- De Clercq, W.; Vergult, A.; Vanrumste, B.; Van Paesschen, W.; Van Huffel, S. Canonical correlation analysis applied to remove muscle artifacts from the electroencephalogram. IEEE Trans. Biomed. Eng. 2006, 53, 2583–2587. [Google Scholar] [CrossRef]
- Banik, S.; Kumar, H.; Ganapathy, N.; Swaminathan, R. Assessment of EEG-PPG Cross Frequency Coherence under Evoked Emotional Arousal. In Current Directions in Biomedical Engineering; De Gruyter: Berlin, Germany, 2024; Volume 10, pp. 49–52. [Google Scholar]
- Zheng, W.L.; Lu, B.L. Investigating critical frequency bands and channels for EEG-based emotion recognition with deep neural networks. IEEE Trans. Auton. Ment. Dev. 2015, 7, 162–175. [Google Scholar] [CrossRef]
- Cohen, I.; Berdugo, B. Noise estimation by minima controlled recursive averaging for robust speech enhancement. IEEE Signal Process. Lett. 2002, 9, 12–15. [Google Scholar] [CrossRef]
- Ephraim, Y.; Van Trees, H.L. A signal subspace approach for speech enhancement. IEEE Trans. Speech Audio Process. 1995, 3, 251–266. [Google Scholar] [CrossRef]
- Hu, Y.; Loizou, P.C. A subspace approach for enhancing speech corrupted by colored noise. In Proceedings of the 2002 IEEE International Conference on Acoustics, Speech, and Signal Processing, Orlando, FL, USA, 13–17 May 2002; IEEE: New York, NY, USA, 2002; Volume 1, pp. I-573–I-576. [Google Scholar]
- Hu, Y.; Loizou, P.C. A generalized subspace approach for enhancing speech corrupted by colored noise. IEEE Trans. Speech Audio Process. 2003, 11, 334–341. [Google Scholar] [CrossRef]
- Priyanka, S.S. A review on adaptive beamforming techniques for speech enhancement. In Proceedings of the 2017 Innovations in Power and Advanced Computing Technologies (i-PACT), Vellore, India, 21–22 April 2017; IEEE: New York, NY, USA, 2017; pp. 1–6. [Google Scholar]
- Souden, M.; Chen, J.; Benesty, J.; Affes, S. An integrated solution for online multichannel noise tracking and reduction. IEEE Trans. Audio Speech Lang. Process. 2011, 19, 2159–2169. [Google Scholar] [CrossRef]
- Hong, J. A multichannel subspace approach with signal presence probability for speech enhancement. Multidimens. Syst. Signal Process. 2019, 30, 2045–2058. [Google Scholar] [CrossRef]
- Li, Y.; Fu, B.; Li, F.; Shi, G.; Zheng, W. A novel transferability attention neural network model for EEG emotion recognition. Neurocomputing 2021, 447, 92–101. [Google Scholar] [CrossRef]
- Li, Y.; Wang, L.; Zheng, W.; Zong, Y.; Qi, L.; Cui, Z.; Zhang, T.; Song, T. A novel bi-hemispheric discrepancy model for EEG emotion recognition. IEEE Trans. Cogn. Dev. Syst. 2020, 13, 354–367. [Google Scholar] [CrossRef]
- Hyvarinen, A. Fast and robust fixed-point algorithms for independent component analysis. IEEE Trans. Neural Netw. 1999, 10, 626–634. [Google Scholar] [CrossRef]
- Lee, T.W.; Girolami, M.; Sejnowski, T.J. Independent component analysis using an extended infomax algorithm for mixed subgaussian and supergaussian sources. Neural Comput. 1999, 11, 417–441. [Google Scholar] [CrossRef]
- Delorme, A.; Makeig, S. EEGLAB: An open source toolbox for analysis of single-trial EEG dynamics including independent component analysis. J. Neurosci. Methods 2004, 134, 9–21. [Google Scholar] [CrossRef]





| Model | SEED | SEED-IV |
|---|---|---|
| SVM [7] | 56.73% ± 16.29% | 37.79% ± 12.52% |
| DAMGCN [9] | 73.21% ± 8.35% | 68.22% ± 7.03% |
| DGCNN [6] | 79.95% ± 9.02% | - |
| RGNN [7] | 85.30% ± 6.72% | 73.84% ± 8.02% |
| TANN [27] | 84.41% ± 8.75% | 68.00% ± 8.35% |
| BiHDM [28] | 85.40% ± 7.53% | 69.03% ± 8.66% |
| GMSS [11] | 86.52% ± 6.22% | 73.48% ± 7.41% |
| SOGNN [8] | 86.81% ± 5.79% | 75.27% ± 8.19% |
| BFE-Net [12] | 92.29% ± 4.65% | 79.81% ± 4.11% |
| NR-BA-BFE-Net (Proposed) | 95.56% ± 4.18% | 83.15% ± 3.40% |
| Dataset | Model | Accuracy | MCC | F1 Score | p-Value |
|---|---|---|---|---|---|
| SEED | BFE-Net | 92.59% ± 5.05% | 89.40% ± 7.08% | 92.37% ± 5.39% | - |
| NR-BA-BFE-Net | 95.56% ± 4.18% | 93.55% ± 5.96% | 95.41% ± 4.28% | 0.032 | |
| SEED-IV | BFE-Net | 80.65% ± 2.46% | 74.65% ± 3.26% | 80.40% ± 2.58% | - |
| NR-BA-BFE-Net | 83.15% ± 3.40% | 77.90% ± 4.49% | 82.99% ± 3.55% | 0.014 |
| Dataset | Model | Accuracy | MCC | F1 Score |
|---|---|---|---|---|
| SEED | BA-BFE-Net (w/o NR) | 93.48% ± 3.19% | 90.71% ± 4.45% | 93.33% ± 3.39% |
| BA-BFE-Net (w/FastICA) | 95.08% ± 4.30% | 92.85% ± 6.19% | 94.99% ± 4.31% | |
| NR-BFE-Net (w/o BA) | 93.17% ± 5.13% | 90.27% ± 7.02% | 92.70% ± 5.76% | |
| SEED-IV | BA-BFE-Net (w/o NR) | 81.85% ± 4.74% | 76.13% ± 6.22% | 81.63% ± 4.83% |
| BA-BFE-Net (w/FastICA) | 80.46% ± 3.23% | 74.21% ± 4.32% | 80.30% ± 3.19% | |
| NR-BFE-Net (w/o BA) | 82.13% ± 3.24% | 76.57% ± 4.30% | 81.95% ± 3.30% |
| Fusion Method | SEED | SEED-IV |
|---|---|---|
| Learnable Scalar Weighting | 93.65% ± 4.81% | 81.30% ± 2.95% |
| Additive Attention | 94.44% ± 3.95% | 80.46% ± 3.30% |
| Single-Head Self-Attention | 95.56% ± 4.18% | 83.15% ± 3.40% |
| Multi-Head Self-Attention | 93.02% ± 4.56% | 81.94% ± 4.30% |
| Method | Complexity (Big-O) | Iterations (I) |
|---|---|---|
| Extended Infomax | 346.5 | |
| FastICA | 98.22 | |
| MIMO | - |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Ji, Y.; Kim, D.-H.; Hong, J. Enhanced EEG Emotion Recognition Using MIMO-Based Denoising and Band-Wise Attention Graph Neural Network. Sensors 2026, 26, 1133. https://doi.org/10.3390/s26041133
Ji Y, Kim D-H, Hong J. Enhanced EEG Emotion Recognition Using MIMO-Based Denoising and Band-Wise Attention Graph Neural Network. Sensors. 2026; 26(4):1133. https://doi.org/10.3390/s26041133
Chicago/Turabian StyleJi, Yujin, Do-Hyung Kim, and Jungpyo Hong. 2026. "Enhanced EEG Emotion Recognition Using MIMO-Based Denoising and Band-Wise Attention Graph Neural Network" Sensors 26, no. 4: 1133. https://doi.org/10.3390/s26041133
APA StyleJi, Y., Kim, D.-H., & Hong, J. (2026). Enhanced EEG Emotion Recognition Using MIMO-Based Denoising and Band-Wise Attention Graph Neural Network. Sensors, 26(4), 1133. https://doi.org/10.3390/s26041133

