SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging.
Details
Serval ID
serval:BIB_77F4DF9FCB07
Type
Article: article from journal or magazin.
Collection
Publications
Institution
Title
SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging.
Journal
IEEE transactions on neural systems and rehabilitation engineering
ISSN
1558-0210 (Electronic)
ISSN-L
1534-4320
Publication state
Published
Issued date
03/2019
Peer-reviewed
Oui
Volume
27
Number
3
Pages
400-410
Language
english
Notes
Publication types: Journal Article ; Research Support, Non-U.S. Gov't
Publication Status: ppublish
Publication Status: ppublish
Abstract
Automatic sleep staging has been often treated as a simple classification problem that aims at determining the label of individual target polysomnography epochs one at a time. In this paper, we tackle the task as a sequence-to-sequence classification problem that receives a sequence of multiple epochs as input and classifies all of their labels at once. For this purpose, we propose a hierarchical recurrent neural network named SeqSleepNet (source code is available at http://github.com/pquochuy/SeqSleepNet). At the epoch processing level, the network consists of a filterbank layer tailored to learn frequency-domain filters for preprocessing and an attention-based recurrent layer designed for short-term sequential modeling. At the sequence processing level, a recurrent layer placed on top of the learned epoch-wise features for long-term modeling of sequential epochs. The classification is then carried out on the output vectors at every time step of the top recurrent layer to produce the sequence of output labels. Despite being hierarchical, we present a strategy to train the network in an end-to-end fashion. We show that the proposed network outperforms the state-of-the-art approaches, achieving an overall accuracy, macro F1-score, and Cohen's kappa of 87.1%, 83.3%, and 0.815 on a publicly available dataset with 200 subjects.
Keywords
Algorithms, Attention, Databases, Factual, Electroencephalography/statistics & numerical data, Electromyography, Electrooculography, Humans, Machine Learning, Neural Networks, Computer, Polysomnography/statistics & numerical data, Reproducibility of Results, Sleep Stages/physiology, Software
Pubmed
Web of science
Create date
11/01/2024 18:05
Last modification date
18/01/2024 14:53