Multi-objective squirrel search algorithm for EEG feature selection

Abstract

Feature selection plays a critical role in the application of Brain Computer Interface (BCI) systems. Many methods have been used to solve the feature selection problem, but they model it as a single-objective problem, considering only classification accuracy or number of features. To close this critical gap, we improve the squirrel search algorithm by combining it with the grid method, and propose a Multi-Objective Squirrel Search Algorithm (MOSSA) to solve the feature selection problem in BCI. We conduct experiments on three publicly available motion imagery datasets, and the experimental results reveal the best classification results of the method on dataset 1. The average classification accuracy of dataset 2 is 96.71%, with the number of selected features reduced to 18 on average. The highest classification accuracy of dataset 3 is 83.57% on the training set and 82.86% on the test set. In addition, we compare MOSSA with other algorithms and the results show the superiority of our proposed method in solving the feature selection problem. Finally, we combine MOSSA with an online application of BCI, where subjects visualize controlling the robot to perform the corresponding actions by the left and right hand movements. The average recognition rate of the three subjects is approximately 70%. In summary, the MOSSA is an effective method for solving the feature selection problem and is useful for the development of online applications of BCI.