EEG-Based Emotion Classification Using Stacking Ensemble Approach

Sensors (Basel). 2022 Nov 6;22(21):8550. doi: 10.3390/s22218550.

Abstract

Rapid advancements in the medical field have drawn much attention to automatic emotion classification from EEG data. People's emotional states are crucial factors in how they behave and interact physiologically. The diagnosis of patients' mental disorders is one potential medical use. When feeling well, people work and communicate more effectively. Negative emotions can be detrimental to both physical and mental health. Many earlier studies that investigated the use of the electroencephalogram (EEG) for emotion classification have focused on collecting data from the whole brain because of the rapidly developing science of machine learning. However, researchers cannot understand how various emotional states and EEG traits are related. This work seeks to classify EEG signals' positive, negative, and neutral emotional states by using a stacking-ensemble-based classification model that boosts accuracy to increase the efficacy of emotion classification using EEG. The selected features are used to train a model that was created using a random forest, light gradient boosting machine, and gradient-boosting-based stacking ensemble classifier (RLGB-SE), where the base classifiers random forest (RF), light gradient boosting machine (LightGBM), and gradient boosting classifier (GBC) were used at level 0. The meta classifier (RF) at level 1 is trained using the results from each base classifier to acquire the final predictions. The suggested ensemble model achieves a greater classification accuracy of 99.55%. Additionally, while comparing performance indices, the suggested technique outperforms as compared with the base classifiers. Comparing the proposed stacking strategy to state-of-the-art techniques, it can be seen that the performance for emotion categorization is promising.

Keywords: EEG data; deep learning; emotion classification; gradient boosting classifier; lightGBM; random forest; stacking ensemble classifier.

MeSH terms

  • Brain
  • Electroencephalography* / methods
  • Emotions* / physiology
  • Humans
  • Machine Learning