Improving Sleep Stage Classification Performance by Single-Channel EEG Data Augmentation via Spectral Band Blending

Choel Hui Lee, Hyun Ji Kim, Jae Wook Heo, Hakseung Kim, Dong Joo Kim

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Classification of sleep stage for neurophysiological signals acquired during polysomnography is an important task in the diagnosis of sleep disorders and sleep research. However, current manual sleep stage labeling by trained experts involves heavy labour intensity. Furthermore, previous models regarding the automation of sleep stage classification using machine learning are highly susceptible to class imbalance problem, which is prevalent in polysomnography recordings. This study proposes an automated sleep stage classification method based on data augmentation via spectral band blending to address class imbalance problem, and thereby improving the performance of machine learning models for sleep stage classification. In this endeavour, EEG recordings from 153 healthy subjects were utilized to develop the EEGNet-BiLSTM as the baseline model. When compared the performance of sleep stage classification between EEGNet-BiLSTM with and without data augmentation, EEGNet-BiLSTM without data augmentation yielded approximately 82% accuracy and 0.70 kappa-value, whereas the proposed method showed 87% accuracy and 0.73 kappa-value. Our proposed method, EEGNet-BiLSTM with data augmentation, is superior in terms of accuracy and consistency compared to the conventional baseline model.

Original languageEnglish
Title of host publication9th IEEE International Winter Conference on Brain-Computer Interface, BCI 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728184852
DOIs
Publication statusPublished - 2021 Feb 22
Event9th IEEE International Winter Conference on Brain-Computer Interface, BCI 2021 - Gangwon, Korea, Republic of
Duration: 2021 Feb 222021 Feb 24

Publication series

Name9th IEEE International Winter Conference on Brain-Computer Interface, BCI 2021

Conference

Conference9th IEEE International Winter Conference on Brain-Computer Interface, BCI 2021
Country/TerritoryKorea, Republic of
CityGangwon
Period21/2/2221/2/24

Keywords

  • class imbalance problem
  • data augmentation
  • deep learning
  • electroencephalography
  • sleep stage
  • sleep stage classification

ASJC Scopus subject areas

  • Artificial Intelligence
  • Human-Computer Interaction
  • Signal Processing

Fingerprint

Dive into the research topics of 'Improving Sleep Stage Classification Performance by Single-Channel EEG Data Augmentation via Spectral Band Blending'. Together they form a unique fingerprint.

Cite this