CAT-Net: A Co-Adaptive Transfer Learning Network for BCI-Assisted Neurorehabilitation

Abstract

Brain-computer interfaces (BCIs) hold great potential for motor recovery in post-stroke patients. However, the motor imagery decoding accuracy is limited by the non-stationarity of EEG signals across subjects and sessions. We propose CAT-Net: a Co-Adaptive Transfer learning network to simultaneously address the inter-subject variability and inter-session nonstationarity in EEG data. The proposed method selects a relevant subset of data from all the available subjects’ data to train an initial model, followed by subject-specific transfer learning from the initial model to the target subject to establish a pretrain model. Subsequently, online adaptive training is then applied to incrementally train the pretrain model using the data from previous sessions for the target subject. This proposed network using this unique co-adaptive training method is then evaluated on both upper and lower-limb neurorehabilitation EEG datasets comprising 358 sessions from 33 stroke patients. The results showed significant accuracy improvements, achieving averaged accuracies of 70.6% and 72.3% on the respective datasets, surpassing the state-of-the-art baselines.