arXiv:2502.00547cs.CVcs.AI2025-02被引 2

融合脑电与表情的多实例学习框架,提升情感识别准确率

Milmer: a Framework for Multiple Instance Learning based Multimodal Emotion Recognition

  • 用Transformer融合脑电与面部表情,引入跨模态注意力平衡特征
  • 在DEAP数据集上达到96.72%四类情绪识别准确率
  • 适合做多模态情感计算、人机交互相关研究者参考

情感在人类行为和决策中起关键作用,是人机交互中的重要研究方向。本文提出一种新型多模态情感识别框架Milmer,整合面部表情分析与脑电(EEG)信号,采用基于Transformer的融合方法。框架包含脑电预处理模块、面部特征提取与平衡模块、跨模态融合模块。为增强视觉特征提取,我们在情感相关数据集上微调了预训练的Swin Transformer。引入跨注意力机制平衡各模态的令牌表示,确保有效特征融合。核心创新在于采用多实例学习(MIL)方法,从多帧面部表情图像中提取时序动态信息,弥补以往研究对时间变化的忽略。在DEAP数据集上的大量实验表明,该框架在四类情绪识别任务中达到96.72%的分类准确率。消融实验验证了各模块贡献,凸显先进特征提取与融合策略的重要性。代码已开源:https://github.com/liangyubuaa/Milmer。

原文摘要 · Abstract (English)

Emotions play a crucial role in human behavior and decision-making, making emotion recognition a key area of interest in human-computer interaction (HCI). This study addresses the challenges of emotion recognition by integrating facial expression analysis with electroencephalogram (EEG) signals, introducing a novel multimodal framework-Milmer. The proposed framework employs a transformer-based fusion approach to effectively integrate visual and physiological modalities. It consists of an EEG preprocessing module, a facial feature extraction and balancing module, and a cross-modal fusion module. To enhance visual feature extraction, we fine-tune a pre-trained Swin Transformer on emotion-related datasets. Additionally, a cross-attention mechanism is introduced to balance token representation across modalities, ensuring effective feature integration. A key innovation of this work is the adoption of a multiple instance learning (MIL) approach, which extracts meaningful information from multiple facial expression images over time, capturing critical temporal dynamics often overlooked in previous studies. Extensive experiments conducted on the DEAP dataset demonstrate the superiority of the proposed framework, achieving a classification accuracy of 96.72% in the four-class emotion recognition task. Ablation studies further validate the contributions of each module, highlighting the significance of advanced feature extraction and fusion strategies in enhancing emotion recognition performance. Our code are available at https://github.com/liangyubuaa/Milmer.

多模态情感识别TransformerMIL

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