An Attention-Based Deep Learning Approach for Sleep Stage Classification With Single-Channel EEG

Citations Over TimeTop 1% of 2021 papers

Abstract

Automatic sleep stage mymargin classification is of great importance to measure sleep quality. In this paper, we propose a novel attention-based deep learning architecture called AttnSleep to classify sleep stages using single channel EEG signals. This architecture starts with the feature extraction module based on multi-resolution convolutional neural network (MRCNN) and adaptive feature recalibration (AFR). The MRCNN can extract low and high frequency features and the AFR is able to improve the quality of the extracted features by modeling the inter-dependencies between the features. The second module is the temporal context encoder (TCE) that leverages a multi-head attention mechanism to capture the temporal dependencies among the extracted features. Particularly, the multi-head attention deploys causal convolutions to model the temporal relations in the input features. We evaluate the performance of our proposed AttnSleep model using three public datasets. The results show that our AttnSleep outperforms state-of-the-art techniques in terms of different evaluation metrics. Our source codes, experimental data, and supplementary materials are available at https://github.com/emadeldeen24/AttnSleep.

Related Papers

• → Sleep Stage Estimation by Non-invasive Bio-measurement(2006)1 cited
• Auto sleep staging and sleep quality estimation(2004)
• → Survey on Sleep Stage Scoring Techniques using Feature Discovery(2020)
• Pre-sleep/Sleep/Post-sleep stages(2021)
• → P051 Evaluating sleep stage concordance of CardioRespiratary Sleep Staging (CReSS) on level III portable sleep study via simultaneous deployment of polysomnograph and level III portable sleep study(2022)