Emotion Recognition with Machine Learning Using EEG Signals

被引:0
作者
Bazgir, Omid [1 ]
Mohammadi, Zeynab [2 ]
Habibi, Seyed Amir Hassan [3 ]
机构
[1] Texas Tech Univ, Dept Elect & Comp Engn, Lubbock, TX 79409 USA
[2] Univ Tabriz, Dept Elect & Comp Engn, Tabriz, Iran
[3] Iran Univ Med Sci, Rasool Akram Hosp, Dept Neurol, Tehran, Iran
来源
2018 25TH IRANIAN CONFERENCE ON BIOMEDICAL ENGINEERING AND 2018 3RD INTERNATIONAL IRANIAN CONFERENCE ON BIOMEDICAL ENGINEERING (ICBME) | 2018年
关键词
Emotion; Machine Learning; Valence-arousal; EEG; DWT; PCA; SVM; KNN; ENTROPY;
D O I
暂无
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
In this research, an emotion recognition system is developed based on valence/arousal model using electroencephalography (EEG) signals. EEG signals are decomposed into the gamma, beta, alpha and theta frequency bands using discrete wavelet transform (DWT), and spectral features are extracted from each frequency band. Principle component analysis (PCA) is applied to the extracted features by preserving the same dimensionality, as a transform, to make the features mutually uncorrelated. Support vector machine (SVM), K-nearest neighbor (KNN) and artificial neural network (ANN) are used to classify emotional states. The cross-validated SVM with radial basis function (RBF) kernel using extracted features of 10 EEG channels, performs with 91.3% accuracy for arousal and 91.1% accuracy for valence, both in the beta frequency band. Our approach shows better performance compared to existing algorithms applied to the "DEAP" dataset.
引用
收藏
页码:149 / 153
页数:5
相关论文
共 18 条
[1]  
Ali M, 2016, INT CONF UBIQ FUTUR, P946, DOI 10.1109/ICUFN.2016.7536936
[2]  
[Anonymous], 2000, HDB COGNITION EMOTIO
[3]  
Bazgir O., 2018, J MED SIGNALS SENSOR, V8
[4]  
Bazgir O, 2015, 2015 22ND IRANIAN CONFERENCE ON BIOMEDICAL ENGINEERING (ICBME), P1, DOI 10.1109/ICBME.2015.7404105
[5]   Voluntary facial expression and hemispheric asymmetry over the frontal cortex [J].
Coan, JA ;
Allen, JJB ;
Harmon-Jones, E .
PSYCHOPHYSIOLOGY, 2001, 38 (06) :912-925
[6]  
Duan RN, 2013, I IEEE EMBS C NEUR E, P81, DOI 10.1109/NER.2013.6695876
[7]   Emotion recognition based on the sample entropy of EEG [J].
Jie, Xiang ;
Rui, Cao ;
Li, Li .
BIO-MEDICAL MATERIALS AND ENGINEERING, 2014, 24 (01) :1185-1192
[8]   DEAP: A Database for Emotion Analysis Using Physiological Signals [J].
Koelstra, Sander ;
Muhl, Christian ;
Soleymani, Mohammad ;
Lee, Jong-Seok ;
Yazdani, Ashkan ;
Ebrahimi, Touradj ;
Pun, Thierry ;
Nijholt, Anton ;
Patras, Ioannis .
IEEE TRANSACTIONS ON AFFECTIVE COMPUTING, 2012, 3 (01) :18-31
[9]  
Li X, 2016, IEEE INT C BIOINFORM, P352, DOI 10.1109/BIBM.2016.7822545
[10]   Wavelet-based emotion recognition system using EEG signal [J].
Mohammadi, Zeynab ;
Frounchi, Javad ;
Amiri, Mahmood .
NEURAL COMPUTING & APPLICATIONS, 2017, 28 (08) :1985-1990