Electroencephalogram-Based Emotion Classification Using Machine Learning and Deep Learning Techniques
Gst Ayu Vida Mastrika Giri(1*), Made Leo Radhitya(2)
(1) Program Studi Informatika, Fakultas MIPA, Universitas Udayana, Bali
(2) Program Studi Teknik Informatika, Institut Bisnis dan Teknologi Indonesia, Bali
(*) Corresponding Author
Abstract
Electroencephalogram (EEG) records brain activity as electrical currents to discern emotions. As interest in human-computer emotional connections rises, reliable and implementable emotion recognition algorithms are essential. This study classifies EEG waves using machine and deep learning. A four-channel Muse EEG headband recorded neutral, negative, and positive emotions for the publicly available Feeling Emotions EEG dataset. Convolutional Neural Networks (CNN), Long Short-Term Memory (LSTM), and Gated Recurrent Unit (GRU) were utilized for deep learning, while SVM, K-NN, and MLP were used for machine learning. The models were assessed for accuracy, precision, recall, and F1-Score. SVM, K-NN, and MLP have accuracy scores of 0.98, 0.95, and 0.97. Deep learning methods CNN, LSTM, and GRU had 0.98, 0.82, and 0.97 accuracy. SVM and CNN surpassed other approaches in accuracy, precision, recall, and F1-Score. The research shows that machine learning and deep learning can classify EEG signals to identify emotions. High accuracy results, especially from SVM and CNN, suggest these models could be used in emotion-aware human-computer interaction systems. This study adds to EEG-based emotion classification research by revealing model selection and parameter tweaking strategies for better categorization.
Keywords
Full Text:
PDFReferences
N. Kumari, S. Anwar, and V. Bhattacharjee, “A Comparative Analysis of Machine and Deep Learning Techniques for EEG Evoked Emotion Classification,” Wireless Personal Communications. Accessed: Dec. 13, 2023. [Online]. Available: https://link.springer.com/article/10.1007/s11277-022-10076-7 [2] J. J. Bird, L. J. Manso, E. P. Ribeiro, A. Ekart, and D. R. Faria, “A Study on Mental State Classification using EEG-based Brain-Machine Interface,” 9th International Conference on Intelligent Systems 2018: Theory, Research and Innovation in Applications, IS 2018 - Proceedings. Accessed: Dec. 13, 2023. [Online]. Available: https://ieeexplore.ieee.org/document/8710576/ [3] Jordan J. Bird, “EEG Brainwave Dataset: Feeling Emotions.” Accessed: Dec. 13, 2023. [Online]. Available: https://www.kaggle.com/datasets/birdy654/eeg-brainwave-dataset-feeling-emotions [4] J. J. Bird, A. Ekart, and D. R. Faria, “Mental Emotional Sentiment Classification with an EEG-based Brain-machine Interface HANDLE Project (EU FP7) View project EMG-controlled 3D Printed Prosthetic Hand for Academia View project,” Proceedings of theInternational Conference on Digital Image and Signal Processing (DISP’19). Accessed: Dec. 13, 2023. [Online]. Available: http://jordanjamesbird.com/publications/Mental-Emotional-Sentiment-Classification-with-an-EEG-based-Brain-machine-Interface.pdf [5] N. S. Suhaimi, J. Mountstephens, and J. Teo, “EEG-Based Emotion Recognition: A State-of-the-Art Review of Current Trends and Opportunities,” Computational Intelligence and Neuroscience. Accessed: Dec. 12, 2023. [Online]. Available: https://www.hindawi.com/journals/cin/2020/8875426/ [6] F. K. Bardak, M. N. Seyman, and F. Temurtaş, “EEG Based Emotion Prediction with Neural Network Models,” Tehnicki Glasnik. Accessed: Dec. 10, 2023. [Online]. Available: https://hrcak.srce.hr/283785 [7] Shashank Joshi and Falak Joshi, “Human Emotion Classification based on EEG Signals Using Recurrent Neural Network And KNN,” International Journal of Next-Generation Computing. Accessed: Dec. 15, 2023. [Online]. Available: https://ijngc.perpetualinnovation.net/index.php/ijngc/article/view/691 [8] A. S. M. Miah, J. Shin, M. M. Islam, Abdullah, and M. K. I. Molla, “Natural Human Emotion Recognition Based on Various Mixed Reality(MR) Games and Electroencephalography (EEG) Signals,” 5th IEEE Eurasian Conference on Educational Innovation 2022, ECEI 2022. Accessed: Dec. 13, 2023. [Online]. Available: https://ieeexplore.ieee.org/document/9829482 [9] A. A. Rahman et al., “Detection of Mental State from EEG Signal Data: An Investigation with Machine Learning Classifiers,” KST 2022 - 2022 14th International Conference on Knowledge and Smart Technology. Accessed: Dec. 15, 2023. [Online]. Available: https://ieeexplore.ieee.org/document/9729084 [10] R. Qiao, C. Qing, T. Zhang, X. Xing, and X. Xu, “A novel deep-learning based framework for multi-subject emotion recognition,” in ICCSS 2017 - 2017 International Conference on Information, Cybernetics, and Computational Social Systems, 2017. doi: 10.1109/ICCSS.2017.8091408. [11] Z. Jiao, X. Gao, Y. Wang, J. Li, and H. Xu, “Deep Convolutional Neural Networks for mental load classification based on EEG data,” Pattern Recognition. Accessed: Dec. 12, 2023. [Online]. Available: https://www.sciencedirect.com/science/article/abs/pii/S0031320317304879 [12] M. K. Chowdary, J. Anitha, and D. J. Hemanth, “Emotion Recognition from EEG Signals Using Recurrent Neural Networks,” Electronics (Switzerland). Accessed: Dec. 14, 2023. [Online]. Available: https://www.mdpi.com/2079-9292/11/15/2387 [13] J. J. Bird, D. R. Faria, L. J. Manso, A. Ekárt, and C. D. Buckingham, “A deep evolutionary approach to bioinspired classifier optimization for brain-machine interaction,” Complexity. Accessed: Dec. 13, 2023. [Online]. Available: https://dx.doi.org/10.1155/2019/4316548 [14] S. M. Alarcão and M. J. Fonseca, “Emotions recognition using EEG signals: A survey,” IEEE Transactions on Affective Computing. Accessed: Dec. 12, 2023. [Online]. Available: https://ieeexplore.ieee.org/document/7946165 [15] M. Saeidi et al., “Neural decoding of EEG signals with machine learning: A systematic review,” Brain Sciences. Accessed: Dec. 14, 2023. [Online]. Available: https://www.mdpi.com/2076-3425/11/11/1525 [16] D. Jung, J. Choi, J. Kim, S. Cho, and S. Han, “EEG‐Based Identification of Emotional Neural State Evoked by Virtual Environment Interaction,” International Journal of Environmental Research and Public Health. Accessed: Dec. 18, 2023. [Online]. Available: https://www.mdpi.com/1660-4601/19/4/2158
DOI: https://doi.org/10.22146/ijccs.96665
Article Metrics
Abstract views : 221 | views : 178Refbacks
- There are currently no refbacks.
Copyright (c) 2024 IJCCS (Indonesian Journal of Computing and Cybernetics Systems)
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
View My Stats1