Li Z, Zhang R, Zeng Y, Tong L, Lu R, Yan B. MST-net: A multi-scale swin transformer network for EEG-based cognitive load assessment.
Brain Res Bull 2024;
206:110834. [PMID:
38049039 DOI:
10.1016/j.brainresbull.2023.110834]
[Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [What about the content of this article? (0)] [Affiliation(s)] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 07/20/2023] [Revised: 10/22/2023] [Accepted: 11/29/2023] [Indexed: 12/06/2023]
Abstract
Cognitive load assessment plays a crucial role in monitoring safe production, resource allocation, and subjective initiative in human-computer interaction. Due to its high time resolution and convenient acquisition, Electroencephalography (EEG) is widely applied in brain monitoring and cognitive state assessment. In this study, a multi-scale Swin Transformer network (MST-Net) was proposed for cognitive load assessment, which extracts local features with different sensory fields using a multi-scale parallel convolution model and introduces the attention mechanism of the Swin Transformer to obtain the feature correlations among multi-scale local features. The performance of the proposed network was validated using the EEG signals collected during cognitive tasks and N-back tasks with three different load levels. Results show that the MST-Net network achieved the best classification accuracy on both local and public datasets, and was higher than the mainstream Swin Transformer and CNN. Furthermore, results of ablation experiments and feature visualization revealed that the proposed MST-Net could well characterize different cognitive loads, which not only provided novel and powerful tools for cognitive load assessment but also showed potential for broad application in brain-computer interface (BCI) systems.
Collapse