Electroencephalogram-Based Emotion Classification Using Machine Learning and Deep Learning Techniques

https://doi.org/10.22146/ijccs.96665

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


classification; deep learning; electroencephalogram; emotion; machine learning

Full Text:

PDF


References

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 : 178

Refbacks

  • There are currently no refbacks.




Copyright (c) 2024 IJCCS (Indonesian Journal of Computing and Cybernetics Systems)

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.



Copyright of :
IJCCS (Indonesian Journal of Computing and Cybernetics Systems)
ISSN 1978-1520 (print); ISSN 2460-7258 (online)
is a scientific journal the results of Computing
and Cybernetics Systems
A publication of IndoCEISS.
Gedung S1 Ruang 416 FMIPA UGM, Sekip Utara, Yogyakarta 55281
Fax: +62274 555133
email:ijccs.mipa@ugm.ac.id | http://jurnal.ugm.ac.id/ijccs



View My Stats1
View My Stats2