Domain Adaptation with Source Selection for Motor-Imagery based BCI

Recent successes of deep learning methods in various applications have inspired BCI researchers for their use in EEG classification. However, data insufficiency and high intra- and inter-subject variabilities hinder from taking their advantage of discovering complex patterns inherent in data, which can be potentially useful to enhance EEG classification accuracy. In this paper, we devise a novel framework of training a deep network by adapting samples of other subjects as a means of domain adaptation. Assuming that there are EEG trials of motor-imagery tasks from multiple subjects available, we first select a subject whose EEG signal characteristics are similar to the target subject based on their power spectral density in resting-state EEG signals. We then use EEG signals of both the selected subject (called a source subject) and the target subject jointly in training a deep network. Rather than training a single path network, we adopt a multi-path network architecture, where the shared bottom layers are used to discover common features for both source and target subjects, while the upper layers branch out into (1) source-target subject identification, (2) label prediction optimized for a source subject, and (3) label prediction optimized for a target subject. Based on our experimental results over the BCI Competition IV-IIa dataset, we validated the effectiveness of the proposed framework in various aspects.

[1]  Chao Wu,et al.  DeepSleepNet: A Model for Automatic Sleep Stage Scoring Based on Raw Single-Channel EEG , 2017, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[2]  Xiaofeng Xie,et al.  Motor Imagery Classification Based on Bilinear Sub-Manifold Learning of Symmetric Positive-Definite Matrices , 2017, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[3]  Yuan-Pin Lin,et al.  A subject-transfer framework for obviating inter- and intra-subject variability in EEG-based drowsiness detection , 2018, NeuroImage.

[4]  Klaus-Robert Müller,et al.  Neurophysiological predictor of SMR-based BCI performance , 2010, NeuroImage.

[5]  R. Leeb,et al.  BCI Competition 2008 { Graz data set B , 2008 .

[6]  Cuntai Guan,et al.  Filter Bank Common Spatial Pattern (FBCSP) in Brain-Computer Interface , 2008, 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence).

[7]  Qisong Wang,et al.  Unsupervised domain adaptation techniques based on auto-encoder for non-stationary EEG-based emotion recognition , 2016, Comput. Biol. Medicine.

[8]  Mohammed Yeasin,et al.  Learning Representations from EEG with Deep Recurrent-Convolutional Neural Networks , 2015, ICLR.

[9]  Brendan Z. Allison,et al.  Brain-Computer Interfaces: A Gentle Introduction , 2009 .

[10]  Klaus-Robert Müller,et al.  Covariate Shift Adaptation by Importance Weighted Cross Validation , 2007, J. Mach. Learn. Res..

[11]  Bernhard Schölkopf,et al.  Transfer Learning in Brain-Computer Interfaces , 2015, IEEE Computational Intelligence Magazine.

[12]  Cuntai Guan,et al.  Regularizing Common Spatial Patterns to Improve BCI Designs: Unified Theory and New Algorithms , 2011, IEEE Transactions on Biomedical Engineering.

[13]  Heung-Il Suk,et al.  Deep recurrent spatio-temporal neural network for motor imagery based BCI , 2018, 2018 6th International Conference on Brain-Computer Interface (BCI).

[14]  T. Jung,et al.  Improving EEG-Based Emotion Classification Using Conditional Transfer Learning , 2017, Front. Hum. Neurosci..

[15]  T. Jung,et al.  Translation of EEG Spatial Filters from Resting to Motor Imagery Using Independent Component Analysis , 2012, PloS one.

[16]  Motoaki Kawanabe,et al.  Learning a common dictionary for subject-transfer decoding with resting calibration , 2015, NeuroImage.

[17]  Christian Jutten,et al.  Transfer Learning: A Riemannian Geometry Framework With Applications to Brain–Computer Interfaces , 2018, IEEE Transactions on Biomedical Engineering.

[18]  François Laviolette,et al.  Domain-Adversarial Training of Neural Networks , 2015, J. Mach. Learn. Res..

[19]  P. Welch The use of fast Fourier transform for the estimation of power spectra: A method based on time averaging over short, modified periodograms , 1967 .

[20]  Robert T. Knight,et al.  Parameterizing neural power spectra , 2018, bioRxiv.

[21]  Wolfram Burgard,et al.  Deep learning with convolutional neural networks for EEG decoding and visualization , 2017, Human brain mapping.