EEG-SCMM: Soft Contrastive Masked Modeling for Cross-Corpus EEG-Based Emotion Recognition

Read original: arXiv:2408.09186 - Published 8/20/2024 by Qile Liu, Weishan Ye, Yulu Liu, Zhen Liang
Total Score

0

EEG-SCMM: Soft Contrastive Masked Modeling for Cross-Corpus EEG-Based Emotion Recognition

Sign in to get full access

or

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

Overview

  • This paper proposes a novel approach called EEG-SCMM (Soft Contrastive Masked Modeling) for cross-corpus EEG-based emotion recognition.
  • The key ideas are to use soft contrastive learning and masked modeling to leverage unlabeled data and improve model performance across different datasets.
  • The authors evaluate their method on multiple EEG emotion recognition datasets and demonstrate superior performance compared to state-of-the-art techniques.

Plain English Explanation

The paper tackles the challenge of recognizing human emotions from electroencephalography (EEG) data, which measures electrical activity in the brain. The researchers developed a new machine learning model called EEG-SCMM that can learn patterns in EEG data to accurately identify different emotional states.

A key innovation in EEG-SCMM is the use of soft contrastive learning and masked modeling. Soft contrastive learning means the model tries to learn features that distinguish between different emotions, but does so in a more flexible way compared to traditional contrastive methods. Masked modeling involves randomly hiding parts of the input EEG data and then training the model to predict the missing information. This helps the model learn more robust and transferable features from the data.

By using these techniques, the researchers were able to leverage unlabeled EEG data to improve the model's performance, even when training on limited labeled data from a single dataset. This is important because collecting high-quality labeled EEG data for emotion recognition can be expensive and time-consuming.

The authors evaluated EEG-SCMM on multiple publicly available EEG emotion recognition datasets and showed that it outperformed existing state-of-the-art methods. This suggests the approach could be useful for building real-world EEG-based emotion recognition systems that work well across different users and contexts.

Technical Explanation

The core of the EEG-SCMM approach is a novel neural network architecture that combines soft contrastive learning and masked modeling to extract more robust and transferable features from EEG data for emotion recognition.

In the soft contrastive learning component, the model learns to map EEG samples corresponding to the same emotion closer in the feature space, while pushing apart samples from different emotions. However, instead of using a hard contrastive loss that enforces a fixed margin between positive and negative pairs, EEG-SCMM uses a soft contrastive loss that allows for more flexibility in the learned representations.

The masked modeling component trains the model to predict the missing parts of the input EEG data given the observed context. This encourages the model to learn general underlying patterns in the EEG signals that are useful for emotion recognition, rather than just memorizing superficial associations.

The authors integrate the soft contrastive learning and masked modeling objectives into a unified framework that can be optimized end-to-end. They also introduce several techniques to further improve the model's cross-corpus generalization ability, such as domain adversarial training and instance-level mixup.

The experimental results show that EEG-SCMM outperforms a range of baselines, including state-of-the-art supervised, semi-supervised, and transfer learning approaches for cross-corpus EEG-based emotion recognition. This demonstrates the effectiveness of the proposed soft contrastive masked modeling strategy in leveraging unlabeled data to learn more transferable representations.

Critical Analysis

The authors have provided a thorough evaluation of their EEG-SCMM approach, testing it on multiple publicly available datasets and comparing it to a range of existing methods. The results are quite promising and suggest the technique could be useful for building practical EEG-based emotion recognition systems.

However, the paper does not fully address some potential limitations and areas for future work. For example, the authors do not discuss how the model would perform on real-world, in-the-wild EEG data that may be noisier or have different characteristics than the curated datasets used in the experiments. Additionally, the paper does not explore the model's interpretability or provide insights into which EEG features it is learning to focus on for emotion recognition.

It would also be valuable to see an analysis of the model's computational and memory requirements, as well as its inference speed, to understand the practical deployment feasibility, especially for edge devices or mobile applications.

Overall, the EEG-SCMM approach represents an interesting and promising step forward in cross-corpus EEG-based emotion recognition. Further research to address the limitations and explore real-world applications could help solidify the technique's potential impact.

Conclusion

This paper introduces a novel EEG-based emotion recognition model called EEG-SCMM that leverages soft contrastive learning and masked modeling to learn robust and transferable features from both labeled and unlabeled data. The authors demonstrate the effectiveness of their approach through extensive experiments on multiple EEG emotion recognition datasets, outperforming state-of-the-art methods.

The key innovation of EEG-SCMM is its ability to harness the power of unsupervised learning techniques like masked modeling to extract more generalizable representations from EEG signals. This is particularly useful for emotion recognition, where labeled data can be scarce and expensive to obtain.

If further developed and deployed, EEG-SCMM could enable more accurate and robust EEG-based emotion recognition systems that can work reliably across different users and contexts. This could have important applications in mental health monitoring, human-computer interaction, and various other domains where understanding human emotions is crucial.



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

