A Lightweight Network with Domain Adaptation for Motor Imagery Recognition.

Abstract

Brain-computer interfaces (BCI) are an effective tool for recognizing motor imagery and have been widely applied in the motor control and assistive operation domains. However, traditional intention-recognition methods face several challenges, such as prolonged training times and limited cross-subject adaptability, which restrict their practical application. This paper proposes an innovative method that combines a lightweight convolutional neural network (CNN) with domain adaptation. A lightweight feature extraction module is designed to extract key features from both the source and target domains, effectively reducing the model's parameters and improving the real-time performance and computational efficiency. To address differences in sample distributions, a domain adaptation strategy is introduced to optimize the feature alignment. Furthermore, domain adversarial training is employed to promote the learning of domain-invariant features, significantly enhancing the model's cross-subject generalization ability. The proposed method was evaluated on an fNIRS motor imagery dataset, achieving an average accuracy of 87.76% in a three-class classification task. Additionally, lightweight experiments were conducted from two perspectives: model structure optimization and data feature selection. The results demonstrated the potential advantages of this method for practical applications in motor imagery recognition systems.