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 条
[31]  
Hamm J., Lee D.D., Grassmann discriminant analysis: a unifying view on subspace-based learning, Proc. 25th. Int. Conf. Mach. Learn., pp. 376-383, (2008)
[32]  
Gong B., Shi Y., Sha F., Grauman K., Geodesic flow kernel for unsupervised domain adaptation,”, Proc of IEEE Conf. On Comput. Vis. Pattern. Recognit, pp. 2066-2073, (2012)
[33]  
Wang J., Feng W., Chen Y., Et al., Visual domain adaptation with manifold embedded distribution alignment, Proc 26th ACM Int. Conf. Multimedia, pp. 402-410, (2018)
[34]  
Long M., Wang J J., Ding G., Sun J., Et al., Transfer feature learning with joint distribution adaptation, Proc of IEEE. Int. Conf. On Comput. Vis., pp. 2200-2207, (2013)
[35]  
Wang J., Chen Y., Hao S., Et al., Balanced distribution adaptation for transfer learning, Proc of IEEE. Int. Conf. On Data Mining (ICDM), pp. 1129-1134, (2017)
[36]  
Zheng W., Lu B., Investigating critical frequency bands and channels for EEG-based emotion recognition with deep neural networks, IEEE. Trans. Auton. Ment. Dev., 7, 3, pp. 162-175, (2015)
[37]  
Zheng W., Liu W., Lu Y., Lu B., Cichocki A., EmotionMeter: a multimodal framework for recognizing human emotions, IEEE Trans. Cybern., 49, 3, pp. 1110-1122, (2019)
[38]  
Koelstra S., Muhl C., Soleymani M., Lee J.-S., Et al., DEAP: a database for emotion analysis
[39]  
using physiological signals, IEEE. Trans. Affect. Comput, 3, 1, pp. 18-31, (2012)
[40]  
Duan R., Zhu J., Lu B., Differential entropy feature for EEG-based emotion classification, Proc. 6th. Int. IEEE/EMBS Conf. Neural. Eng (NER), pp. 81-84, (2013)