Light-Weight Gated Recurrent Unit for Electroencephalogram Emotion Classification
This experiment investigates the impact of reducing GRU units and dense layer neurons in a lightweight GRU architecture (LW-GRU-RU) compared to a baseline GRU model for electroencephalogram (EEG) emotion classification. A baseline GRU model is used as a reference, with an optimized GRU variant utilizing feature selection through a Random Forest-based algorithm. Experiments are conducted on a labeled emotion dataset, comparing accuracy and training efficiency across five trials. Results highlight the trade-off between model complexity, accuracy, and computational efficiency, providing insights for practical applications. Both models are evaluated on an emotion dataset for accuracy and training efficiency. The lightweight model achieves a competitive accuracy of 97.486% while reducing the average training time to 0.19 seconds per epoch, showcasing its potential for efficient real-world applications.
Authors
- Wenzhang Du (ORCID: https://orcid.org/0009-0004-4834-7895)
- Suchada Sitjongsataporn (ORCID: https://orcid.org/0000-0002-2357-2365)
- Pipat Sakarin
Institutions
- Mahanakorn University of Technology (TH)
Publication Details
- Published
- 2025-03-05
- DOI
- https://doi.org/10.1109/ieecon64081.2025.10987663
- Primary Topic
- EEG and Brain-Computer Interfaces
- Type
- article
- Field-Weighted Citation Impact
- 0.00