A Supervised Information Enhanced Multi-Granularity Contrastive Learning Framework for EEG Based Emotion Recognition

Xiang Li, Jian Song, Zhigang Zhao, Chunxiao Wang, Dawei Song +1 more
5/12/2024
cs.LGcs.AIeess.SP

No official code found. Best implementation: muzixiang/SI-CLEER (17 stars, last commit 2023-10-05).

Abstract

This study introduces a novel Supervised Info-enhanced Contrastive Learning framework for EEG based Emotion Recognition (SICLEER). SI-CLEER employs multi-granularity contrastive learning to create robust EEG contextual representations, potentiallyn improving emotion recognition effectiveness. Unlike existing methods solely guided by classification loss, we propose a joint learning model combining self-supervised contrastive learning loss and supervised classification loss. This model optimizes both loss functions, capturing subtle EEG signal differences specific to emotion detection. Extensive experiments demonstrate SI-CLEER's robustness and superior accuracy on the SEED dataset compared to state-of-the-art methods. Furthermore, we analyze electrode performance, highlighting the significance of central frontal and temporal brain region EEGs in emotion detection. This study offers an universally applicable approach with potential benefits for diverse EEG classification tasks.

View on arXivView PDF

Code Implementations(1)

muzixiang/SI-CLEERCommunity100%

The source code and data for the paper 'A Supervised Information Enhanced Multi-granularity Contrastive Learning Framework for EEG based Emotion Recognition'

173PythonSep 7, 20233 years ago

Similar Papers

Cite this paper

@article{li2024supervised,
  title  = {A Supervised Information Enhanced Multi-Granularity Contrastive Learning Framework for EEG Based Emotion Recognition},
  author = {Xiang Li and Jian Song and Zhigang Zhao and Chunxiao Wang and Dawei Song and Bin Hu},
  year   = {2024},
  eprint = {2405.07260},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.07260v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.07260.svg)](https://codeofpaper.com/paper/2405.07260)

Discussion