GCD-JFSE: Graph-based Class-Domain Knowledge Joint Feature Selection and Ensemble Learning for EEG-based Emotion Recognition

Gang Luo,Yutong Han,Weichu Xie,Fuze Tian,Lixian Zhu,Kun Qian,Xiaowei Li,Shuting Sun,Bin Hu
DOI: https://doi.org/10.1016/j.knosys.2024.112770
IF: 8.139
2024-01-01
Knowledge-Based Systems
Abstract:Feature selection has demonstrated strong performance in emotion recognition using intrasubject electroencephalography (EEG) data. However, it faces challenges due to individual differences and the nonstationarity of EEG signals in cross-subject and cross-session emotion recognition. Currently, research on incorporating domain information into feature selection for cross-domain (subject or session) emotion recognition remains limited. To address this issue, we propose a graph-based class-domain knowledge joint feature selection and ensemble learning approach. Firstly, an undirected, fully connected weighted graph is constructed to capture the relationship between features. Then, some metrics such as domain scatter, domain correlation, and domain standard deviation are introduced to guide feature selection. Subsequently, soft voting ensemble learning is employed to enhance recognition performance. To validate the effectiveness of our method, we conduct experiments on public datasets (SEED, SEED_IV, DREAMER), achieving accuracies of 78.67% on SEED, 58.98% on SEED_IV, 61.11% of valence and 72.46% of arousal on DREAMER in a cross-subject scenario. In the cross-session scenario, we obtain 87.11% on SEED and 60.74% on SEED_IV. The proposed method outperforms state-of-the-art approaches. This study not only expands the application of feature selection in emotion recognition but also provides a potential strategy to enhance the performance of real-world EEG-based emotion recognition applications.
What problem does this paper attempt to address?