arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.14014cs.LGcs.AI
收藏

基于互交叉注意力机制的特征融合用于脑电情绪识别

Feature Fusion Based on Mutual-Cross-Attention Mechanism for EEG Emotion Recognition

School of Computing and Artificial Intelligence, Southwest Jiaotong University(西南交通大学计算机与人工智能学院)

Yimin Zhao, Jin Gu

AI总结:

针对现有EEG情绪识别系统复杂度高、精度一般等问题,提出结合MCA互交叉注意力特征融合机制与定制3D-CNN的方法,搭配新设计的Channel-PSD-DE三维特征,在DEAP数据集上实现了效价99.49%、唤醒99.30%的高准确率。

摘要:

客观准确的情绪诊断参考对心理学家至关重要,尤其是在应对因病理性原因难以沟通的患者时。然而,当前用于情绪识别的基于脑电图(EEG)数据的系统存在模型复杂度过高、精度一般以及可解释性有限等问题。为此,我们提出了一种新颖且有效的特征融合机制,名为互交叉注意力(Mutual-Cross-Attention, MCA)。这种纯数学机制与专门定制的三维卷积神经网络(3D-CNN)相结合,能够巧妙地发掘EEG数据中时域特征与频域特征之间的互补关系。此外,新设计的Channel-PSD-DE三维特征也为高性能表现做出了贡献。该方法最终在DEAP数据集上取得了99.49%(效价维度)和99.30%(唤醒维度)的准确率。

英文摘要:

An objective and accurate emotion diagnostic reference is vital to psychologists, especially when dealing with patients who are difficult to communicate with for pathological reasons. Nevertheless, current systems based on Electroencephalography (EEG) data utilized for sentiment discrimination have some problems, including excessive model complexity, mediocre accuracy, and limited interpretability. Consequently, we propose a novel and effective feature fusion mechanism named Mutual-Cross-Attention (MCA). Combining with a specially customized 3D Convolutional Neural Network (3D-CNN), this purely mathematical mechanism adeptly discovers the complementary relationship between time-domain and frequency-domain features in EEG data. Furthermore, the new designed Channel-PSD-DE 3D feature also contributes to the high performance. The proposed method eventually achieves 99.49% (valence) and 99.30% (arousal) accuracy on DEAP dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