Contrastive Learning of Subject-Invariant EEG Representations for Cross-Subject Emotion Recognition

被引:115
作者
Shen, Xinke [1 ,2 ]
Liu, Xianggen [3 ]
Hu, Xin [4 ]
Zhang, Dan [4 ]
Song, Sen [1 ,2 ]
机构
[1] Tsinghua Univ, Dept Biomed Engn, Beijing 100084, Peoples R China
[2] Tsinghua Univ, Tsinghua Lab Brain & Intelligence, Beijing 100084, Peoples R China
[3] Sichuan Univ, Coll Comp Sci, Chengdu 610065, Sichuan, Peoples R China
[4] Tsinghua Univ, Dept Psychol, Beijing 100084, Peoples R China
基金
中国国家自然科学基金;
关键词
EEG; emotion recognition; brain-computer interface; cross-subject; contrastive learning; DIFFERENTIAL ENTROPY FEATURE; BRAIN; PERSONALITY; DISCRETE; DATABASE; MODEL;
D O I
10.1109/TAFFC.2022.3164516
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
EEG signals have been reported to be informative and reliable for emotion recognition in recent years. However, the inter-subject variability of emotion-related EEG signals still poses a great challenge for the practical applications of EEG-based emotion recognition. Inspired by recent neuroscience studies on inter-subject correlation, we proposed a Contrastive Learning method for Inter-Subject Alignment (CLISA) to tackle the cross-subject emotion recognition problem. Contrastive learning was employed to minimize the inter-subject differences by maximizing the similarity in EEG signkal representations across subjects when they received the same emotional stimuli in contrast to different ones. Specifically, a convolutional neural network was applied to learn inter-subject aligned spatiotemporal representations from EEG time series in contrastive learning. The aligned representations were subsequently used to extract differential entropy features for emotion classification. CLISA achieved state-of-the-art cross-subject emotion recognition performance on our THU-EP dataset with 80 subjects and the publicly available SEED dataset with 15 subjects. It could generalize to unseen subjects or unseen emotional stimuli in testing. Furthermore, the spatiotemporal representations learned by CLISA could provide insights into the neural mechanisms of human emotion processing.
引用
收藏
页码:2496 / 2511
页数:16
相关论文
共 102 条
[1]   Neural systems for recognizing emotion [J].
Adolphs, R .
CURRENT OPINION IN NEUROBIOLOGY, 2002, 12 (02) :169-177
[2]   Emotions Recognition Using EEG Signals: A Survey [J].
Alarcao, Soraia M. ;
Fonseca, Manuel J. .
IEEE TRANSACTIONS ON AFFECTIVE COMPUTING, 2019, 10 (03) :374-393
[3]  
Amari S, 1996, ADV NEUR IN, V8, P757
[4]   Uncovering the structure of clinical EEG signals with self-supervised learning [J].
Banville, Hubert ;
Chehab, Omar ;
Hyvarinen, Aapo ;
Engemann, Denis-Alexander ;
Gramfort, Alexandre .
JOURNAL OF NEURAL ENGINEERING, 2021, 18 (04)
[5]   AFFECT AS A PSYCHOLOGICAL PRIMITIVE [J].
Barrett, Lisa Feldman ;
Bliss-Moreau, Eliza .
ADVANCES IN EXPERIMENTAL SOCIAL PSYCHOLOGY, VOL 41, 2009, 41 :167-218
[6]   AN INFORMATION MAXIMIZATION APPROACH TO BLIND SEPARATION AND BLIND DECONVOLUTION [J].
BELL, AJ ;
SEJNOWSKI, TJ .
NEURAL COMPUTATION, 1995, 7 (06) :1129-1159
[7]   Improved epileptic seizure detection combining dynamic feature normalization with EEG novelty detection [J].
Bogaarts, J. G. ;
Hilkman, D. M. W. ;
Gommer, E. D. ;
van Kranen-Mastenbroek, V. H. J. M. ;
Reulen, J. P. H. .
MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2016, 54 (12) :1883-1892
[8]   Dynamic feature scaling for online learning of binary classifiers [J].
Bollegala, Danushka .
KNOWLEDGE-BASED SYSTEMS, 2017, 129 :97-105
[9]   Cortical Response Similarities Predict which Audiovisual Clips Individuals Viewed, but Are Unrelated to Clip Preference [J].
Bridwell, David A. ;
Roth, Cullen ;
Gupta, Cota Navin ;
Calhoun, Vince D. .
PLOS ONE, 2015, 10 (06)
[10]  
Parra LC, 2019, Arxiv, DOI arXiv:1801.08881