ERP-WGAN: A data augmentation method for EEG single-trial detection.

ABSTRACT

Brain computer interaction based on EEG presents great potential and becomes the research hotspots. However, the insufficient scale of EEG database limits the BCI system performance, especially the positive and negative sample imbalance caused by oddball paradigm. To alleviate the bottleneck problem of scarce EEG sample, we propose a data augmentation method based on generative adversarial network to improve the performance of EEG signal classification. Taking the characteristics of EEG into account in wasserstein generative adversarial networks (WGAN), the problems of model collapse and poor quality of artificial data were solved by using resting noise, smoothing and random amplitude. The quality of artificial data was comprehensively evaluated from verisimilitude, diversity and accuracy. Compared with the three artificial data methods and two data sampling methods, the proposed ERP-WGAN framework significantly improve the performance of both subject and general classifiers, especially the accuracy of general classifiers trained by less than 5 dimensional features is improved by 20-25%. Moreover, we evaluate the training sets performance with different mixing ratios of artificial and real samples. ERP-WGAN can reduced at least 73% of the real subject data and acquisition cost, which greatly saves the test cycle and research cost.