EEG-SCMM: Soft Contrastive Masked Modeling for Cross-Corpus EEG-Based Emotion Recognition
Total Score

0

EEG-SCMM: Soft Contrastive Masked Modeling for Cross-Corpus EEG-Based Emotion Recognition

Qile Liu, Weishan Ye, Yulu Liu, Zhen Liang

Emotion recognition using electroencephalography (EEG) signals has garnered widespread attention in recent years. However, existing studies have struggled to develop a sufficiently generalized model suitable for different datasets without re-training (cross-corpus). This difficulty arises because distribution differences across datasets far exceed the intra-dataset variability. To solve this problem, we propose a novel Soft Contrastive Masked Modeling (SCMM) framework. Inspired by emotional continuity, SCMM integrates soft contrastive learning with a new hybrid masking strategy to effectively mine the short-term continuity characteristics inherent in human emotions. During the self-supervised learning process, soft weights are assigned to sample pairs, enabling adaptive learning of similarity relationships across samples. Furthermore, we introduce an aggregator that weightedly aggregates complementary information from multiple close samples based on pairwise similarities among samples to enhance fine-grained feature representation, which is then used for original sample reconstruction. Extensive experiments on the SEED, SEED-IV and DEAP datasets show that SCMM achieves state-of-the-art (SOTA) performance, outperforming the second-best method by an average accuracy of 4.26% under two types of cross-corpus conditions (same-class and different-class) for EEG-based emotion recognition.

Read more

8/20/2024

Multi-modal Mood Reader: Pre-trained Model Empowers Cross-Subject Emotion Recognition
Total Score

0

Multi-modal Mood Reader: Pre-trained Model Empowers Cross-Subject Emotion Recognition

Yihang Dong, Xuhang Chen, Yanyan Shen, Michael Kwok-Po Ng, Tao Qian, Shuqiang Wang

Emotion recognition based on Electroencephalography (EEG) has gained significant attention and diversified development in fields such as neural signal processing and affective computing. However, the unique brain anatomy of individuals leads to non-negligible natural differences in EEG signals across subjects, posing challenges for cross-subject emotion recognition. While recent studies have attempted to address these issues, they still face limitations in practical effectiveness and model framework unity. Current methods often struggle to capture the complex spatial-temporal dynamics of EEG signals and fail to effectively integrate multimodal information, resulting in suboptimal performance and limited generalizability across subjects. To overcome these limitations, we develop a Pre-trained model based Multimodal Mood Reader for cross-subject emotion recognition that utilizes masked brain signal modeling and interlinked spatial-temporal attention mechanism. The model learns universal latent representations of EEG signals through pre-training on large scale dataset, and employs Interlinked spatial-temporal attention mechanism to process Differential Entropy(DE) features extracted from EEG data. Subsequently, a multi-level fusion layer is proposed to integrate the discriminative features, maximizing the advantages of features across different dimensions and modalities. Extensive experiments on public datasets demonstrate Mood Reader's superior performance in cross-subject emotion recognition tasks, outperforming state-of-the-art methods. Additionally, the model is dissected from attention perspective, providing qualitative analysis of emotion-related brain areas, offering valuable insights for affective research in neural signal processing.

Read more

5/31/2024

Joint Contrastive Learning with Feature Alignment for Cross-Corpus EEG-based Emotion Recognition
Total Score

0

Joint Contrastive Learning with Feature Alignment for Cross-Corpus EEG-based Emotion Recognition

Qile Liu, Zhihao Zhou, Jiyuan Wang, Zhen Liang

The integration of human emotions into multimedia applications shows great potential for enriching user experiences and enhancing engagement across various digital platforms. Unlike traditional methods such as questionnaires, facial expressions, and voice analysis, brain signals offer a more direct and objective understanding of emotional states. However, in the field of electroencephalography (EEG)-based emotion recognition, previous studies have primarily concentrated on training and testing EEG models within a single dataset, overlooking the variability across different datasets. This oversight leads to significant performance degradation when applying EEG models to cross-corpus scenarios. In this study, we propose a novel Joint Contrastive learning framework with Feature Alignment (JCFA) to address cross-corpus EEG-based emotion recognition. The JCFA model operates in two main stages. In the pre-training stage, a joint domain contrastive learning strategy is introduced to characterize generalizable time-frequency representations of EEG signals, without the use of labeled data. It extracts robust time-based and frequency-based embeddings for each EEG sample, and then aligns them within a shared latent time-frequency space. In the fine-tuning stage, JCFA is refined in conjunction with downstream tasks, where the structural connections among brain electrodes are considered. The model capability could be further enhanced for the application in emotion detection and interpretation. Extensive experimental results on two well-recognized emotional datasets show that the proposed JCFA model achieves state-of-the-art (SOTA) performance, outperforming the second-best method by an average accuracy increase of 4.09% in cross-corpus EEG-based emotion recognition tasks.

Read more

4/16/2024

👁️

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