A subject-independent portable emotion recognition system using synchrosqueezing wavelet transform maps of EEG signals and ResNet-18

被引:20
作者
Bagherzadeh, Sara [1 ,2 ]
Norouzi, Mohammad Reza [2 ,3 ]
Hampa, Sepideh Bahri [2 ,4 ]
Ghasri, Amirhesam [2 ,4 ]
Kouroshi, Pouya Tolou [2 ,5 ]
Hosseininasab, Saman [2 ,3 ]
Zadeh, Mohammad Amin Ghasem [2 ,6 ]
Nasrabadi, Ali Motie [2 ,5 ]
机构
[1] Islamic Azad Univ, Dept Biomed Engn, Sci & Res Branch, Tehran, Iran
[2] Mirolab Ltd, Tehran, Iran
[3] Univ Tehran, Coll Sci, Dept Biotechnol, Tehran, Iran
[4] Tarbiat Modares Univ, Dept Biol Sci, Tehran, Iran
[5] Shahed Univ, Fac Engn, Dept Biomed Engn, Tehran, Iran
[6] Islamic Azad Univ, Fac Humanities, Dept Gen Psychol, West Tehran Branch, Tehran, Iran
关键词
Electroencephalogram (EEG); Emotion recognition; Synchrosqueezing wavelet transform (SSWT); Convolutional neural network (CNN); MACHINE; NETWORKS; FEATURES;
D O I
10.1016/j.bspc.2023.105875
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective: Designing a portable Brain-Computer Interface (aBCI) using EEG signals is challenging due to the numerous channels, though not all are vital for emotional recognition. We aimed to simplify this by creating a two-channel portable aBCI using advanced time-frequency analysis and deep learning. Method: Our approach involved utilizing the time-frequency analysis named synchrosqueezing wavelet transform (SSWT), which provides better frequency resolution for fluctuations of EEG signal than common wavelet transform. Using the ResNet-18 Convolutional Neural Network, we fine-tuned for sadness and happiness classification. The two best channels were identified across four databases: SEED-IV, SEED-V, SEED-GER, and SEED-FRA, using the Leave-One-Subject-Out method. Results: Finally, we achieved an average accuracy over sad and happy emotions using the SSWT-ResNet18 model of 76.66%, 78.12%, 81.25%, and 75.00% for the SEED-IV, SEED-V, SEED-GER, and SEED-FRA databases, respectively. Conclusion: Overall, our study demonstrates the potential for developing a rapid aBCI by utilizing a precise time-frequency method and deep learning technique from the least number of channels. Significance: Our approach has promising implications for future real-world applications in emotional recognition.
引用
收藏
页数:13
相关论文
共 52 条
[1]   Motor imagery EEG signal classification using image processing technique over GoogLeNet deep learning algorithm for controlling the robot manipulator [J].
Ak, Ayca ;
Topuz, Vedat ;
Midi, Ipek .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 72
[2]   Database for an emotion recognition system based on EEG signals and various computer games - GAMEEMO [J].
Alakus, Talha Burak ;
Gonen, Murat ;
Turkoglu, Ibrahim .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 60
[3]   Identifying emotions in images from valence and arousal ratings [J].
Alarcao, Soraia M. ;
Fonseca, Manuel J. .
MULTIMEDIA TOOLS AND APPLICATIONS, 2018, 77 (13) :17413-17435
[4]   Emotion recognition in EEG signals using the continuous wavelet transform and CNNs [J].
Almanza-Conejo, Oscar ;
Luz Almanza-Ojeda, Dora ;
Luis Contreras-Hernandez, Jose ;
Alberto Ibarra-Manzano, Mario .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (02) :1409-1422
[5]   COVID-19 diagnosis using state-of-the-art CNN architecture features and Bayesian Optimization [J].
Aslan, Muhammet Fatih ;
Sabanci, Kadir ;
Durdu, Akif ;
Unlersen, Muhammed Fahri .
COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 142
[6]   CNN based efficient approach for emotion recognition [J].
Aslan, Muzaffer .
JOURNAL OF KING SAUD UNIVERSITY COMPUTER AND INFORMATION SCIENCES, 2022, 34 (09) :7335-7346
[7]  
Ba J, 2014, ACS SYM SER
[8]   A Hybrid EEG-based Emotion Recognition Approach Using Wavelet Convolutional Neural Networks and Support Vector Machine [J].
Bagherzadeh, Sara ;
Maghooli, Keivan ;
Shalbaf, Ahmad ;
Maghsoudi, Arash .
BASIC AND CLINICAL NEUROSCIENCE, 2023, 14 (01) :87-102
[9]   Recognition of emotional states using frequency effective connectivity maps through transfer learning approach from electroencephalogram signals [J].
Bagherzadeh, Sara ;
Maghooli, Keivan ;
Shalbaf, Ahmad ;
Maghsoudi, Arash .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 75
[10]   Emotion recognition using effective connectivity and pre-trained convolutional neural networks in EEG signals [J].
Bagherzadeh, Sara ;
Maghooli, Keivan ;
Shalbaf, Ahmad ;
Maghsoudi, Arash .
COGNITIVE NEURODYNAMICS, 2022, 16 (05) :1087-1106