Multi-Source Transfer Learning via Optimal Transport Feature Ranking for EEG Classification

Abstract

Motor imagery (MI) brain-computer interface (BCI) paradigms have been extensively used in neurological rehabilitation. However, due to the required long calibration time and non-stationary nature of electroencephalogram (EEG) signals, it is challenging to obtain a substantial sample size of EEG data. Transfer learning (TL), which leverages knowledge from existing subjects to enhance the learning performance in new subjects, has been employed to solve this problem. Previous studies often extract sample features directly from the two domains for transfer learning, leading to poor features and negative transfer. To overcome this limitation, an optimal transport feature selection method was developed in this study to select the most suitable features for migration to enhance the generalization capability of the TL. This was achieved by analyzing the diagonal value similarity of the optimal transport coupling matrix. First, the Riemannian tangent space mapping method was used to map the sample covariance matrix of EEG trials from the Riemannian manifold to tangent space. Secondly, the optimal subset of EEG features was selected via the optimal transport feature selection method. Subsequently, the separate distribution alignment method was employed to minimize dissimilarities between the source and target domains. Finally, the weighted voting mechanisms were integrated for decision fusion. The proposed multi-source transfer feature learning (MSTFL) method was validated based on two public BCI Competition IV datasets. Our results demonstrated superior classification accuracy of 85.93% and 79.48%, respectively, on the two public datasets, compared to state-of-the-art models. These findings indicate the effectiveness of our proposed MSTFL model for both feature extraction and classification of MI signals.