机构:[1]School of Computer Science, Hubei University of Technology, Wuhan 430068, China[2]Wuhan Third Hospital (Tongren Hospital of Wuhan University), Wuhan 430074, China[3]Key Laboratory of Brain Machine Collaborative Intelligence of Zhejiang Province, Hangzhou 310018, China
Motor Imagery EEG (MI-EEG) classification plays an important role in different Brain Computer Interface (BCI) systems. Recently, deep learning has been widely used in the MI-EEG classification tasks, however this technology requires a large number of labeled training samples which are difficult to obtain, and insufficient labeled training samples will result in a degradation of the classification performance. To address the degradation problem, we investigate a Self-Supervised Learning (SSL) based MI-EEG classification method to reduce the dependence on a large number of labeled training samples. The proposed method includes a pretext task and a downstream classification one. In the pretext task, each MI-EEG is rearranged according to the temporal characteristic. A network is pre-trained using the original and rearranged MI-EEGs. In the downstream task, a MI EEG classification network is firstly initialized by the network learned in the pretext task and then trained using a small number of the labeled training samples. A series of experiments are conducted on Data sets 1 and 2b of BCI competition IV and IVa of BCI competition III. In the case of one third of the labeled training samples, the proposed method can obtain an obvious improvement compared to the baseline network without using SSL. In the experiments under different percentages of the labeled training samples, the results show that the designed SSL strategy is effective and beneficial to improving the classification performance.
基金:
High-level Talents Fund of Hubei University of Technology [GCRC2020016]; Key Laboratory of Brain Machine Collaborative Intelligence of Zhejiang Province [2020E10010-02]; Natural Science Foundation of Hubei Province [2021CFB282]; Open Funding Project of the State Key Laboratory of Biocatalysis and Enzyme Engineering [SKLBEE2021020, SKLBEE2020020]
第一作者机构:[1]School of Computer Science, Hubei University of Technology, Wuhan 430068, China
共同第一作者:
通讯作者:
通讯机构:[1]School of Computer Science, Hubei University of Technology, Wuhan 430068, China[3]Key Laboratory of Brain Machine Collaborative Intelligence of Zhejiang Province, Hangzhou 310018, China
推荐引用方式(GB/T 7714):
Ou Yanghan,Sun Siqin,Gan Haitao,et al.An improved self-supervised learning for EEG classification[J].MATHEMATICAL BIOSCIENCES AND ENGINEERING.2022,19(7):6907-6922.doi:10.3934/mbe.2022325.
APA:
Ou, Yanghan,Sun, Siqin,Gan, Haitao,Zhou, Ran&Yang, Zhi.(2022).An improved self-supervised learning for EEG classification.MATHEMATICAL BIOSCIENCES AND ENGINEERING,19,(7)
MLA:
Ou, Yanghan,et al."An improved self-supervised learning for EEG classification".MATHEMATICAL BIOSCIENCES AND ENGINEERING 19..7(2022):6907-6922