Classification of left-versus right-hand motor imagery in stroke patients using supplementary data generated by CycleGAN

Fangzhou Xu, Fenqi Rong, Jiancai Leng, Tao Sun, Yang Zhang, Siddharth Siddharth, Tzyy-Ping Jung
2021 IEEE transactions on neural systems and rehabilitation engineering  
Acquiring Electroencephalography (EEG) data is often time-consuming, laborious, and costly, posing practical challenges to train powerful but data-demanding deep learning models. This study proposes a surrogate EEG data-generation system based on cycle-consistent adversarial networks (CycleGAN) that can expand the number of training data. This study used EEG2Image based on a modified S-transform (MST) to convert EEG data into EEG-topography. This method retains the frequency-domain
more » ... cs and spatial information of the EEG signals. Then, the CycleGAN is used to learn and generate motor-imagery EEG data of stroke patients. From the visual inspection, there is no difference between the EEG topographies of the generated and original EEG data collected from the stroke patients. Finally, we used convolutional neural networks (CNN) to evaluate and analyze the generated EEG data. The experimental results show that the generated data effectively improved the classification accuracy.
doi:10.1109/tnsre.2021.3123969 pmid:34710045 fatcat:2bu77dpyd5h7dk725thaparypu