arXiv:2510.22197cs.LGcs.AI2025-10NeurIPS被引 4

多数据集联合预训练提升情绪识别泛化能力,无需大量标注。

Multi-dataset Joint Pre-training of Emotional EEG Enables Generalizable Affective Computing

  • 用跨数据集协方差对齐损失,统一不同数据的统计特性。
  • 少样本下AUROC提升4.57%,零样本跨数据集准确率提高11.92%。
  • 适合需要高泛化情绪计算的研究者,尤其关注跨被试、跨数据集场景。

当任务表征与通用预训练特征不一致时,任务特定的预训练至关重要。现有通用预训练脑电模型在复杂任务如情绪识别上表现不佳,主要因任务特征与广义预训练方法之间存在不匹配。本文提出一种面向跨数据集情绪识别的任务特定多数据集联合预训练框架,解决大范围数据分布差异、情绪类别定义不一致及显著个体差异问题。引入跨数据集协方差对齐损失,对齐各数据集的二阶统计特性,实现无需大量标签或个体校准的鲁棒泛化。为捕捉脑电信号的长期依赖与复杂动态,设计混合编码器,结合类Mamba线性注意力通道编码器与时空动力学模型。实验表明,该方法在少样本情绪识别中平均提升4.57% AUROC,零样本跨数据集泛化准确率提升11.92%。预训练数据集数量越多,性能越优;相比单数据集训练,多数据集联合预训练提升8.55%。本工作提供可扩展的任务特定预训练框架,凸显其在可泛化情感计算中的价值。代码已开源:https://github.com/ncclab-sustech/mdJPT_nips2025。

原文摘要 · Abstract (English)

Task-specific pre-training is essential when task representations diverge from generic pre-training features. Existing task-general pre-training EEG models struggle with complex tasks like emotion recognition due to mismatches between task-specific features and broad pre-training approaches. This work aims to develop a task-specific multi-dataset joint pre-training framework for cross-dataset emotion recognition, tackling problems of large inter-dataset distribution shifts, inconsistent emotion category definitions, and substantial inter-subject variability. We introduce a cross-dataset covariance alignment loss to align second-order statistical properties across datasets, enabling robust generalization without the need for extensive labels or per-subject calibration. To capture the long-term dependency and complex dynamics of EEG, we propose a hybrid encoder combining a Mamba-like linear attention channel encoder and a spatiotemporal dynamics model. Our method outperforms state-of-the-art large-scale EEG models by an average of 4.57% in AUROC for few-shot emotion recognition and 11.92% in accuracy for zero-shot generalization to a new dataset. Performance scales with the increase of datasets used in pre-training. Multi-dataset joint pre-training achieves a performance gain of 8.55% over single-dataset training. This work provides a scalable framework for task-specific pre-training and highlights its benefit in generalizable affective computing. Our code is available at https://github.com/ncclab-sustech/mdJPT_nips2025.

情绪识别脑电预训练泛化

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