EEG emotion recognition using fusion model of graph convolutional neural networks and LSTM

被引:261
作者
Yin, Yongqiang [1 ,2 ]
Zheng, Xiangwei [1 ,2 ]
Hu, Bin [1 ]
Zhang, Yuang [1 ,2 ]
Cui, Xinchun [3 ]
机构
[1] Shandong Normal Univ, Sch Informat Sci & Engn, Jinan, Peoples R China
[2] Shandong Prov Key Lab Distributed Comp Software N, Jinan, Peoples R China
[3] Qufu Normal Univ, Sch Comp Sci, Rizhao, Peoples R China
基金
中国国家自然科学基金;
关键词
EEG; Emotion recognition; Long-short term memory neural network; Graph convolutional neural network; Differential entropy; CLASSIFICATION;
D O I
10.1016/j.asoc.2020.106954
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent years, graph convolutional neural networks have become research focus and inspired new ideas for emotion recognition based on EEG. Deep learning has been widely used in emotion recognition, but it is still challenging to construct models and algorithms in practical applications. In this paper, we propose a novel emotion recognition method based on a novel deep learning model (ERDL). Firstly, EEG data is calibrated by 3s baseline data and divided into segments with 6s time window, and then differential entropy is extracted from each segment to construct feature cube. Secondly, the feature cube of each segment serves as input of the novel deep learning model which fuses graph convolutional neural network (GCNN) and long-short term memories neural networks (LSTM). In the fusion model, multiple GCNNs are applied to extract graph domain features while LSTM cells are used to memorize the change of the relationship between two channels within a specific time and extract temporal features, and Dense layer is used to attain the emotion classification results. At last, we conducted extensive experiments on DEAP dataset and experimental results demonstrate that the proposed method has better classification results than the state-of-the-art methods. We attained the average classification accuracy of 90.45% and 90.60% for valence and arousal in subject-dependent experiments while 84.81% and 85.27% in subject-independent experiments. (c) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:12
相关论文
共 57 条
[1]   Analysis of evoked EEG synchronization and desynchronization in conditions of emotional activation in humans: Temporal and topographic characteristics [J].
Aftanas L.I. ;
Reva N.V. ;
Varlamov A.A. ;
Pavlov S.V. ;
Makhnev V.P. .
Neuroscience and Behavioral Physiology, 2004, 34 (8) :859-867
[2]   Emotions Recognition Using EEG Signals: A Survey [J].
Alarcao, Soraia M. ;
Fonseca, Manuel J. .
IEEE TRANSACTIONS ON AFFECTIVE COMPUTING, 2019, 10 (03) :374-393
[3]  
Alhagry S, 2017, INT J ADV COMPUT SC, V8, P355, DOI 10.14569/IJACSA.2017.081046
[4]   Emotion Recognition Based on High-Resolution EEG Recordings and Reconstructed Brain Sources [J].
Becker, Hanna ;
Fleureau, Julien ;
Guillotel, Philippe ;
Wendling, Fabrice ;
Merlet, Isabelle ;
Albera, Laurent .
IEEE TRANSACTIONS ON AFFECTIVE COMPUTING, 2020, 11 (02) :244-257
[5]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[6]  
Candra H, 2015, IEEE ENG MED BIO, P7250, DOI 10.1109/EMBC.2015.7320065
[7]  
Chang C.-C., 2011, ACM T INTEL SYST TEC, V2, P1, DOI DOI 10.1145/1961189.1961199
[8]   Accurate EEG-Based Emotion Recognition on Combined Features Using Deep Convolutional Neural Networks [J].
Chen, J. X. ;
Zhang, P. W. ;
Mao, Z. J. ;
Huang, Y. F. ;
Jiang, D. M. ;
Zhang, Andy N. .
IEEE ACCESS, 2019, 7 :44317-44328
[9]   Electroencephalogram-based emotion assessment system using ontology and data mining techniques [J].
Chen, Jing ;
Hu, Bin ;
Moore, Philip ;
Zhang, Xiaowei ;
Ma, Xu .
APPLIED SOFT COMPUTING, 2015, 30 :663-674
[10]   What does the prefrontal cortex "do" in affect: perspectives on frontal EEG asymmetry research [J].
Davidson, RJ .
BIOLOGICAL PSYCHOLOGY, 2004, 67 (1-2) :219-233