Brain-Computer Interface with Corrupted EEG Data: a Tensor Completion Approach

Abstract

One of the current issues in brain-computer interface (BCI) is how to deal with noisy electroencephalography (EEG) measurements organized as multidimensional datasets (tensors). On the other hand, recently, significant advances have been made in multidimensional signal completion algorithms that exploit tensor decomposition models to capture the intricate relationship among entries in a multidimensional signal. We propose to use tensor completion applied to EEG data for improving the classification performance in a motor imagery BCI system with corrupted measurements. Noisy measurements (electrode misconnections, subject movements, etc.) are considered as unknowns (missing samples) that are inferred from a tensor decomposition model (tensor completion). We evaluate the performance of four recently proposed tensor completion algorithms, CP-WOPT (Acar et al. Chemom Intell Lab Syst. 106:41-56, 2011), 3DPB-TC (Caiafa et al. 2013), BCPF (Zhao et al. IEEE Trans Pattern Anal Mach Intell. 37(9):1751-1763, 2015), and HaLRT (Liu et al. IEEE Trans Pattern Anal Mach Intell. 35(1):208-220, 2013), plus a simple interpolation strategy, first with random missing entries and then with missing samples constrained to have a specific structure (random missing channels), which is a more realistic assumption in BCI applications. We measured the ability of these algorithms to reconstruct the tensor from observed data. Then, we tested the classification accuracy of imagined movement in a BCI experiment with missing samples. We show that for random missing entries, all tensor completion algorithms can recover missing samples increasing the classification performance compared to a simple interpolation approach. For the random missing channels case, we show that tensor completion algorithms help to reconstruct missing channels, significantly improving the accuracy in the classification of motor imagery (MI), however, not at the same level as clean data. Summarizing, compared to the interpolation case, all tensor completion algorithms succeed to increase the classification performance by 7–9% (LDA–SVD) for random missing entries and 15–8% (LDA–SVD) for random missing channels. Tensor completion algorithms are useful in real BCI applications. The proposed strategy could allow using motor imagery BCI systems even when EEG data is highly affected by missing channels and/or samples, avoiding the need of new acquisitions in the calibration stage.