Automated Sleep Staging via Parallel Frequency-Cut Attention

Stage-based sleep screening is a widely-used tool in both healthcare and neuroscientific research, as it allows for the accurate assessment of sleep patterns and stages. In this paper, we propose a novel framework that is based on authoritative guidance in sleep medicine and is designed to automatically capture the time-frequency characteristics of sleep electroencephalogram (EEG) signals in order to make staging decisions. Our framework consists of two main phases: a feature extraction process that partitions the input EEG spectrograms into a sequence of time-frequency patches, and a staging phase that searches for correlations between the extracted features and the defining characteristics of sleep stages. To model the staging phase, we utilize a Transformer model with an attention-based module, which allows for the extraction of global contextual relevance among time-frequency patches and the use of this relevance for staging decisions. The proposed method is validated on the large-scale Sleep Heart Health Study dataset and achieves new state-of-the-art results for the wake, N2, and N3 stages, with respective F1 scores of 0.93, 0.88, and 0.87 using only EEG signals. Our method also demonstrates high inter-rater reliability, with a kappa score of 0.80. Moreover, we provide visualizations of the correspondence between sleep staging decisions and features extracted by our method, which enhances the interpretability of the proposal. Overall, our work represents a significant contribution to the field of automated sleep staging and has important implications for both healthcare and neuroscience research.

Medienart:

E-Artikel

Erscheinungsjahr:

2023

Erschienen:

2023

Enthalten in:

Zur Gesamtaufnahme - volume:31

Enthalten in:

IEEE transactions on neural systems and rehabilitation engineering : a publication of the IEEE Engineering in Medicine and Biology Society - 31(2023) vom: 08., Seite 1974-1985

Sprache:

Englisch

Beteiligte Personen:

Chen, Zheng [VerfasserIn]
Yang, Ziwei [VerfasserIn]
Zhu, Lingwei [VerfasserIn]
Chen, Wei [VerfasserIn]
Tamura, Toshiyo [VerfasserIn]
Ono, Naoaki [VerfasserIn]
Altaf-Ul-Amin, Md [VerfasserIn]
Kanaya, Shigehiko [VerfasserIn]
Huang, Ming [VerfasserIn]

Links:

Volltext

Themen:

Journal Article
Research Support, Non-U.S. Gov't

Anmerkungen:

Date Completed 14.04.2023

Date Revised 08.05.2023

published: Print-Electronic

Citation Status MEDLINE

doi:

10.1109/TNSRE.2023.3243589

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM355275198