Design of a Cascade Stacking Ensemble Algorithm Based on Multi -Domain Feature Fusion of MI-EEG

Yue Zhang,Weihai Chen,Chu-Liang Lin,Zhongcai Pei,Jianer Chen,Hongfei Shi
DOI: https://doi.org/10.1109/iciea54703.2022.10006088
2022-01-01
Abstract:Electroencephalogram (EEG) signal is a complex bioelectrical signal with non-stationary, nonlinear, high dimension, and low signal-to-noise ratio. Single time domain or frequency domain analysis has great limitations, and the wavelet transform combined with time -frequency analysis cannot completely distinguish the artefact components like the EEG signals. All these problems will reduce the classification recognition rate. In view of this, an ensemble learning algorithm based on multi-domain features is proposed to realize the classification of MI-EEG. Four complementary features of time domain, frequency domain, spatial domain and time-frequency feature are calculated and fused as feature vectors, and then classified by an ensemble algorithm. Then, a cascade classification model based on a Stacking integrated algorithm is proposed. The first level is composed of Random Forest (RF), Gradient Boosting Decision Tree (GBDT) and Extreme Gradient Boosting (XGBoost) model, and the second level is composed of logistic regression (LR) to output the results. The output results show that the classification results of fusion features are significantly improved compared with those of single features. the accuracy of the Stacking integrated algorithm is up to 90.12%. Compared with a single classification algorithm, the average classification accuracy is improved by 3%.
What problem does this paper attempt to address?