Suppr超能文献

基于脑电图的情绪识别:使用具有双重注意力机制的图卷积神经网络

EEG-based emotion recognition using graph convolutional neural network with dual attention mechanism.

作者信息

Chen Wei, Liao Yuan, Dai Rui, Dong Yuanlin, Huang Liya

机构信息

College of Electronic and Optical Engineering & College of Flexible Electronics (Future Technology), Nanjing University of Posts and Telecommunications, Nanjing, China.

出版信息

Front Comput Neurosci. 2024 Jul 19;18:1416494. doi: 10.3389/fncom.2024.1416494. eCollection 2024.

Abstract

EEG-based emotion recognition is becoming crucial in brain-computer interfaces (BCI). Currently, most researches focus on improving accuracy, while neglecting further research on the interpretability of models, we are committed to analyzing the impact of different brain regions and signal frequency bands on emotion generation based on graph structure. Therefore, this paper proposes a method named Dual Attention Mechanism Graph Convolutional Neural Network (DAMGCN). Specifically, we utilize graph convolutional neural networks to model the brain network as a graph to extract representative spatial features. Furthermore, we employ the self-attention mechanism of the Transformer model which allocates more electrode channel weights and signal frequency band weights to important brain regions and frequency bands. The visualization of attention mechanism clearly demonstrates the weight allocation learned by DAMGCN. During the performance evaluation of our model on the DEAP, SEED, and SEED-IV datasets, we achieved the best results on the SEED dataset, showing subject-dependent experiments' accuracy of 99.42% and subject-independent experiments' accuracy of 73.21%. The results are demonstrably superior to the accuracies of most existing models in the realm of EEG-based emotion recognition.

摘要

基于脑电图(EEG)的情绪识别在脑机接口(BCI)中变得至关重要。目前,大多数研究都集中在提高准确率上,而忽视了对模型可解释性的进一步研究。我们致力于基于图结构分析不同脑区和信号频段对情绪产生的影响。因此,本文提出了一种名为双注意力机制图卷积神经网络(DAMGCN)的方法。具体而言,我们利用图卷积神经网络将脑网络建模为一个图,以提取具有代表性的空间特征。此外,我们采用Transformer模型的自注意力机制,为重要的脑区和频段分配更多的电极通道权重和信号频段权重。注意力机制的可视化清晰地展示了DAMGCN学习到的权重分配。在我们的模型对DEAP、SEED和SEED-IV数据集进行性能评估时,我们在SEED数据集上取得了最佳结果,在基于受试者的实验中准确率达到99.42%,在不依赖受试者的实验中准确率达到73.21%。这些结果明显优于基于EEG的情绪识别领域中大多数现有模型的准确率。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/7172/11294218/8c1d64401ab9/fncom-18-1416494-g001.jpg

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验