Dynamic time warping-based transfer learning for improving common spatial patterns in brain–computer interface

OBJECTIVE Common spatial patterns (CSP) is a prominent feature extraction algorithm in motor imagery (MI)-based brain-computer interfaces (BCIs). However, CSP is computed using sample-based covariance-matrix estimation. Hence, its performance deteriorates if the number of training trials is small. To address this problem, this paper proposes a novel regularized covariance matrix estimation framework for CSP (i.e. DTW-RCSP) based on dynamic time warping (DTW) and transfer learning. APPROACH The proposed framework combines the subject-specific covariance matrix (Σss) estimated using the few available trials from the new subject, with a novelDTW-based transferred covariance matrix (ΣDTW) estimated using previous subjects' trials. In the proposedΣDTW, the available labelled trials from the previous subjects are temporally aligned to the average of the few available trials of the new subject from the same class using DTW. This alignment aims to reduce temporal variations and non-stationarities between previous subjects trials and the available few trials from the new subjects. Moreover, to tackle the problem of regularization parameter selection when only few trials are available for training, an online method is proposed, where the best regularization parameter is selected based on the confidence scores of the trained classifier on upcoming first few labelled testing trials. MAIN RESULTS The proposed framework is evaluated on two datasets against two baseline algorithms. The obtained results reveal that DTW-RCSP significantly outperformed the baseline algorithms at various testing scenarios, particularly, when only a few trials are available for training. SIGNIFICANCE Impressively, our results show that successful BCI interactions could be achieved with a calibration session as small as only one trial per class.

[1]  Saeid Nahavandi,et al.  Multiclass Informative Instance Transfer Learning Framework for Motor Imagery-Based Brain-Computer Interface , 2018, Comput. Intell. Neurosci..

[2]  K.-R. Muller,et al.  Optimizing Spatial filters for Robust EEG Single-Trial Analysis , 2008, IEEE Signal Processing Magazine.

[3]  PerlovskyLeonid 2009 Special Issue , 2009 .

[4]  E. Curran,et al.  Learning to control brain activity: A review of the production and control of EEG components for driving brain–computer interface (BCI) systems , 2003, Brain and Cognition.

[5]  David Lee,et al.  Filter Bank Regularized Common Spatial Pattern Ensemble for Small Sample Motor Imagery Classification , 2018, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[6]  Mahnaz Arvaneh,et al.  Weighted Transfer Learning for Improving Motor Imagery-Based Brain–Computer Interface , 2019, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[7]  Yasuharu Koike,et al.  Application of Covariate Shift Adaptation Techniques in Brain–Computer Interfaces , 2010, IEEE Transactions on Biomedical Engineering.

[8]  Shiliang Sun,et al.  A subject transfer framework for EEG classification , 2012, Neurocomputing.

[9]  Mahnaz Arvaneh,et al.  Robust Common Spatial Patterns Estimation Using Dynamic Time Warping to Improve BCI Systems , 2019, ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[10]  Sriram Subramanian,et al.  Predicting Mental Imagery-Based BCI Performance from Personality, Cognitive Profile and Neurophysiological Patterns , 2015, PloS one.

[11]  Motoaki Kawanabe,et al.  Toward Unsupervised Adaptation of LDA for Brain–Computer Interfaces , 2011, IEEE Transactions on Biomedical Engineering.

[12]  G. Pfurtscheller,et al.  Brain-Computer Interfaces for Communication and Control. , 2011, Communications of the ACM.

[13]  Klaus-Robert Müller,et al.  Subject-independent mental state classification in single trials , 2009, Neural Networks.

[14]  Cuntai Guan,et al.  Optimizing Spatial Filters by Minimizing Within-Class Dissimilarities in Electroencephalogram-Based Brain–Computer Interface , 2013, IEEE Transactions on Neural Networks and Learning Systems.

[15]  G. Pfurtscheller,et al.  Optimal spatial filtering of single trial EEG during imagined hand movement. , 2000, IEEE transactions on rehabilitation engineering : a publication of the IEEE Engineering in Medicine and Biology Society.

[16]  Eamonn J. Keogh,et al.  Exact indexing of dynamic time warping , 2002, Knowledge and Information Systems.

[17]  Cuntai Guan,et al.  Facilitating motor imagery-based brain–computer interface for stroke patients using passive movement , 2016, Neural Computing and Applications.

[18]  J. Friedman Regularized Discriminant Analysis , 1989 .

[19]  Cuntai Guan,et al.  Learning from other subjects helps reducing Brain-Computer Interface calibration time , 2010, 2010 IEEE International Conference on Acoustics, Speech and Signal Processing.

[20]  Motoaki Kawanabe,et al.  Divergence-Based Framework for Common Spatial Patterns Algorithms , 2014, IEEE Reviews in Biomedical Engineering.

[21]  Seungjin Choi,et al.  Composite Common Spatial Pattern for Subject-to-Subject Transfer , 2009, IEEE Signal Processing Letters.

[22]  T P Speed,et al.  A score test for the linkage analysis of qualitative and quantitative traits based on identity by descent data from sib-pairs. , 2000, Biostatistics.

[23]  Ahmed M. Azab,et al.  A review on transfer learning approaches in brain–computer interface , 2018 .

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

[25]  K.-R. Muller,et al.  BCI meeting 2005-workshop on BCI signal processing: feature extraction and translation , 2006, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[26]  Ya-Ju Fan,et al.  On the Time Series $K$-Nearest Neighbor Classification of Abnormal Brain Activity , 2007, IEEE Transactions on Systems, Man, and Cybernetics - Part A: Systems and Humans.

[27]  Fabien Lotte,et al.  Why standard brain-computer interface (BCI) training protocols should be changed: an experimental study , 2016, Journal of neural engineering.

[28]  S. Chiba,et al.  Dynamic programming algorithm optimization for spoken word recognition , 1978 .

[29]  Wojciech Samek,et al.  Transferring Subspaces Between Subjects in Brain--Computer Interfacing , 2012, IEEE Transactions on Biomedical Engineering.

[30]  F. Itakura,et al.  Minimum prediction residual principle applied to speech recognition , 1975 .