通过成对学习与对抗训练,提升跨数据集情绪识别精度。
Unsupervised Pairwise Learning Optimization Framework for Cross-Corpus EEG-Based Emotion Recognition Based on Prototype Representation
- 设计双对抗分类器,通过三阶段对抗训练对齐跨数据集情绪特征。
- 在多个公开数据集上平均准确率提升4.76%和3.97%。
- 适合需要高鲁棒性跨域情绪识别的研究者与应用开发者。
情感计算是脑机接口领域快速发展的交叉方向。近年来深度学习的引入极大推动了情绪识别的发展。然而,由于个体生理差异以及实验环境与设备的变化,跨数据集情绪识别面临严峻挑战,尤其在决策边界附近的样本。为此,我们提出基于域对抗迁移学习的优化方法——最大分类器差异成对学习(McdPL)框架,实现细粒度情绪特征对齐。McdPL设计双对抗分类器(Ada分类器与RMS分类器),采用三阶段对抗训练以最大化分类差异并最小化特征分布冲突,尤其针对决策边界附近样本。两个分类器在训练中保持对抗关系,最终实现精确跨数据集特征对齐。同时,成对学习将样本分类转化为相似性判断,缓解标签噪声影响。我们在SEED、SEED-IV和SEED-V公开数据集上进行了系统评估,结果表明McdPL在跨数据集情绪识别任务中优于其他基线模型,平均准确率分别提升4.76%和3.97%。本工作为跨数据集情绪识别提供了有前景的解决方案。源码见https://github.com/WuCB-BCI/Mcd_PL。
原文摘要 · Abstract (English)
Affective computing is a rapidly developing interdisciplinary research direction in the field of brain-computer interface. In recent years, the introduction of deep learning technology has greatly promoted the development of the field of emotion recognition. However, due to physiological differences between subjects, as well as the variations in experimental environments and equipment, cross-corpus emotion recognition faces serious challenges, especially for samples near the decision boundary. To solve the above problems, we propose an optimization method based on domain adversarial transfer learning to fine-grained alignment of affective features, named Maximum classifier discrepancy with Pairwise Learning (McdPL) framework. In McdPL, we design a dual adversarial classifier (Ada classifier and RMS classifier), and apply a three-stage adversarial training to maximize classification discrepancy and minimize feature distribution to align controversy samples near the decision boundary. In the process of domain adversarial training, the two classifiers also maintain an adversarial relationship, ultimately enabling precise cross-corpus feature alignment. In addition, the introduction of pairwise learning transforms the classification problem of samples into a similarity problem between samples, alleviating the influence of label noise. We conducted systematic experimental evaluation of the model using publicly available SEED, SEED-IV and SEED-V databases. The results show that the McdPL model is superior to other baseline models in the cross-corpus emotion recognition task, and the average accuracy improvements of 4.76\% and 3.97\%, respectively. Our work provides a promising solution for emotion recognition cross-corpus. The source code is available at https://github.com/WuCB-BCI/Mcd_PL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。