Emotion Recognition From Multi-Channel EEG via Deep Forest

Juan Cheng,Meiyao Chen,Chang Li,Yu Liu,Rencheng Song,Aiping Liu,Xun Chen
DOI: https://doi.org/10.1109/jbhi.2020.2995767
IF: 7.7
2021-02-01
IEEE Journal of Biomedical and Health Informatics
Abstract:Recently, deep neural networks (DNNs) have been applied to emotion recognition tasks based on electroencephalography (EEG), and have achieved better performance than traditional algorithms. However, DNNs still have the disadvantages of too many hyperparameters and lots of training data. To overcome these shortcomings, in this article, we propose a method for multi-channel EEG-based emotion recognition using deep forest. First, we consider the effect of baseline signal to preprocess the raw artifact-eliminated EEG signal with baseline removal. Secondly, we construct 2$D$ frame sequences by taking the spatial position relationship across channels into account. Finally, 2$D$ frame sequences are input into the classification model constructed by deep forest that can mine the spatial and temporal information of EEG signals to classify EEG emotions. The proposed method can eliminate the need for feature extraction in traditional methods and the classification model is insensitive to hyperparameter settings, which greatly reduce the complexity of emotion recognition. To verify the feasibility of the proposed model, experiments were conducted on two public DEAP and DREAMER databases. On the DEAP database, the average accuracies reach to 97.69% and 97.53% for valence and arousal, respectively; on the DREAMER database, the average accuracies reach to 89.03%, 90.41%, and 89.89% for valence, arousal and dominance, respectively. These results show that the proposed method exhibits higher accuracy than the state-of-art methods.
computer science, interdisciplinary applications,mathematical & computational biology,medical informatics, information systems
What problem does this paper attempt to address?
The paper mainly addresses the task of emotion recognition based on electroencephalogram (EEG) and attempts to solve the following key issues: 1. **Improving the limitations of traditional methods**: The paper points out that traditional feature extraction methods only focus on time-domain or frequency-domain information, and traditional machine learning algorithms (such as decision trees, support vector machines, etc.) have limited generalization ability when dealing with complex EEG signals. 2. **Overcoming the drawbacks of deep neural networks (DNNs)**: Although DNNs have shown significant advantages in EEG-based emotion recognition tasks, they still require extensive hyperparameter tuning and a large amount of training data. However, in the field of EEG-based emotion recognition, large-scale labeled databases are relatively limited. 3. **Proposing a new method**: To overcome the above challenges, the paper proposes a method using deep forest for multi-channel EEG emotion recognition. Specifically, a deep forest model named Multi-Grained Cascade Forest (gcForest) is adopted and customized according to the characteristics of multi-channel EEG. gcForest has features such as fewer hyperparameters, no need for backpropagation, and insensitivity to hyperparameter settings. Its model complexity depends on the data scale, making it suitable for training on datasets of different sizes. In summary, the core objective of this paper is to improve the performance of EEG-based emotion recognition by proposing a new method based on deep forest, particularly addressing the limitations of traditional methods and deep neural network methods. The effectiveness and superiority of the proposed method are validated through experiments on two public databases (DEAP and DREAMER).