Cognitive state detection using electroencephalogram(EEG)signals for various tasks has attracted significant research attention.However,it is difficult to further improve the performance of crosssubject cognitive stat...Cognitive state detection using electroencephalogram(EEG)signals for various tasks has attracted significant research attention.However,it is difficult to further improve the performance of crosssubject cognitive state detection.Further,most of the existing deep learning models will degrade significantly when limited training samples are given,and the feature hierarchical relationships are ignored.To address the above challenges,we propose an efficient interpretation model based on multiple capsule networks for cross-subject EEG cognitive state detection,termed as Efficient EEG-based Multi-Capsule Framework(E3GCAPS).Specifically,we use a selfexpression module to capture the potential connections between samples,which is beneficial to alleviate the sensitivity of outliers that are caused by the individual differences of cross-subject EEG.In addition,considering the strong correlation between cognitive states and brain function connection mode,the dynamic subcapsule-based spatial attention mechanism is introduced to explore the spatial relationship of multi-channel 1D EEG data,in which multichannel 1D data greatly improving the training efficiency while preserving the model performance.The effectiveness of the E3GCAPS is validated on the Fatigue-Awake EEG Dataset(FAAD)and the SJTU Emotion EEG Dataset(SEED).Experimental results show E3GCAPS can achieve remarkable results on the EEG-based cross-subject cognitive state detection under different tasks.展开更多
基金supported by NSFC with grant No.62076083Firstly,the authors would like to express thanks to the Key Laboratory of Brain Machine Collaborative Intelligence of Zhejiang Province with grant No.2020E10010Industrial Neuroscience Laboratory of Sapienza University of Rome.
文摘Cognitive state detection using electroencephalogram(EEG)signals for various tasks has attracted significant research attention.However,it is difficult to further improve the performance of crosssubject cognitive state detection.Further,most of the existing deep learning models will degrade significantly when limited training samples are given,and the feature hierarchical relationships are ignored.To address the above challenges,we propose an efficient interpretation model based on multiple capsule networks for cross-subject EEG cognitive state detection,termed as Efficient EEG-based Multi-Capsule Framework(E3GCAPS).Specifically,we use a selfexpression module to capture the potential connections between samples,which is beneficial to alleviate the sensitivity of outliers that are caused by the individual differences of cross-subject EEG.In addition,considering the strong correlation between cognitive states and brain function connection mode,the dynamic subcapsule-based spatial attention mechanism is introduced to explore the spatial relationship of multi-channel 1D EEG data,in which multichannel 1D data greatly improving the training efficiency while preserving the model performance.The effectiveness of the E3GCAPS is validated on the Fatigue-Awake EEG Dataset(FAAD)and the SJTU Emotion EEG Dataset(SEED).Experimental results show E3GCAPS can achieve remarkable results on the EEG-based cross-subject cognitive state detection under different tasks.
