Please use this identifier to cite or link to this item:
Title: An attention-based deep learning approach for sleep stage classification with single-channel EEG
Authors: Eldele, Emadeldeen
Chen, Zhenghua
Liu, Chengyu
Wu, Min
Kwoh, Chee Keong
Li, Xiaoli
Guan, Cuntai
Keywords: Computer and Information Science
Issue Date: 2021
Source: Eldele, E., Chen, Z., Liu, C., Wu, M., Kwoh, C. K., Li, X. & Guan, C. (2021). An attention-based deep learning approach for sleep stage classification with single-channel EEG. IEEE Transactions On Neural Systems and Rehabilitation Engineering, 29, 809-818.
Journal: IEEE Transactions on Neural Systems and Rehabilitation Engineering
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
ISSN: 1534-4320
DOI: 10.1109/TNSRE.2021.3076234
DOI (Related Dataset): 10.21979/N9/EUHGHS
Schools: School of Computer Science and Engineering 
Rights: © 2021 The Author(s). Published by IEEE. This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see
Fulltext Permission: open
Fulltext Availability: With Fulltext
Appears in Collections:SCSE Journal Articles

Citations 1

Updated on Feb 21, 2024

Web of ScienceTM
Citations 5

Updated on Oct 24, 2023

Page view(s)

Updated on Feb 27, 2024

Download(s) 50

Updated on Feb 27, 2024

Google ScholarTM




Items in DR-NTU are protected by copyright, with all rights reserved, unless otherwise indicated.