Enhancing Domain Diversity of Transfer Learning-Based SSVEP-BCIs by the Reconstruction of Channel Correlation

Wenlong Ding,Aiping Liu,Chengjuan Xie,Kai Wang,Xun Chen
DOI: https://doi.org/10.1109/TBME.2024.3458389
2024-09-10
Abstract:Objective: The application of transfer learning, specifically pre-training and fine-tuning, in steady-state visual evoked potential (SSVEP)-based brain-computer interfaces (BCIs) has been demonstrated to effectively improve the classification performance of deep learning methods with limited calibration data. However, effectively learning task-related knowledge from source domains during the pre-training phase remains challenging. To address this issue, this study proposes an effective data augmentation method called Reconstruction of Channel Correlation (RCC) to optimize the utilization of the source domain data. Methods: Concretely, RCC reconstructs training samples using probabilistically mixed eigenvector matrices derived from covariance matrices across source domains. This process manipulates the channel correlation of training samples, implicitly creating novel synthesized domains. By increasing the diversity of source domains, RCC aims to enhance the domain generalizability of the pre-trained model. The effectiveness of RCC is validated through subject-independent and subject-adaptive classification experiments. Results: The results of subject-independent classification demonstrate that RCC significantly improves the classification performance of the pre-trained model on unseen target subjects. Moreover, when compared to the fine-tuning process using the RCC-absent pre-trained model, the fine-tuning process using the RCC-enhanced pre-trained model yields significantly improved performance in the subject-adaptive classification. Conclusion: RCC proves to enhance the performance of transfer learning by optimizing the utilization of the source domain data. Significance: The RCC-enhanced transfer learning has the potential to facilitate the practical implementation of SSVEP-BCIs in real-world scenarios.
What problem does this paper attempt to address?