Semi-Supervised Dual-Stream Self-Attentive Adversarial Graph Contrastive Learning for Cross-Subject EEG-based Emotion Recognition

Read original: arXiv:2308.11635 - Published 8/6/2024 by Weishan Ye, Zhiguo Zhang, Fei Teng, Min Zhang, Jianhong Wang, Dong Ni, Fali Li, Peng Xu, Zhen Liang
Total Score

0

👁️

Sign in to get full access

or

If you already have an account, we'll log you in

Overview

  • Electroencephalography (EEG) is a technique for measuring brain activity
  • EEG-based emotion recognition has promising applications, but is limited by the lack of labeled data
  • This paper proposes a semi-supervised framework called DS-AGC to address the challenge of limited labeled data in cross-subject EEG-based emotion recognition

Plain English Explanation

The paper describes a new method for recognizing emotions based on EEG signals. EEG is a way to measure the electrical activity in the brain. The researchers developed a framework called DS-AGC that can learn to recognize emotions from EEG data, even when there is limited labeled data available.

Typically, machine learning models for emotion recognition need a lot of labeled data (data where the emotions are already identified) to learn effectively. However, collecting and labeling large EEG datasets for emotion recognition is difficult and expensive. The DS-AGC framework addresses this challenge by using a semi-supervised approach, which means it can learn from both labeled and unlabeled data.

The key ideas in the DS-AGC framework are:

  • Using two parallel "streams" to extract both non-structural and structural features from the EEG data
  • Employing a semi-supervised multi-domain adaptation method to bridge the gap between the labeled source data, unlabeled source data, and the target (unknown) data
  • Developing a graph contrastive learning method to extract effective graph-based feature representations from the EEG channels
  • Using a self-attentive fusion module to highlight the EEG features most relevant to emotions and identify data samples in the labeled source domain that are similar to the target domain

By combining these ideas, the DS-AGC framework is able to achieve better emotion recognition performance compared to existing methods, even when the amount of labeled data is limited.

Technical Explanation

The DS-AGC framework has two parallel streams for extracting EEG features: a non-structural stream and a structural stream.

The non-structural stream uses a semi-supervised multi-domain adaptation method to learn representations that can bridge the gap between the labeled source data, unlabeled source data, and the unknown target data. This helps overcome the distribution discrepancy between these domains.

The structural stream leverages a graph contrastive learning approach to extract effective graph-based feature representations from the multiple EEG channels. This exploits the spatial relationships between the channels in a semi-supervised manner.

The outputs from the two streams are then fed into a self-attentive fusion module, which highlights the EEG features that are most relevant to emotions and selects data samples from the labeled source domain that are closer to the target domain. This helps the model focus on the most informative features and data for cross-subject emotion recognition.

The researchers evaluated the DS-AGC framework on two benchmark EEG-based emotion recognition datasets (SEED and SEED-IV) using a semi-supervised cross-subject leave-one-subject-out evaluation scheme. The results show that DS-AGC outperforms existing methods by an average of 5.83% on SEED and 6.99% on SEED-IV, demonstrating its effectiveness in addressing the label scarcity problem in cross-subject EEG-based emotion recognition.

Critical Analysis

The paper presents a comprehensive and well-designed solution to the challenge of limited labeled data in EEG-based emotion recognition. The key strengths of the DS-AGC framework include its ability to leverage both labeled and unlabeled data, as well as its effective feature extraction and fusion mechanisms.

However, the paper does not provide a detailed analysis of the limitations of the proposed approach. For example, it would be helpful to understand the computational complexity of the framework and its scalability to larger datasets. Additionally, the authors could discuss potential biases or edge cases that may arise when applying DS-AGC in real-world scenarios.

Further research could also explore the generalizability of the DS-AGC framework to other modalities of emotion recognition, such as video-based or multimodal approaches. Investigating the framework's performance on a wider range of datasets and tasks would also help establish its robustness and versatility.

Conclusion

This paper presents a promising semi-supervised framework called DS-AGC for addressing the challenge of limited labeled data in cross-subject EEG-based emotion recognition. By leveraging both labeled and unlabeled data, as well as innovative feature extraction and fusion techniques, the DS-AGC framework is able to outperform existing methods and demonstrate the potential of EEG-based emotion recognition, even in scenarios with scarce labeled data.

The proposed approach could have significant implications for the development of practical EEG-based emotion recognition systems that can be deployed in a wide range of applications, from mental health monitoring to human-computer interaction. Further research and development in this area could lead to more robust and accessible emotion recognition technologies that can benefit individuals and society.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

👁️

Total Score

0

Semi-Supervised Dual-Stream Self-Attentive Adversarial Graph Contrastive Learning for Cross-Subject EEG-based Emotion Recognition

Weishan Ye, Zhiguo Zhang, Fei Teng, Min Zhang, Jianhong Wang, Dong Ni, Fali Li, Peng Xu, Zhen Liang

