An Attention-Based Deep Learning Approach for Sleep Stage Classification With Single-Channel EEG (2021-04-28T00:00:00.000000Z)

TL;DR

A novel attention-based deep learning architecture called AttnSleep is proposed to classify sleep stages using single channel EEG signals to outperforms state-of-the-art techniques in terms of different evaluation metrics.

Abstract

Automatic sleep stage mymargin classification is of great importance to measure sleep quality. In this paper, we propose a novel attention-based deep learning architecture called AttnSleep to classify sleep stages using single channel EEG signals. This architecture starts with the feature extraction module based on multi-resolution convolutional neural network (MRCNN) and adaptive feature recalibration (AFR). The MRCNN can extract low and high frequency features and the AFR is able to improve the quality of the extracted features by modeling the inter-dependencies between the features. The second module is the temporal context encoder (TCE) that leverages a multi-head attention mechanism to capture the temporal dependencies among the extracted features. Particularly, the multi-head attention deploys causal convolutions to model the temporal relations in the input features. We evaluate the performance of our proposed AttnSleep model using three public datasets. The results show that our AttnSleep outperforms state-of-the-art techniques in terms of different evaluation metrics. Our source codes, experimental data, and supplementary materials are available at https://github.com/emadeldeen24/AttnSleep.

Authors

Zhenghua Chen

5 papers

Cuntai Guan

4 papers

Chengyu Liu

3 papers

TL;DR

Abstract

Authors

References46 items

A Novel Ensemble Deep Learning Approach for Sleep-Wake Detection Using Heart Rate Variability and Acceleration

End-to-end sleep staging using convolutional neural network in raw single-channel EEG

Deep Learning for Automated Feature Discovery and Classification of Sleep Stages

An Attention Based CNN-LSTM Approach for Sleep-Wake Detection With Heterogeneous Sensors

Convolution- and Attention-Based Neural Network for Automated Sleep Stage Classification

An improved deep convolutional neural network with multi-scale information for bearing fault diagnosis

SleepEEGNet: Automated sleep stage scoring with sequence to sequence deep learning approach

Cascaded LSTM recurrent neural network for automated sleep stage classification using single-channel EEG signals

Automatic Human Sleep Stage Scoring Using Deep Neural Networks

Deep Convolutional Network Method for Automatic Sleep Stage Classification Based on Neurophysiological Signals

SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging

A new automatic sleep staging system based on statistical behavior of local extrema using single channel EEG signal

The National Sleep Research Resource: towards a sleep data commons

Joint Classification and Prediction CNN Framework for Automatic Sleep Stage Classification

A convolutional neural network for sleep stage scoring from raw single-channel EEG

HyCLASSS: A Hybrid Classifier for Automatic Sleep Stage Scoring

Squeeze-and-Excitation Networks

A novel, fast and efficient single-sensor automatic sleep-stage classification based on complementary cross-frequency coupling estimates

A Deep Learning Architecture for Temporal Sleep Stage Classification Using Multivariate and Multimodal Time Series

Cardiorespiratory Sleep Stage Detection Using Conditional Random Fields

Attention is All you Need

DeepSleepNet: A Model for Automatic Sleep Stage Scoring Based on Raw Single-Channel EEG

Automatic Sleep Stage Scoring with Single-Channel EEG Using Convolutional Neural Networks

Very deep convolutional neural networks for raw waveforms

Layer Normalization

Computer-aided sleep staging using Complete Ensemble Empirical Mode Decomposition with Adaptive Noise and bootstrap aggregating

Empirical Evaluation of Rectified Activations in Convolutional Network

Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift

Adam: A Method for Stochastic Optimization

Analysis and Classification of Sleep Stages Based on Difference Visibility Graphs From a Single-Channel EEG Signal

Sleep: a health imperative.

Sleep and Metabolism: An Overview

SVMs Modeling for Highly Imbalanced Classification

The relationships between memory systems and sleep stages

Relationship between blood pressure, sleep K-complexes, and muscle sympathetic nerve activity in humans.

The Sleep Heart Health Study: design, rationale, and methods.

A Coefficient of Agreement for Nominal Scales

BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

A Novel Multi-Class EEG-Based Sleep Stage Classification System

Partial Sleep Deprivation Attenuates the Positive Affective System: Effects Across Multiple Measurement Modalities

“An overview of polysomnography,”

and F

SMOTE: Synthetic Minority Over-sampling Technique

Physionet: components of a new research resource for complex physiologic signals

We design a class-aware loss function to efﬁciently handle the class imbalance without introducing additional computations

We propose a novel temporal context encoder that deploys a multi-head self-attention with causal convo-lutions to efﬁciently capture the temporal dependencies in the extracted features

Field of Study

Journal Information

Name

Page

Volume

Venue Information

Name

Type

URL

Alternate Names