Convolution- and Attention-Based Neural Network for Automated Sleep Stage Classification

Tianqi Zhu, Wei Luo and Feng Yu
Additional contact information
Tianqi Zhu: College of Biomedical Engineering and Instrument Science, Zhejiang University, Hangzhou 310027, China
Wei Luo: College of Biomedical Engineering and Instrument Science, Zhejiang University, Hangzhou 310027, China
Feng Yu: College of Biomedical Engineering and Instrument Science, Zhejiang University, Hangzhou 310027, China

IJERPH, 2020, vol. 17, issue 11, 1-13

Abstract: Analyzing polysomnography (PSG) is an effective method for evaluating sleep health; however, the sleep stage scoring required for PSG analysis is a time-consuming effort for an experienced medical expert. When scoring sleep epochs, experts pay attention to find specific signal characteristics (e.g., K-complexes and spindles), and sometimes need to integrate information from preceding and subsequent epochs in order to make a decision. To imitate this process and to build a more interpretable deep learning model, we propose a neural network based on a convolutional network (CNN) and attention mechanism to perform automatic sleep staging. The CNN learns local signal characteristics, and the attention mechanism excels in learning inter- and intra-epoch features. In experiments on the public sleep-edf and sleep-edfx databases with different training and testing set partitioning methods, our model achieved overall accuracies of 93.7% and 82.8%, and macro-average F1-scores of 84.5 and 77.8, respectively, outperforming recently reported machine learning-based methods.

Keywords: sleep stage classification; convolutional neural network; attention mechanism (search for similar items in EconPapers)
JEL-codes: I I1 I3 Q Q5 (search for similar items in EconPapers)
Date: 2020
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (2)

Downloads: (external link)
https://www.mdpi.com/1660-4601/17/11/4152/pdf (application/pdf)
https://www.mdpi.com/1660-4601/17/11/4152/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jijerp:v:17:y:2020:i:11:p:4152-:d:369793

Access Statistics for this article

IJERPH is currently edited by Ms. Jenna Liu

More articles in IJERPH from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().