arXiv:2412.00082cs.LGcs.AI2024-12被引 3

提出新框架,让脑电情绪识别在未知目标条件下仍保持高精度。

PL-DCP: A Pairwise Learning framework with Domain and Class Prototypes for EEG emotion recognition under unseen target conditions

  • 通过解耦特征分离出领域与类别原型,提升模型泛化能力。
  • 在三个公开数据集上准确率分别达82.88%、65.15%、61.29%。
  • 适合脑机接口中跨被试情绪识别,尤其目标数据完全未知时使用。

脑电信号在情感脑机接口(aBCI)中具有重要作用。近年来深度学习推动了aBCI的发展,但现有基于深度迁移学习的情绪识别方法存在对源域和目标域双重依赖的问题,且易受标签噪声影响,制约模型性能与泛化能力。为此,本文提出一种面向未知目标条件下的脑电情绪识别配对学习框架PL-DCP,融合特征解耦与原型推理思想。该框架通过特征解耦模块提取并分离情绪特征,形成领域特征与类别特征,并构建双原型表示:领域原型捕捉个体差异,类别原型保留跨被试情绪共性。同时,采用配对学习策略有效抑制错误标签带来的噪声影响。在SEED、SEED-IV和SEED-V三个公开数据集上进行系统评估,准确率分别为82.88%、65.15%和61.29%。结果表明,尽管训练阶段未接触目标域,其性能仍优于需源/目标数据的SOTA迁移学习方法。该工作为情绪识别提供了一种有效且鲁棒的解决方案。代码已开源:https://github.com/WuCB-BCI/PL_DCP。

原文摘要 · Abstract (English)

Electroencephalogram (EEG) signals serve as a powerful tool in affective Brain-Computer Interfaces (aBCIs) and play a crucial role in affective computing. In recent years, the introduction of deep learning techniques has significantly advanced the development of aBCIs. However, the current emotion recognition methods based on deep transfer learning face the challenge of the dual dependence of the model on source domain and target domain, As well as being affected by label noise, which seriously affects the performance and generalization ability of the model. To overcome this limitation, we proposes a Pairwise Learning framework with Domain and Category Prototypes for EEG emotion recognition under unseen target conditions (PL-DCP), and integrating concepts of feature disentanglement and prototype inference. Here, the feature disentanglement module extracts and decouples the emotional EEG features to form domain features and class features, and further calculates the dual prototype representation. The Domain-pprototype captures the individual variations across subjects, while the class-prototype captures the cross-individual commonality of emotion categories. In addition, the pairwise learning strategy effectively reduces the noise effect caused by wrong labels. The PL-DCP framework conducts a systematic experimental evaluation on the published datasets SEED, SEED-IV and SEED-V, and the accuracy are 82.88\%, 65.15\% and 61.29\%, respectively. The results show that compared with other State-of-the-Art(SOTA) Methods, the PL-DCP model still achieves slightly better performance than the deep transfer learning method that requires both source and target data, although the target domain is completely unseen during the training. This work provides an effective and robust potential solution for emotion recognition. The source code is available at https://github.com/WuCB-BCI/PL_DCP.

脑电识别情绪识别迁移学习原型学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。