Augmenting Electroencephalogram Transformer for Steady-State Visually Evoked Potential-Based Brain-Computer Interfaces.

Abstract

Objective: Advancing high-speed steady-state visually evoked potential (SSVEP)-based brain-computer interface (BCI) systems requires effective electroencephalogram (EEG) decoding through deep learning. However, challenges persist due to data sparsity and the unclear neural basis of most augmentation techniques. Furthermore, effective processing of dynamic EEG signals and accommodating augmented data require a more sophisticated model tailored to the unique characteristics of EEG signals. Approach: This study introduces background EEG mixing (BGMix), a novel data augmentation technique grounded in neural principles that enhances training samples by replacing background noise between different classes. Building on this, we propose the augment EEG Transformer (AETF), a Transformer-based model designed to capture the temporal, spatial, and frequential features of EEG signals, leveraging the advantages of Transformer architectures. Main results: Experimental evaluations of 2 publicly available SSVEP datasets show the efficacy of the BGMix strategy and the AETF model. The BGMix approach notably improved the average classification accuracy of 4 distinct deep learning models, with increases ranging from 11.06% to 21.39% and 4.81% to 25.17% in the respective datasets. Furthermore, the AETF model outperformed state-of-the-art baseline models, excelling with short training data lengths and achieving the highest information transfer rates (ITRs) of 205.82 ± 15.81 bits/min and 240.03 ± 14.91 bits/min on the 2 datasets. Significance: This study introduces a novel EEG augmentation method and a new approach to designing deep learning models informed by the neural processes of EEG. These innovations significantly improve the performance and practicality of high-speed SSVEP-based BCI systems.