Enhancing cross-subject EEG emotion recognition through multi-source manifold metric transfer learning

被引:3
作者
Shi X. [1 ]
She Q. [1 ,4 ]
Fang F. [2 ]
Meng M. [1 ,4 ]
Tan T. [3 ]
Zhang Y. [2 ]
机构
[1] School of Automation, Hangzhou Dianzi University, Zhejiang, Hangzhou
[2] Department of Biomedical Engineering, University of Miami, Coral Gables, FL
[3] Department of Rehabilitation, Medicine, Zhejiang Provincial People's Hospital, People's Hospital of Hangzhou Medical College, Zhejiang, Hangzhou
[4] International Joint Research Laboratory for Autonomous Robotic Systems, Zhejiang, Hangzhou
基金
中国国家自然科学基金;
关键词
Affective brain-computer interface (aBCI); Electroencephalogram (EEG); Emotion recognition; Metric transfer learning;
D O I
10.1016/j.compbiomed.2024.108445
中图分类号
学科分类号
摘要
Transfer learning (TL) has demonstrated its efficacy in addressing the cross-subject domain adaptation challenges in affective brain-computer interfaces (aBCI). However, previous TL methods usually use a stationary distance, such as Euclidean distance, to quantify the distribution dissimilarity between two domains, overlooking the inherent links among similar samples, potentially leading to suboptimal feature mapping. In this study, we introduced a novel algorithm called multi-source manifold metric transfer learning (MSMMTL) to enhance the efficacy of conventional TL. Specifically, we first selected the source domain based on Mahalanobis distance to enhance the quality of the source domains and then used manifold feature mapping approach to map the source and target domains on the Grassmann manifold to mitigate data drift between domains. In this newly established shared space, we optimized the Mahalanobis metric by maximizing the inter-class distances while minimizing the intra-class distances in the target domain. Recognizing that significant distribution discrepancies might persist across different domains even on the manifold, to ensure similar distributions between the source and target domains, we further imposed constraints on both domains under the Mahalanobis metric. This approach aims to reduce distributional disparities and enhance the electroencephalogram (EEG) emotion recognition performance. In cross-subject experiments, the MSMMTL model exhibits average classification accuracies of 88.83 % and 65.04 % for SEED and DEAP, respectively, underscoring the superiority of our proposed MSMMTL over other state-of-the-art methods. MSMMTL can effectively solve the problem of individual differences in EEG-based affective computing. © 2024 Elsevier Ltd
引用
收藏
相关论文
共 53 条
[21]  
She Q., Zhang C., Fang F., Ma Y., Zhang Y., Multisource associate domain adaptation for cross-subject and cross-session EEG emotion recognition, IEEE Trans. Instrum. Meas., 72, pp. 1-12, (2023)
[22]  
Pan S.J., Yang Q., A survey on transfer learning, IEEE Trans. Knowl. Data Eng., 22, 10, pp. 1345-1359, (2010)
[23]  
Yao Y., Doretto G., Boosting for transfer learning with multiple sources, 2010 IEEE. Compute. Society. Conf. On Comput. Vis. Pattern. Recognit, pp. 1855-1862, (2010)
[24]  
Kaya M., Bilge H.S., Deep metric learning: a survey, Symmetry-Basel, 11, 9, (2019)
[25]  
Xu Y., Et al., A unified framework for metric transfer learning, IEEE Trans. Knowl. Data Eng., 29, 6, pp. 1158-1171, (2017)
[26]  
Zhang X., She Q., Tan T., Gao Y., Et al., Multi-source geometric metric transfer learning for EEG classification, Biomed. Signal. Proces., 81, (2023)
[27]  
De Maesschalck R., Jouan-Rimbaud D., Massart D.L., The mahalanobis distance, Chemometr. Intell. Lab. Syst., 50, 1, pp. 1-18, (2000)
[28]  
Peng H., Long F., Ding C., Feature selection based on mutual information criteria of max-dependency, max-relevance, and min-redundancy, IEEE Trans. Pattern Anal. Mach. Intell., 27, 8, pp. 1226-1238, (2005)
[29]  
Li W., Huan W., Hou B., Tian Y., Zhang Z., Song A., Can emotion be transferred?-A review on transfer learning for EEG-Based emotion recognition, IEEE. Trans. Cogn. Dev. Syst., 14, 3, pp. 833-846, (2022)
[30]  
Belkin M., Niyogi P., Sindhwani V., Manifold regularization: a geometric framework for learning from labeled and unlabeled examples, J. Mach. Learn. Res., 7, 85, pp. 2399-2434, (2006)