PSSS-EEG: A Probabilistic-masking Self-Supervised Swin-transformer model for EEG-based drowsiness recognition

Jiaming Zhang,Fangzuo Zhang,Hongtao Wei
DOI: https://doi.org/10.1016/j.patcog.2024.111005
IF: 8
2024-09-22
Pattern Recognition
Abstract:Drowsiness recognition based on Electroencephalography (EEG) signals is a critical task with wide-ranging applications in areas such as transportation safety. However, existing methods for EEG-based drowsiness recognition have demonstrated limited performance in data-limited scenarios and struggle with capturing the diverse patterns among different subjects. In this paper, we propose a novel approach, the Probabilistic-masking Self-Supervised Swin-transformer model (PSSS-EEG), for accurate and robust EEG-based drowsiness recognition. The PSSS-EEG model integrates data augmentation and self-supervised learning techniques to improve the recognition accuracy and generalization ability of drowsiness recognition. Specifically, we introduce the incomplete probabilistic-masking operation to the data augmentation. Moreover, the swin-transformer architecture is employed as the backbone of the self-supervised network to capture long-range dependencies at different scales in EEG data. The retention weights are introduced into the loss function as a linkage between data augmentation and self-supervised network. Experimental results demonstrate that our model outperforms existing methods in EEG-based drowsiness recognition accuracy, achieving an accuracy exceeding 80%. In addition, this work highlights the importance of advanced data augmentation techniques and self-supervised learning models. It also contributes to filling a crucial gap in the existing literature about the integration of data augmentation and self-supervised learning, providing a more effective and adaptable solution for EEG-based recognition.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?