A regression method for EEG-based cross-dataset fatigue detection

被引:8
作者
Yuan, Duanyang [1 ]
Yue, Jingwei [2 ]
Xiong, Xuefeng [1 ]
Jiang, Yibi [1 ]
Zan, Peng [1 ]
Li, Chunyong [2 ]
机构
[1] Shanghai Univ, Sch Mechatron Engn & Automat, Shanghai Key Lab Power Stn Automat, Shanghai, Peoples R China
[2] Beijing Inst Radiat Med, Acad Mil Med Sci AMMS, Beijing, Peoples R China
关键词
fatigue detection; cross-dataset; EEG; regression method; self-supervised learning;
D O I
10.3389/fphys.2023.1196919
中图分类号
Q4 [生理学];
学科分类号
071003 ;
摘要
Introduction: Fatigue is dangerous for certain jobs requiring continuous concentration. When faced with new datasets, the existing fatigue detection model needs a large amount of electroencephalogram (EEG) data for training, which is resource-consuming and impractical. Although the cross-dataset fatigue detection model does not need to be retrained, no one has studied this problem previously. Therefore, this study will focus on the design of the cross-dataset fatigue detection model.Methods: This study proposes a regression method for EEG-based cross-dataset fatigue detection. This method is similar to self-supervised learning and can be divided into two steps: pre-training and the domain-specific adaptive step. To extract specific features for different datasets, a pretext task is proposed to distinguish data on different datasets in the pre-training step. Then, in the domain-specific adaptation stage, these specific features are projected into a shared subspace. Moreover, the maximum mean discrepancy (MMD) is exploited to continuously narrow the differences in the subspace so that an inherent connection can be built between datasets. In addition, the attention mechanism is introduced to extract continuous information on spatial features, and the gated recurrent unit (GRU) is used to capture time series information.Results: The accuracy and root mean square error (RMSE) achieved by the proposed method are 59.10% and 0.27, respectively, which significantly outperforms state-of-the-art domain adaptation methods.Discussion: In addition, this study discusses the effect of labeled samples. When the number of labeled samples is 10% of the total number, the accuracy of the proposed model can reach 66.21%. This study fills a vacancy in the field of fatigue detection. In addition, the EEG-based cross-dataset fatigue detection method can be used for reference by other EEG-based deep learning research practices.
引用
收藏
页数:16
相关论文
共 45 条
[1]   Uncovering the structure of clinical EEG signals with self-supervised learning [J].
Banville, Hubert ;
Chehab, Omar ;
Hyvarinen, Aapo ;
Engemann, Denis-Alexander ;
Gramfort, Alexandre .
JOURNAL OF NEURAL ENGINEERING, 2021, 18 (04)
[2]   Eye Movement Analysis for Activity Recognition Using Electrooculography [J].
Bulling, Andreas ;
Ward, Jamie A. ;
Gellersen, Hans ;
Troester, Gerhard .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (04) :741-753
[3]   Multi-Source and Multi-Representation Adaptation for Cross-Domain Electroencephalography Emotion Recognition [J].
Cao, Jiangsheng ;
He, Xueqin ;
Yang, Chenhui ;
Chen, Sifang ;
Li, Zhangyu ;
Wang, Zhanxiang .
FRONTIERS IN PSYCHOLOGY, 2022, 12
[4]   Multi-channel EEG recordings during a sustained-attention driving task [J].
Cao, Zehong ;
Chuang, Chun-Hsiang ;
King, Jung-Kai ;
Lin, Chin-Teng .
SCIENTIFIC DATA, 2019, 6 (1) :19
[5]   MS-MDA: Multisource Marginal Distribution Adaptation for Cross-Subject and Cross-Session EEG Emotion Recognition [J].
Chen, Hao ;
Jin, Ming ;
Li, Zhunan ;
Fan, Cunhang ;
Li, Jinpeng ;
He, Huiguang .
FRONTIERS IN NEUROSCIENCE, 2021, 15
[6]   SCA-CNN: Spatial and Channel-wise Attention in Convolutional Networks for Image Captioning [J].
Chen, Long ;
Zhang, Hanwang ;
Xiao, Jun ;
Nie, Liqiang ;
Shao, Jian ;
Liu, Wei ;
Chua, Tat-Seng .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6298-6306
[7]   EEG emotion Enhancement using Task-specific Domain Adversarial Neural Network [J].
Ding, Ke-Ming ;
Kimura, Tsukasa ;
Fukui, Ken-ichi ;
Numao, Masayuki .
2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
[8]   A Product Fuzzy Convolutional Network for Detecting Driving Fatigue [J].
Du, Guanglong ;
Long, Shuaiying ;
Li, Chunquan ;
Wang, Zhiyao ;
Liu, Peter X. .
IEEE TRANSACTIONS ON CYBERNETICS, 2023, 53 (07) :4175-4188
[9]   ADAST: Attentive Cross-Domain EEG-Based Sleep Staging Framework With Iterative Self-Training [J].
Eldele, Emadeldeen ;
Ragab, Mohamed ;
Chen, Zhenghua ;
Wu, Min ;
Kwoh, Chee-Keong ;
Li, Xiaoli ;
Guan, Cuntai .
IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE, 2023, 7 (01) :210-221
[10]   An iterative cross-subject negative-unlabeled learning algorithm for quantifying passive fatigue [J].
Foong, Ruyi ;
Ang, Kai Keng ;
Zhang, Zhuo ;
Quek, Chai .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (05)