Multi-Source geometric metric transfer learning for EEG classification. (March 2023)

• Record Type: Journal Article
• Authors: Zhang, Xianxiong; She, Qingshan; Tan, Tongcai; Gao, Yunyuan; Ma, Yuliang; Zhang, Jianhai
• Is Part Of: Biomedical signal processing and control. Volume 81(2023)
• Publication Date: 2023-03
• Subjects: Signal processing -- Periodicals; Biomedical engineering -- Periodicals; Signal Processing, Computer-Assisted -- Periodicals; Image Processing, Computer-Assisted -- Periodicals; Biomedical Engineering -- Periodicals; 610.28
• DOI: 10.1016/j.bspc.2022.104435
• Languages: English
• ISSNs: 1746-8094

Abstract

Highlights: MSGMTL designs a new objective function, which integrates the idea of metric learning and transfer learning, alleviates the distribution difference and obtains a reliable metric matrix. MSGMTL effectively uses the label information of source domain to establish similar feature pairs and dissimilar feature pairs, so as to promote target domain to learn the discriminative feature representation. MSGMTL calculate the balance factor through metric matrix and dynamically allocate the weights of marginal distribution and conditional distribution. MSGMTL proves the feasibility of metric transfer learning and provides a new idea for EEG classification of BCI. Abstract: Background and Objective: In the brain computer interfaces (BCIs), transfer learning (TL) has proven its effectiveness and attracted more attention in recent research. However, traditional TL algorithms mainly use Euclidean metric to calculate distance between features, not fully exploiting the potential relationship between feature representations, which makes the improvement of performance limited. Methods: This paper proposes a multi-source geometric metric transfer learning (MSGMTL) algorithm. Firstly, multiple sources are aggregated together through Euclidean alignment (EA) to minimize the marginal distribution. Secondly, the tangent space features are extracted from a manifold to obtain the covariance matrices of EEG samples. Thirdly, three optimization components are introduced into a unified function under Mahalanobis distance metric. Namely, MSGMTL integrates pairwise constraints balanced distribution adaption based metric and structure consistency, aiming to preserve discriminative information and geometric structure to improve the performance of motor imagery (MI) classification. Results: Experiments conducted on three datasets show that, compared with other advanced methods, MSGMTL achieves better performance in classification accuracy and computational cost. Conclusion: It comes to the conclusion that the combination of metric learning and transfer learning has achieved superior performance for EEG classification and can be beneficial to advancing the application of MI-based BCIs. Index Terms — Brain computer interface (BCI), metric learning, multi-source geometric metric transfer learning (MSGMTL), Mahalanobis distance.