Fatigue EEG Classification Study Based on Convolutionally Constrained Boltzmann Machine

Xiaoran Dong; Yahan Gao

doi:10.54097/krwvwm87

Authors

Xiaoran Dong
Yahan Gao

DOI:

https://doi.org/10.54097/krwvwm87

Keywords:

N-back, C-RBM, semi-supervised learning, EEG classification.

Abstract

As the demand for the detection of brain fatigue state grows, portable EEG instruments are becoming more and more critical in related research. In this study, we designed a 1-back task paradigm to induce fatigue EEG signals using the EmotivEpoc+14 EEG instrument. The EEG dataset was built based on the subjective rating scale rating scores of the KSS scale and the behavioral analysis results. Construct an improved convolutionally constrained Boltzmann machine model, introducing convolutional operations to the visible and hidden layers to achieve weight sharing. Feature selection using principal component analysis method combined with Pearson's coefficient to retain highly correlated features. Self-training-semi-supervised learning method is used to train the model, and the results show that the features extracted from the C-RBM model achieve 89% and 91% classification accuracy in the frontal and occipital lobes. After reducing the channels it is used for SVM and RF classifiers with the best results, SVM achieves 93% classification accuracy of HM + PSD + PE for occipital lobe and RF achieves 92% classification accuracy of HM + PSD + WE for occipital lobe. This shows that the combination of the C-RBM model proposed in this paper with SVM and RF classifiers can effectively use the reduced dimensionality channel features for fatigue detection, which provides a reference for fatigue detection of feature combinations for sparse channels.

Downloads

Download data is not yet available.

References

[1] MARTIN V P, LOPEZ R, DAUVILLIERS Y, et al. Sleepiness in adults: An umbrella review of a complex construct[J]. Sleep Med Rev, 2023,67: 101718.

[2] GONCALVES M T, MALAFAIA S, SANTOS J M D, et al. Epworth sleepiness scale : A meta-analytic study on the internal consistency[J]. SLEEP MEDICINE, 2023,109: 261-269.

[3] POPP R F, FIERLBECK A K, KNUTTEL H, et al. Daytime sleepiness versus fatigue in patients with multiple sclerosis: A systematic review on the Epworth sleepiness scale as an assessment tool[J]. Sleep Med Rev, 2017,32: 95-108.

[4] XIANG C, FAN X, BAI D, et al. A resting-state EEG dataset for sleep deprivation[J]. Sci Data, 2024,11(1): 427.

[5] ZHANG H, ZHOU Q, CHEN H, et al. The applied principles of EEG analysis methods in neuroscience and clinical neurology[J]. Mil Med Res, 2023,10(1): 67.

[6] Y. P, C. M W, Z. W, et al. Fatigue Detection in SSVEP-BCIs Based on Wavelet Entropy of EEG[J]. IEEE Access, 2021,9: 114905-114913.

[7] LUO H, QIU T, LIU C, et al. Research on fatigue driving detection using forehead EEG based on adaptive multi-scale entropy[J]. Biomedical Signal Processing and Control, 2019,51: 50-58.

[8] HU J. Automated Detection of Driver Fatigue Based on AdaBoost Classifier with EEG Signals[J]. Front Comput Neurosci, 2017,11: 72.

[9] LIU Q, LIU Y, CHEN K, et al. Research on Channel Selection and Multi-Feature Fusion of EEG Signals for Mental Fatigue Detection[J]. Entropy (Basel), 2021,23(4).

[10] GUO H, CHEN S, ZHOU Y, et al. A hybrid critical channels and optimal feature subset selection framework for EEG fatigue recognition[J]. Sci Rep, 2025,15(1): 2139.

[11] HU J, MIN J. Automated detection of driver fatigue based on EEG signals using gradient boosting decision tree model[J]. COGNITIVE NEURODYNAMICS, 2018,12(4): 431-440.

[12] LAMICHHANE B, WESTBROOK A, COLE M W, et al. Exploring brain-behavior relationships in the N-back task[J]. Neuroimage, 2020,212: 116683.

[13] ELTRASS A S, GHANEM N H. A new automated multi-stage system of non-local means and multi-kernel adaptive filtering techniques for EEG noise and artifacts suppression[J]. J Neural Eng, 2021,18(3).

[14] RIAZ F, HASSAN A, REHMAN S, et al. EMD-Based Temporal and Spectral Features for the Classification of EEG Signals Using Supervised Learning[J]. IEEE Trans Neural Syst Rehabil Eng, 2016,24(1): 28-35.

[15] XU X, TANG J, XU T, et al. Mental Fatigue Degree Recognition Based on Relative Band Power and Fuzzy Entropy of EEG[J]. Int J Environ Res Public Health, 2023,20(2).

[16] CHADDAD A, WU Y, KATEB R, et al. Electroencephalography Signal Processing: A Comprehensive Review and Analysis of Methods and Techniques[J]. Sensors (Basel), 2023,23(14).

[17] YAN T, WANG D, XIA T, et al. Investigations on generalized Hjorth's parameters for machine performance degradation assessment[J]. Mechanical Systems and Signal Processing, 2022,168.

[18] HAG A, AL-SHARGIE F, HANDAYANI D, et al. Mental Stress Classification Based on Selected Electroencephalography Channels Using Correlation Coefficient of Hjorth Parameters [J]. Brain Sci, 2023,13(9).

[19] CIZMECI H, OZCAN C, DURGUT R. Channel selection and feature extraction on deep EEG classification using metaheuristic and Welch PSD[J]. Soft Computing, 2022,26(19): 10115-10125.

[20] ZHANG Y, LIU B, JI X, et al. Classification of EEG Signals Based on Autoregressive Model and Wavelet Packet Decomposition[J]. Neural Processing Letters, 2017,45(2): 365-378.

[21] YE B, QIU T, BAI X, et al. Research on Recognition Method of Driving Fatigue State Based on Sample Entropy and Kernel Principal Component Analysis[J]. Entropy (Basel), 2018,20(9).

[22] CHEN S, LUO Z, GAN H. An entropy fusion method for feature extraction of EEG[J]. Neural Computing and Applications, 2018,29(10): 857-863.

[23] LIU Y, XIANG Z, YAN Z, et al. CEEMDAN fuzzy entropy based fatigue driving detection using single-channel EEG[J]. Biomedical Signal Processing and Control, 2024,95.

[24] YE B, QIU T, BAI X, et al. Research on Recognition Method of Driving Fatigue State Based on Sample Entropy and Kernel Principal Component Analysis[J]. Entropy (Basel), 2018,20(9).

[25] PENG Y, WONG C M, WANG Z, et al. Fatigue Detection in SSVEP-BCIs Based on Wavelet Entropy of EEG[J]. IEEE Access, 2021,9: 114905-114913.

[26] ZHANG N, DING S, ZHANG J, et al. An overview on Restricted Boltzmann Machines[J]. Neurocomputing, 2018, 275: 1186-1199.

[27] F. Y, S. L, E. D, et al. How to Reduce Dimension With PCA and Random Projections?[J]. IEEE Transactions on Information Theory, 2021,67(12): 8154-8189.

[28] LAN Z, ZHAO J, LIU P, et al. Driving fatigue detection based on fusion of EEG and vehicle motion information[J]. Biomedical Signal Processing and Control, 2024,92.