用心理情绪理论构建图结构,提升脑电情绪识别准确率与合理性
Graph-Regularized Deep Learning for EEG-Based Emotion Recognition with Psychologically-Grounded Label Structure

- 将情绪建模为图节点,通过图正则化约束预测分布
- 在SEED-IV和SEED-V数据集上最高提升5.42%准确率
- 适合关注情绪可解释性与模型泛化能力的研究者
基于脑电的情绪识别对心理健康监测和情感脑机接口至关重要,但现有深度学习方法常将情绪类别视为孤立标签,忽略了其心理层面的关联性。本文提出一种图正则化学习框架,将情绪视为图中的节点,边表示基于维度情绪理论的接近度。采用三种互补的正则化策略——图标签平滑(直观软标签)、图拉普拉斯的传输距离(谱图理论)和切片沃尔什距离(图上的最优传输),按计算复杂度递增排序,惩罚偏离已知情绪拓扑的模型预测。框架在AudioTransformer(纯Transformer)、Conformer(CNN-Transformer混合)和DCGNN(因果图神经网络)三种代表性主干架构上评估,展现架构无关优势。在SEED-IV(4类)和SEED-V(5类)数据集上均实现一致提升:最佳情况下准确率提高5.42%,心理上不合理误分类减少39%。该框架有助于突破标准方法的性能上限。代码将开源。
原文摘要 · Abstract (English)
EEG-based emotion recognition is critical for mental health monitoring and affective brain-computer interfaces, yet existing deep learning approaches often treat emotion classes as isolated labels, ignoring their psychological interdependencies. We propose a graph-regularized learning framework that conceptualizes emotions as nodes in a graph where edges encode proximity based on dimensional emotion theories. We adapt three complementary regularization strategies--Graph Label Smoothing (intuitive soft labeling), Commuting distance on graph via Graph Laplacian (spectral graph theory), and Sliced Wasserstein Distance (optimal transport on graph)--ordered by increasing computational complexity. These strategies penalize model predictions that deviate from the established emotion topology. Our framework is evaluated across three representative backbone architectures: AudioTransformer (pure transformer), Conformer (CNN-transformer hybrid), and DCGNN (causal graph neural network), demonstrating architecture-agnostic benefits. Experiments on SEED-IV (4 classes) and SEED-V (5 classes) datasets show consistent improvements: best case up to +5.42% accuracy and 39% reduction in psychologically implausible misclassifications. Ultimately, our framework help raise the upper bound of performance achievable with standard approaches. Code will be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。