Emotion Recognition in EEG Signals with Wavelet Transform and CNN


Abstract views: 66 / PDF downloads: 16

Authors

DOI:

https://doi.org/10.5281/zenodo.15088482

Keywords:

Electroencephalography, machine learning, emotion recognition, convolutional neural network

Abstract

In this study, different Wavelet Transform methods were used. Emotion recognition was performed on EEG signals using artificial neural networks and convolutional neural networks with the features obtained by using Wavelet Transform coefficients. A dataset of EEG signals belonging to three different emotions taken from four people was used. It was used to classify stressful, neutral and relaxed emotions. By comparing the results obtained with Continuous Wavelet Transform (CWT), Discrete Wavelet Transform (DWT) for 1D and 2D and Synchrosqueezed Wavelet Transform (SSWT), an appropriate wavelet transform was tried to be determined for emotion recognition on EEG signals. It was found that Synchrosqueezed Wavelet Transform (SSWT) was the most effective algorithm for emotion classification with the highest accuracy, precision, sensitivity, specificity and F1-score.

References

Akter, S., Prodhan, R.A., Pias, T.S., Eisenberg, D., Fresneda Fernandez, J., 2022. M1M2: deep-learning-based real-time emotion recognition from neural activity. Sensors, 22(21): 8467.

Albawi, S., Mohammed, T.A., Al-Zawi S., 2017. Understanding of a convolutional neural network. International Conference on Engineering and Technology, 21 Agust, pp. 1-6.

Anonim, 2015. Keras. (https://keras.io), (Accessed: 04.01.2025).

Anonim, 2020. ssqueezepy, (https://github.com/OverLordGoldDragon/ssqueezepy/), (Accessed: 04.01.2025).

Anonim, 2024. EEG Signal for stress detection, (https://www.kaggle.com/datase ts/ishan8055/eeg-signal-for-stress-dete ction), (Accessed: 04.01.2025).

Bajaj, N., 2020. Wavelets for EEG analysis. Wavelet Theory, 1-16.

Daubechies, I., Lu, J., Wu, H.T., 2011. Synchrosqueezed wavelet transforms: An empirical mode decomposition-like tool. Applied and Computational Harmonic Analysis, 30(2): 243-261.

Fieres, J., Schemmel, J., Meier, K., 2006. Training convolutional networks of threshold neurons suited for low-power hardware implementation. IEEE International Joint Conference on Neural Network, Proceedings Book, pp. 21-28.

Ghosh, A., Sufian, A., Sultana, F., Chakrabarti, A., De, D., 2020. Fundamental concepts of convolutional neural network. Recent Trends and Advances in Artificial Intelligence and Internet of Things, 519-567.

Hunter, J.D., 2007. Matplotlib: A 2D graphics environment. Computing in Science & Engineering, 9(03): 90-95.

Lee, H.K., Choi, Y.S., 2019. Application of continuous wavelet transform and convolutional neural network in decoding motor imagery brain-computer interface. Entropy, 21(12): 1199.

Lee, G., Gommers, R., Waselewski, F., Wohlfahrt, K., O'Leary, A., 2019. PyWavelets: A Python package for wavelet analysis. Journal of Open Source Software, 4(36): 1237.

Mallat, S.G., 1989. A theory for multiresolution signal decomposition: the wavelet representation. IEEE Transactions on Pattern Analysis and Machine Intelligence, 11(7): 674-693.

McKinney, W., 2010. Data structures for statistical computing in Python. SciPy, 445(1): 51-56.

Ozdemir, M.A., Degirmenci, M., Izci, E., Akan, A., 2021. EEG-based emotion recognition with deep convolutional neural networks. Biomedical Engineering/Biomedizinische Technic, 66(1): 43-57.

Pedregosa, F., Varoquaux, G., Gramfort, A., Michel, V., Thirion, B., Grisel, O., Duchesnay, É., 2011. Scikit-learn: Machine learning in Python. The Journal of Machine Learning Research, 12: 2825-2830.

Saranya, K., Sudharson, D., Gokila, N.M., Lakshmipriya, S.R., Suriyavathi, K.A.K., 2023. Emotion Recognition using EEG Signal Classification of seed Dataset. 2nd International Conference on Advancements in Electrical, Electronics, Communication, Computing and Automation (ICAECA), Proceedings Book, pp. 1-6.

Vijayan, A.E., Sen, D., Sudheer, A.P., 2015. EEG-based emotion recognition using statistical measures and auto-regressive modeling. IEEE International Conference on Computational ıntelligence & Communication Technology, pp. 587-591.

Virtanen, P., Gommers, R., Oliphant, T.E., Haberland, M., Reddy, T., Cournapeau, D., Van Mulbregt, P., 2020. SciPy 1.0: fundamental algorithms for scientific computing in Python. Nature Methods, 17(3): 261-272.

Published

2025-03-28

How to Cite

KALAJE , A., & DEMİR, H. (2025). Emotion Recognition in EEG Signals with Wavelet Transform and CNN. MAS Journal of Applied Sciences, 10(1), 81–89. https://doi.org/10.5281/zenodo.15088482

Issue

Section

Articles