Joint Contrastive Learning with Feature Alignment for Cross-Corpus EEG-based Emotion Recognition (2404.09559v1)
Abstract: 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.
- Soraia M Alarcao and Manuel J Fonseca. 2017. Emotions recognition using EEG signals: A survey. IEEE transactions on affective computing 10, 3 (2017), 374–393.
- Mashail Alsolamy and Anas Fattouh. 2016. Emotion estimation from EEG signals during listening to Quran using PSD features. In 2016 7th International Conference on computer science and information technology (CSIT). IEEE, 1–5.
- Cognitive behavioral therapy for anxiety and related disorders: A meta-analysis of randomized placebo-controlled trials. Depression and anxiety 35, 6 (2018), 502–514.
- A simple framework for contrastive learning of visual representations. In International conference on machine learning. PMLR, 1597–1607.
- Emotion recognition in human-computer interaction. IEEE Signal processing magazine 18, 1 (2001), 32–80.
- Convolutional neural networks on graphs with fast localized spectral filtering. Advances in neural information processing systems 29 (2016).
- Differential entropy feature for EEG-based emotion classification. In 2013 6th international IEEE/EMBS conference on neural engineering (NER). IEEE, 81–84.
- Patrick Flandrin. 1998. Time-frequency/time-scale analysis. Academic press.
- Nickolaos Fragopanagos and John G Taylor. 2005. Emotion recognition in human–computer interaction. Neural Networks 18, 4 (2005), 389–405.
- Domain-adversarial training of neural networks. Journal of machine learning research 17, 59 (2016), 1–35.
- Lester I Goldfischer. 1965. Autocorrelation function and power spectral density of laser-produced speckle patterns. Josa 55, 3 (1965), 247–253.
- ASTDF-Net: Attention-Based Spatial-Temporal Dual-Stream Fusion Network for EEG-Based Emotion Recognition. In Proceedings of the 31st ACM International Conference on Multimedia. 883–892.
- Deep mul timodal learning for emotion recognition in spoken language. In 2018 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 5079–5083.
- PGCN: Pyramidal graph convolutional network for EEG emotion recognition. arXiv preprint arXiv:2302.02520 (2023).
- Clocs: Contrastive learning of cardiac signals across space, time, and patients. In International Conference on Machine Learning. PMLR, 5606–5615.
- Domain adaptation techniques for EEG-based emotion recognition: a comparative study on two public datasets. IEEE Transactions on Cognitive and Developmental Systems 11, 1 (2018), 85–94.
- Domain adaptation for EEG emotion recognition based on latent representation similarity. IEEE Transactions on Cognitive and Developmental Systems 12, 2 (2019), 344–353.
- A multi-domain adaptive graph convolutional network for EEG-based emotion recognition. In Proceedings of the 29th ACM International Conference on Multimedia. 5565–5573.
- EEG based emotion recognition: A tutorial and review. Comput. Surveys 55, 4 (2022), 1–57.
- A novel transferability attention neural network model for EEG emotion recognition. Neurocomputing 447 (2021), 92–101.
- A Novel Neural Network Model based on Cerebral Hemispheric Asymmetry for EEG Emotion Recognition. In IJCAI. 1561–1567.
- Contrastive self-supervised representation learning for sensing signals from the time-frequency perspective. In 2021 International Conference on Computer Communications and Networks (ICCCN). IEEE, 1–10.
- Real-time movie-induced discrete emotion recognition from EEG signals. IEEE Transactions on Affective Computing 9, 4 (2017), 550–562.
- Contrastive representation learning for electroencephalogram classification. In Machine Learning for Health. PMLR, 238–253.
- Survey on emotional body gesture recognition. IEEE transactions on affective computing 12, 2 (2018), 505–523.
- Henri J Nussbaumer and Henri J Nussbaumer. 1982. The fast Fourier transform. Springer.
- Antonia Papandreou-Suppappola. 2018. Applications in time-frequency signal processing. CRC press.
- Soheil Rayatdoost and Mohammad Soleymani. 2018. Cross-corpus EEG-based emotion recognition. In 2018 IEEE 28th international workshop on machine learning for signal processing (MLSP). IEEE, 1–6.
- Contrastive learning of subject-invariant EEG representations for cross-subject emotion recognition. IEEE Transactions on Affective Computing (2022).
- Variational instance-adaptive graph for EEG emotion recognition. IEEE Transactions on Affective Computing 14, 1 (2021), 343–356.
- Graph-embedded convolutional neural network for image-based EEG emotion recognition. IEEE Transactions on Emerging Topics in Computing 10, 3 (2021), 1399–1413.
- MPED: A multi-modal physiological emotion database for discrete emotion recognition. IEEE Access 7 (2019), 12177–12191.
- EEG emotion recognition using dynamical graph convolutional neural networks. IEEE Transactions on Affective Computing 11, 3 (2018), 532–541.
- Johan AK Suykens and Joos Vandewalle. 1999. Least squares support vector machine classifiers. Neural processing letters 9 (1999), 293–300.
- Exploring Contrastive Learning in Human Activity Recognition for Healthcare. arXiv preprint arXiv:2011.11542 (2020).
- Laurens Van der Maaten and Geoffrey Hinton. 2008. Visualizing data using t-SNE. Journal of machine learning research 9, 11 (2008).
- Mixing up contrastive learning: Self-supervised representation learning for time series. Pattern Recognition Letters 155 (2022), 54–61.
- Ts2vec: Towards universal representation of time series. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 36. 8980–8987.
- Facial expression recognition via learning deep sparse autoencoders. Neurocomputing 273 (2018), 643–649.
- mixup: Beyond empirical risk minimization. arXiv preprint arXiv:1710.09412 (2017).
- Emotionmeter: A multimodal framework for recognizing human emotions. IEEE transactions on cybernetics 49, 3 (2018), 1110–1122.
- Wei-Long Zheng and Bao-Liang Lu. 2015. Investigating critical frequency bands and channels for EEG-based emotion recognition with deep neural networks. IEEE Transactions on autonomous mental development 7, 3 (2015), 162–175.
- EEG-based emotion recognition using regularized graph neural networks. IEEE Transactions on Affective Computing 13, 3 (2020), 1290–1301.
- PR-PL: A novel prototypical representation based pairwise learning framework for emotion recognition using EEG signals. IEEE Transactions on Affective Computing (2023).
- EEG-based Emotion Style Transfer Network for Cross-dataset Emotion Recognition. arXiv preprint arXiv:2308.05767 (2023).
- Emotion self-regulation training in major depressive disorder using simultaneous real-time fMRI and EEG neurofeedback. NeuroImage: Clinical 27 (2020), 102331.