SeriesSleepNet: an EEG time series model with partial data augmentation for automatic sleep stage scoring

Minji Lee, Heon Gyu Kwak, Hyeong Jin Kim, Dong Ok Won, Seong Whan Lee

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

Introduction: We propose an automatic sleep stage scoring model, referred to as SeriesSleepNet, based on convolutional neural network (CNN) and bidirectional long short-term memory (bi-LSTM) with partial data augmentation. We used single-channel raw electroencephalography signals for automatic sleep stage scoring. Methods: Our framework was focused on time series information, so we applied partial data augmentation to learn the connected time information in small series. In specific, the CNN module learns the time information of one epoch (intra-epoch) whereas the bi-LSTM trains the sequential information between the adjacent epochs (inter-epoch). Note that the input of the bi-LSTM is the augmented CNN output. Moreover, the proposed loss function was used to fine-tune the model by providing additional weights. To validate the proposed framework, we conducted two experiments using the Sleep-EDF and SHHS datasets. Results and Discussion: The results achieved an overall accuracy of 0.87 and 0.84 and overall F1-score of 0.80 and 0.78 and kappa value of 0.81 and 0.78 for five-class classification, respectively. We showed that the SeriesSleepNet was superior to the baselines based on each component in the proposed framework. Our architecture also outperformed the state-of-the-art methods with overall F1-score, accuracy, and kappa value. Our framework could provide information on sleep disorders or quality of sleep to automatically classify sleep stages with high performance.

Original languageEnglish
Article number1188678
JournalFrontiers in Physiology
Volume14
DOIs
Publication statusPublished - 2023

Bibliographical note

Publisher Copyright:
Copyright © 2023 Lee, Kwak, Kim, Won and Lee.

Keywords

  • automatic sleep stage scoring
  • bi-directional long-short term memory
  • convolutional neural network
  • deep learning
  • single-channel EEG

ASJC Scopus subject areas

  • Physiology
  • Physiology (medical)

Fingerprint

Dive into the research topics of 'SeriesSleepNet: an EEG time series model with partial data augmentation for automatic sleep stage scoring'. Together they form a unique fingerprint.

Cite this