1. Transfer Learning and SpecAugment applied to SSVEP Based BCI Classification
- Author
-
Pedro R. A. S. Bassi, Willian Rampazzo, and Romis Attux
- Subjects
Signal Processing (eess.SP) ,FOS: Computer and information sciences ,Computer Science - Machine Learning ,Computer science ,Computer Science - Artificial Intelligence ,0206 medical engineering ,Biomedical Engineering ,Health Informatics ,Context (language use) ,02 engineering and technology ,Electroencephalography ,Convolutional neural network ,Machine Learning (cs.LG) ,03 medical and health sciences ,0302 clinical medicine ,medicine ,FOS: Electrical engineering, electronic engineering, information engineering ,Electrical Engineering and Systems Science - Signal Processing ,Brain–computer interface ,Small data ,medicine.diagnostic_test ,business.industry ,Image and Video Processing (eess.IV) ,Pattern recognition ,Electrical Engineering and Systems Science - Image and Video Processing ,Filter bank ,020601 biomedical engineering ,Support vector machine ,Artificial Intelligence (cs.AI) ,Signal Processing ,Spectrogram ,Artificial intelligence ,business ,Transfer of learning ,030217 neurology & neurosurgery - Abstract
Objective We used deep convolutional neural networks (DCNNs) to classify electroencephalography (EEG) signals in a steady-state visually evoked potentials (SSVEP) based single-channel brain–computer interface (BCI), which does not require calibration on the user. Methods EEG signals were converted to spectrograms and served as input to train DCNNs using the transfer learning technique. We also modified and applied a data augmentation method, SpecAugment, generally employed for speech recognition. Furthermore, for comparison purposes, we classified the SSVEP dataset using support-vector machines (SVMs) and filter bank canonical correlation analysis (FBCCA). Results Excluding the evaluated user's data from the fine-tuning process, we reached 82.2% mean test accuracy and 0.825 mean F1-Score on 35 subjects from an open dataset, using a small data length (0.5 s), only one electrode (Oz) and the DCNN with transfer learning, window slicing (WS) and SpecAugment's time masks. Conclusion The DCNN results surpassed SVM and FBCCA performances, using a single electrode and a small data length. Transfer learning provided minimal accuracy change, but made training faster. SpecAugment created a small performance improvement and was successfully combined with WS, yielding higher accuracies. Significance We present a new methodology to solve the problem of SSVEP classification using DCNNs. We also modified a speech recognition data augmentation technique and applied it to the context of BCIs. The presented methodology surpassed performances obtained with FBCCA and SVMs (more traditional SSVEP classification methods) in BCIs with small data lengths and one electrode. This type of BCI can be used to develop small and fast systems.
- Published
- 2020