Electroencephalogram classification in motor-imagery brain–computer interface applications based on double-constraint nonnegative matrix factorization

Abstract

Objective: Brain–computer interfaces (BCIs) are aimed at providing a new way of communication between the human brain and external devices. One of the major tasks associated with the BCI system is to improve classification performance of the motor imagery (MI) signal. Electroencephalogram (EEG) signals are widely used for the MI BCI system. The raw EEG signals are usually non-stationary time series with weak class properties, degrading the classification performance. Approach: Nonnegative matrix factorization (NMF) has been successfully applied to pattern extraction which provides meaningful data presentation. However, NMF is unsupervised and cannot make use of the label information. Based on the label information of MI EEG data, we propose a novel method, called double-constrained nonnegative matrix factorization (DCNMF), to improve the classification performance of NMF on MI BCI. The proposed method constructs a couple of label matrices as the constraints on the NMF procedure to make the EEGs with the same class labels have the similar representation in the low-dimensional space, while the EEGs with different class labels have dissimilar representations as much as possible. Accordingly, the extracted features obtain obvious class properties, which are optimal to the classification of MI EEG. Main results: This study is conducted on the BCI competition III datasets (I and IVa). The proposed method helps to achieve a higher average accuracy across two datasets (79.00% for dataset I, 77.78% for dataset IVa); its performance is about 10% better than the existing studies in the literature. Significance: Our study provides a novel solution for MI BCI analysis from the perspective of label constraint; it provides convenience for semi-supervised learning of features and significantly improves the classification performance.