Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Automated Sleep Staging via Parallel Frequency-Cut Attention
(Submitted on 7 Apr 2022 (v1), last revised 12 Jan 2023 (this version, v3))
Abstract: This paper proposes a novel framework for automatically capturing the time-frequency nature of electroencephalogram (EEG) signals of human sleep based on the authoritative sleep medicine guidance. The framework consists of two parts: the first part extracts informative features by partitioning the input EEG spectrograms into a sequence of time-frequency patches. The second part is constituted by an attention-based architecture to efficiently search for the correlation between partitioned time-frequency patches and defining factors of sleep stages in parallel. The proposed pipeline is validated on the Sleep Heart Health Study dataset with new state-of-the-art results for the stages wake, N2, and N3, obtaining respective F1 scores of 0.93, 0.88, and 0.87, with only EEG signals used. The proposed method also has a high inter-rater reliability of 0.80 kappa. We also visualize the correspondence between sleep staging decisions and features extracted by the proposed method, providing strong interpretability for our model.
Submission history
From: Zheng Chen [view email][v1] Thu, 7 Apr 2022 02:48:13 GMT (4603kb,D)
[v2] Sun, 26 Jun 2022 08:33:34 GMT (24534kb,D)
[v3] Thu, 12 Jan 2023 07:46:13 GMT (36435kb,D)
Link back to: arXiv, form interface, contact.