Investigating the Effect of Flickering Frequency Pair and Mother Wavelet Selection in Steady-State Visually-Evoked Potentials on Two-Command Brain-Computer Interfaces

Abstract

IntroductionSteady-state visually evoked potentials (SSVEPs) have become popular in brain-computer interface (BCI) applications in addition to many other applications on clinical neuroscience (neurodegenerative disorders, schizophrenia, epilepsy, etc.), cognitive (visual attention, working memory, brain rhythms, etc.), and use of engineering researches. Among available methods to measure brain activities, SSVEPs have advantages like higher information transfer rate, simplicity in structure, and short training time. SSVEP-based BCIs use flickering stimuli at different frequencies to discriminate distinct commands in real life. Some features are extracted from the SSVEP signals before these commands are classified. The wavelet transform (WT) has attracted researchers among feature extraction methods since it utilizes the non-stationary signals well. In the WT, a sample function (named mother wavelet) represents the SSVEP signal in both time and frequency domains. Unfortunately, there is no universal mother wavelet function that fits all the signals. Therefore, choosing an appropriate mother wavelet function may be a challenge in WT-related studies. Although there are such studies in three- and seven-command SSVEP-based studies, there is no study for two-command systems in our knowledge. Materials and MethodsIn this study, two user commands flickered at the combinations of seven different frequencies were tested to determine which frequency pairs give the highest performance. For this purpose, three well-known wavelet features (energy, entropy, and variance) were calculated for each of derived EEG frequency bands from the discrete WT coefficients of SSVEP signals. The WT was repeated for six different mother wavelet functions (Haar, Db4, Sym4, Coif1, Bior3.5, and Rbior2.8). Then, four feature sets (every three features, and all together) were applied to seven commonly-used machine learning algorithms (Decision Tree, Discriminant Analysis, Logistic Regression, Naive Bayes, Support Vector Machines, Nearest Neighbors, and Ensemble Classifiers). Results and DiscussionWe achieved 100% accuracies among these 3,528 runs (7 classifiers x 4 feature sets x 6 mother wavelets x 21 flickering frequency pairs) using the mother wavelet function of Haar and the Ensemble Learner classifier. The highest classifier performances are 100% when two commands have the flickering frequency pairs of (6.0 and 10 Hz), (6.5 and 8.2 Hz), or (6.5 and 10.0 Hz). ConclusionWe obtained three main outcomes from this study. First, the most representative mother wavelet function was Haar, while the worst one was Symlet 4. Second, the Ensemble Learner classifier gave the maximum classifier performance in a two-command SSVEP-based BCI system. Besides, two user commands from SSVEP should be one of the frequency pairs of (6.0 and 10.0 Hz), (6.5 and 8.2 Hz), and (6.5 and 10.0 Hz) to achieve the maximum accuracy.