Electroencephalography (EEG) is an objective tool for emotion recognition with promising applications. However, the scarcity of labeled data remains a major challenge in this field, limiting the widespread use of EEG-based emotion recognition. In this paper, a semi-supervised Dual-stream Self-Attentive Adversarial Graph Contrastive learning framework (termed as DS-AGC) is proposed to tackle the challenge of limited labeled data in cross-subject EEG-based emotion recognition. The DS-AGC framework includes two parallel streams for extracting non-structural and structural EEG features. The non-structural stream incorporates a semi-supervised multi-domain adaptation method to alleviate distribution discrepancy among labeled source domain, unlabeled source domain, and unknown target domain. The structural stream develops a graph contrastive learning method to extract effective graph-based feature representation from multiple EEG channels in a semi-supervised manner. Further, a self-attentive fusion module is developed for feature fusion, sample selection, and emotion recognition, which highlights EEG features more relevant to emotions and data samples in the labeled source domain that are closer to the target domain. Extensive experiments conducted on two benchmark databases (SEED and SEED-IV) using a semi-supervised cross-subject leave-one-subject-out cross-validation evaluation scheme show that the proposed model outperforms existing methods under different incomplete label conditions (with an average improvement of 5.83% on SEED and 6.99% on SEED-IV), demonstrating its effectiveness in addressing the label scarcity problem in cross-subject EEG-based emotion recognition.

Read more

8/6/2024

👁️

Total Score

0

EEGMatch: Learning with Incomplete Labels for Semi-Supervised EEG-based Cross-Subject Emotion Recognition

Rushuang Zhou, Weishan Ye, Zhiguo Zhang, Yanyang Luo, Li Zhang, Linling Li, Gan Huang, Yining Dong, Yuan-Ting Zhang, Zhen Liang

Electroencephalography (EEG) is an objective tool for emotion recognition and shows promising performance. However, the label scarcity problem is a main challenge in this field, which limits the wide application of EEG-based emotion recognition. In this paper, we propose a novel semi-supervised learning framework (EEGMatch) to leverage both labeled and unlabeled EEG data. First, an EEG-Mixup based data augmentation method is developed to generate more valid samples for model learning. Second, a semi-supervised two-step pairwise learning method is proposed to bridge prototype-wise and instance-wise pairwise learning, where the prototype-wise pairwise learning measures the global relationship between EEG data and the prototypical representation of each emotion class and the instance-wise pairwise learning captures the local intrinsic relationship among EEG data. Third, a semi-supervised multi-domain adaptation is introduced to align the data representation among multiple domains (labeled source domain, unlabeled source domain, and target domain), where the distribution mismatch is alleviated. Extensive experiments are conducted on two benchmark databases (SEED and SEED-IV) under a cross-subject leave-one-subject-out cross-validation evaluation protocol. The results show the proposed EEGmatch performs better than the state-of-the-art methods under different incomplete label conditions (with 6.89% improvement on SEED and 1.44% improvement on SEED-IV), which demonstrates the effectiveness of the proposed EEGMatch in dealing with the label scarcity problem in emotion recognition using EEG signals. The source code is available at https://github.com/KAZABANA/EEGMatch.

Read more

9/2/2024

👨‍🏫

Total Score

0

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

Xiang Li, Jian Song, Zhigang Zhao, Chunxiao Wang, Dawei Song, Bin Hu

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.

Read more

5/14/2024

Multi-Source EEG Emotion Recognition via Dynamic Contrastive Domain Adaptation
Total Score

0

Multi-Source EEG Emotion Recognition via Dynamic Contrastive Domain Adaptation

Yun Xiao, Yimeng Zhang, Xiaopeng Peng, Shuzheng Han, Xia Zheng, Dingyi Fang, Xiaojiang Chen

Electroencephalography (EEG) provides reliable indications of human cognition and mental states. Accurate emotion recognition from EEG remains challenging due to signal variations among individuals and across measurement sessions. To address these challenges, we introduce a multi-source dynamic contrastive domain adaptation method (MS-DCDA), which models coarse-grained inter-domain and fine-grained intra-class adaptations through a multi-branch contrastive neural network and contrastive sub-domain discrepancy learning. Our model leverages domain knowledge from each individual source and a complementary source ensemble and uses dynamically weighted learning to achieve an optimal tradeoff between domain transferability and discriminability. The proposed MS-DCDA model was evaluated using the SEED and SEED-IV datasets, achieving respectively the highest mean accuracies of $90.84%$ and $78.49%$ in cross-subject experiments as well as $95.82%$ and $82.25%$ in cross-session experiments. Our model outperforms several alternative domain adaptation methods in recognition accuracy, inter-class margin, and intra-class compactness. Our study also suggests greater emotional sensitivity in the frontal and parietal brain lobes, providing insights for mental health interventions, personalized medicine, and development of preventive strategies.

Read more

8/21/2024