IMGWOFS: A Feature Selector with Trade-off Between Conflict Objectives for EEG-based Emotion Recognition

Gang Luo,Shuting Sun,Chang Yan,Shanshan Qu,Dixin Wang,Na Chu,Xuesong Liu,Fuze Tian,Kun Qian,Xiaowei Li,Bin Hu
DOI: https://doi.org/10.1109/taffc.2024.3450573
IF: 13.99
2024-01-01
IEEE Transactions on Affective Computing
Abstract:Feature selection is a crucial step in EEG emotion recognition. However, it was often used as a single objective problem to either reduce the number of features or maximize classification accuracy, while neglecting their balance. To address the issue, we proposed Improved Multi-objective Grey Wolf Optimization Feature Selection (IMGWOFS). Firstly, we designed a population initialization operator via discriminability and independence of features to accelerate search speed. Secondly, we employed a two-stage update strategy to improve the global search capabilities of the EEG feature subsets. Finally, we incorporated an adaptive mutation operator to escape the local optima. We conducted experiments on SEED and DEAP datasets, and the accuracy were 86.87 $\pm$ 1.62 % and 60.65 $\pm$ 1.51 % in the beta band using a smaller number of EEG features. In addition, the frontal lobe was related to emotion processing. In conclusion, IMGWOFS is an effective and feasible feature selection method for EEG-based emotion recognition.
What problem does this paper attempt to address?