Improved Domain Adaptation Network Based on Wasserstein Distance for Motor Imagery EEG Classification

Qingshan She, Tie Chen, Feng Fang, Jianhai Zhang, Yunyuan Gao, Yingchun Zhang

Research output: Contribution to journalArticlepeer-review

14 Scopus citations

Abstract

Motor Imagery (MI) paradigm is critical in neural rehabilitation and gaming. Advances in brain-computer interface (BCI) technology have facilitated the detection of MI from electroencephalogram (EEG). Previous studies have proposed various EEG-based classification algorithms to identify the MI, however, the performance of prior models was limited due to the cross-subject heterogeneity in EEG data and the shortage of EEG data for training. Therefore, inspired by generative adversarial network (GAN), this study aims to propose an improved domain adaption network based on Wasserstein distance, which utilizes existing labeled data from multiple subjects (source domain) to improve the performance of MI classification on a single subject (target domain). Specifically, our proposed framework consists of three components, including a feature extractor, a domain discriminator, and a classifier. The feature extractor employs an attention mechanism and a variance layer to improve the discrimination of features extracted from different MI classes. Next, the domain discriminator adopts the Wasserstein matrix to measure the distance between source domain and target domain, and aligns the data distributions of source and target domain via adversarial learning strategy. Finally, the classifier uses the knowledge acquired from the source domain to predict the labels in the target domain. The proposed EEG-based MI classification framework was evaluated by two open-source datasets, the BCI Competition IV Datasets 2a and 2b. Our results demonstrated that the proposed framework could enhance the performance of EEG-based MI detection, achieving better classification results compared with several state-of-the-art algorithms. In conclusion, this study is promising in helping the neural rehabilitation of different neuropsychiatric diseases.

Original languageEnglish (US)
Pages (from-to)1137-1148
Number of pages12
JournalIEEE Transactions on Neural Systems and Rehabilitation Engineering
Volume31
DOIs
StatePublished - 2023

Keywords

  • Motor imagery (MI)
  • adversarial learning
  • deep neural network
  • domain adaptation
  • electroencephalogram (EEG)
  • machine learning

ASJC Scopus subject areas

  • Internal Medicine
  • Neuroscience(all)
  • Biomedical Engineering
  • Rehabilitation

Fingerprint

Dive into the research topics of 'Improved Domain Adaptation Network Based on Wasserstein Distance for Motor Imagery EEG Classification'. Together they form a unique fingerprint.

Cite this