|
Frontiers of Information Technology & Electronic Engineering
ISSN 2095-9184 (print), ISSN 2095-9230 (online)
2024 Vol.25 No.11 P.1497-1514
VG-DOCoT: a novel DO-Conv and transformer framework via VAE-GAN technique for EEG emotion recognition
Abstract: Human emotions are intricate psychological phenomena that reflect an individual’s current physiological and psychological state. Emotions have a pronounced influence on human behavior, cognition, communication, and decision-making. However, current emotion recognition methods often suffer from suboptimal performance and limited scalability in practical applications. To solve this problem, a novel electroencephalogram (EEG) emotion recognition network named VG-DOCoT is proposed, which is based on depthwise over-parameterized convolutional (DO-Conv), transformer, and variational automatic encoder-generative adversarial network (VAE-GAN) structures. Specifically, the differential entropy (DE) can be extracted from EEG signals to create mappings into the temporal, spatial, and frequency information in preprocessing. To enhance the training data, VAE-GAN is employed for data augmentation. A novel convolution module DO-Conv is used to replace the traditional convolution layer to improve the network. A transformer structure is introduced into the network framework to reveal the global dependencies from EEG signals. Using the proposed model, a binary classification on the DEAP dataset is carried out, which achieves an accuracy of 92.52% for arousal and 92.27% for valence. Next, a ternary classification is conducted on SEED, which classifies neutral, positive, and negative emotions; an impressive average prediction accuracy of 93.77% is obtained. The proposed method significantly improves the accuracy for EEG-based emotion recognition.
Key words: Emotion recognition; Electroencephalogram (EEG); Depthwise over-parameterized convolutional (DO-Conv); Transformer; Variational automatic encoder-generative adversarial network (VAE-GAN)
南京信息工程大学电子与信息工程学院,中国南京市,210044
摘要:人类情绪是反映个体当前生理和心理状态的复杂心理现象。情绪对人类的行为、认知、交流和决策产生了显著的影响。然而,目前的情绪识别方法在实际应用中往往存在性能不佳和可扩展性有限的问题。为此,我们提出一种新颖的脑电图(EEG)情绪识别框架VG-DOCoT,它基于深度过参数化卷积(DO-Conv)、变换器和变分自编码器-生成对抗网络(VAE-GAN)结构。具体来说,在预处理中,可以从EEG信号中提取微分熵(DE),以映射到时间、空间和频率信息中。为了增强训练数据,采用VAE-GAN进行数据增强。使用一种新颖的卷积模块DO-Conv替代传统的卷积层,以提高网络性能。在网络框架中引入了变换器结构,以揭示EEG信号中的全局依赖性。使用所提出的模型,在DEAP数据集上进行了二分类任务仿真,唤醒度和效价度的准确率分别达到92.52%和92.27%。另外,在SEED数据集上进行了三分类任务测试,包括中性、积极和消极三种情绪,获得的平均预测准确率为93.77%。所提出的方法显著提高了脑电情绪识别的准确率。
关键词组:
References:
Open peer comments: Debate/Discuss/Question/Opinion
<1>
DOI:
10.1631/FITEE.2300781
CLC number:
TP183
Download Full Text:
Downloaded:
624
Download summary:
<Click Here>Downloaded:
240Clicked:
1355
Cited:
0
On-line Access:
2024-12-26
Received:
2023-11-17
Revision Accepted:
2024-02-26
Crosschecked:
2025-01-24