EEG-Based Emotion Classification Using Stacking Ensemble Approach

被引:28
作者
Chatterjee, Subhajit [1 ]
Byun, Yung-Cheol [2 ]
机构
[1] Jeju Natl Univ, Dept Comp Engn, Jeju 63243, South Korea
[2] Jeju Natl Univ, Inst Informat Sci & Technol, Dept Comp Engn, Elect Engn, Jeju 63243, South Korea
关键词
deep learning; emotion classification; EEG data; stacking ensemble classifier; random forest; lightGBM; gradient boosting classifier; BRAIN ACTIVITY; RECOGNITION; FOREST;
D O I
10.3390/s22218550
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
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.
引用
收藏
页数:15
相关论文
共 46 条
[1]  
Alhagry S, 2017, INT J ADV COMPUT SC, V8, P355, DOI 10.14569/IJACSA.2017.081046
[2]   A novel sample and feature dependent ensemble approach for Parkinson's disease detection [J].
Ali, Liaqat ;
Chakraborty, Chinmay ;
He, Zhiquan ;
Cao, Wenming ;
Imrana, Yakubu ;
Rodrigues, Joel J. P. C. .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (22) :15997-16010
[3]   A real-time automated system for the recognition of human facial expressions [J].
Anderson, K ;
McOwan, PW .
IEEE TRANSACTIONS ON SYSTEMS MAN AND CYBERNETICS PART B-CYBERNETICS, 2006, 36 (01) :96-105
[4]  
[Anonymous], ENSEMBLE METHODSBAGG
[5]  
[Anonymous], 1999, P ART NEUR NETW ENG
[6]  
[Anonymous], 2005, ELECTROEN CLIN NEURO
[7]  
[Anonymous], 2009, P 1 ACM INT WORKSH M, DOI [10.1145/1631111.1631118, DOI 10.1145/1631111.1631118]
[8]   Random forest in remote sensing: A review of applications and future directions [J].
Belgiu, Mariana ;
Dragut, Lucian .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2016, 114 :24-31
[9]   A random forest guided tour [J].
Biau, Gerard ;
Scornet, Erwan .
TEST, 2016, 25 (02) :197-227
[10]  
Bird J.J., 2019, Proceedings of theInternational Conference on Digital Image and Signal Processing (DISP19)