Meng M, Yin X, She Q, Gao Y, Kong W, Luo Z. Sparse representation-based classification with two-dimensional dictionary optimization for motor imagery EEG pattern recognition.
J Neurosci Methods 2021;
361:109274. [PMID:
34229027 DOI:
10.1016/j.jneumeth.2021.109274]
[Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Journal Information] [Subscribe] [Scholar Register] [Received: 03/07/2021] [Revised: 06/19/2021] [Accepted: 07/01/2021] [Indexed: 11/25/2022]
Abstract
BACKGROUND
Sparse representation-based classification (SRC) has more advantages in motor imagery EEG pattern recognition, and the quality of dictionary construction directly determines the performance of SRC. In this paper, we proposed a two-dimensional dictionary optimization (TDDO) method to directly improve the performance of SRC.
NEW METHOD
Firstly, an initial dictionary was constructed with multi-band features extracted by filter band common spatial pattern (FBCSP). Then Lasso regression is used to select significant features in each atom synchronously in the horizontal direction, and the KNN-based method is used to clean up noise atoms in the vertical direction. Finally, an SRC method by training samples linearly representing test samples was implemented in classification.
RESULTS
The results show the necessity and rationality of TDDO-SRC method. The highest average classification accuracy of 86.5% and 92.4% is obtained on two public datasets.
COMPARISON WITH EXISTING METHOD(S)
The proposed method has more superior classification accuracy compared to traditional methods and existing winners' methods.
CONCLUSIONS
The quality of dictionary construction has a great impact on the robustness of SRC. And compared with the original SRC, the classification accuracy of the optimized TDDO-SRC is greatly improved.
Collapse