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 · 发表于:IEEE Transactions on Biomedical Engineering · 年份:2024 · DOI:10.1109/tbme.2024.3458389 · 被引用次数:8 · 研究领域:EEG and Brain-Computer Interfaces、Advanced Memory and Neural Computing、Functional Brain Connectivity Studies
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 ...