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

Abstract

We used convolutional neural networks (CNNs) for automatic sleep stage scoring based on single-channel electroencephalography (EEG) to learn task-specific filters for classification without using prior domain knowledge. We used an openly available dataset from 20 healthy young adults for evaluation and applied 20-fold cross-validation. We used class-balanced random sampling within the stochastic gradient descent (SGD) optimization of the CNN to avoid skewed performance in favor of the most represented sleep stages. We achieved high mean F1-score (81%, range 79-83%), mean accuracy across individual sleep stages (82%, range 80-84%) and overall accuracy (74%, range 71-76%) over all subjects. By analyzing and visualizing the filters that our CNN learns, we found that rules learned by the filters correspond to sleep scoring criteria in the American Academy of Sleep Medicine (AASM) manual that human experts follow. Our method's performance is balanced across classes and our results are comparable to state-of-the-art methods with hand-engineered features. We show that, without using prior domain knowledge, a CNN can automatically learn to distinguish among different normal sleep stages.

12 pages

Cited by in corpus (21)

• DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG
• SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging
• A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series
• SleepEEGNet: Automated Sleep Stage Scoring with Sequence to Sequence Deep Learning Approach
• Joint Classification and Prediction CNN Framework for Automatic Sleep Stage Classification
• SleepTransformer: Automatic Sleep Staging with Interpretability and Uncertainty Quantification
• XSleepNet: Multi-View Sequential Model for Automatic Sleep Staging
• Towards More Accurate Automatic Sleep Staging via Deep Transfer Learning
• Automatic Sleep Staging of EEG Signals: Recent Development, Challenges, and Future Directions
• SleePyCo: Automatic Sleep Scoring with Feature Pyramid and Contrastive Learning
• Edge Deep Learning for Neural Implants
• Do Not Sleep on Traditional Machine Learning: Simple and Interpretable Techniques Are Competitive to Deep Learning for Sleep Scoring
• Automatic detection of microsleep episodes with deep learning
• A Survey on Deep Learning-based Non-Invasive Brain Signals:Recent Advances and New Frontiers
• Addressing Class Imbalance in Classification Problems of Noisy Signals by using Fourier Transform Surrogates
• Diffuse to fuse EEG spectra -- intrinsic geometry of sleep dynamics for classification
• Sleep Stage Classification Based on Multi-level Feature Learning and Recurrent Neural Networks via Wearable Device
• A Structured Learning Approach with Neural Conditional Random Fields for Sleep Staging
• Compensated Integrated Gradients to Reliably Interpret EEG Classification
• DeepSleepNet-Lite: A Simplified Automatic Sleep Stage Scoring Model with Uncertainty Estimates
• Sleep Staging Based on Multi Scale Dual Attention Network