sleep-stages-classification-papers

This repo contains a list of papers for sleep stage classification using deep learning based techniques starting from 2017-afterwards.

I classified them generally according to the proposed DL technique.

If you have any suggested papers, please contact me
emad0002{at}e.ntu.edu.sg

Review papers

year	Paper	PDF
2022	Representations of temporal sleep dynamics: Review and synthesis of the literature Sleep Medicine Reviews	PDF
2022	A comprehensive evaluation of contemporary methods used for automatic sleep staging Biomedical Signal Processing and Control	PDF.
2021	Automatic Sleep Staging: Recent Development, Challenges, and Future Directions Arxiv	PDF
2020	Automated Detection of Sleep Stages Using Deep Learning Techniques: A Systematic Review of the Last Decade (2010–2020) Applied Sciences	PDF
2019	Automated sleep scoring: A review of the latest approaches Sleep Medicine Reviews	PDF
2019	A review of automated sleep stage scoring based on physiological signals for the new millennia Computer Methods and Programs in Biomedicine	PDF

CNN only

year	Paper	PDF	Code
2022	Enhancing Contextual Encoding With Stage-Confusion and Stage-Transition Estimation for EEG-Based Sleep Staging (v2) IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)	PDF	github
2021	SalientSleepNet: Multimodal Salient Wave Detection Network for Sleep Staging IJCAI-21	PDF	github
2020	SleepPrintNet: A Multivariate Multimodal Neural Network based on Physiological Time-series for Automatic Sleep Staging IEEE Transactions on Artificial Intelligence	PDF	github
2020	Orthogonal convolutional neural networks for automatic sleep stage classification based on single-channel EEG Computer Methods and Programs in Biomedicine	PDF	-
2020	TRIER: Template-Guided Neural Networks for Robust and Interpretable Sleep Stage Identification from EEG Recordings Arxiv	PDF	-
2020	Computation-Efficient Multi-Model Deep Neural Network for Sleep Stage Classification Asia Service Sciences and Software Engineering Conference	PDF	-
2019	U-Time: A Fully Convolutional Network for Time Series Segmentation Applied to Sleep Staging Advances in Neural Information Processing Systems (NeurIPS)	PDF	github
2019	Deep learning for automated feature discovery and classification of sleep stages IEEE Transactions On Computational Biology and Bioinformatics	PDF	-
2019	A Deep Learning Model for Automated Sleep Stages Classification Using PSG Signals International Journal of Environmental Research and Public Health	PDF	-
2019	Joint Classification and Prediction CNN Framework for Automatic Sleep Stage Classification IEEE Transactions on Biomedical Engineering	PDF	github
2019	An Image Based Prediction Model for Sleep Stage Identification IEEE International Conference on Image Processing (ICIP)	PDF	-
2018	DNN Filter Bank Improves 1-Max Pooling CNN for Single-Channel EEG Automatic Sleep Stage Classification IEEE Engineering in Medicine and Biology Society (EMBC)	PDF	-
2018	Deep residual networks for automatic sleep stage classification of raw polysomnographic waveforms IEEE Engineering in Medicine and Biology Society (EMBC)	PDF	-

CNN with LSTM

year	Paper	PDF	Code
2021	CCRRSleepNet: A Hybrid Relational Inductive Biases Network for Automatic Sleep Stage Classification on Raw Single-Channel EEG Brain Sciences	PDF	github
2020	An Automatic Sleep Staging Method Using a Multi-head and Sequence Network International Conference on Biological Information and Biomedical Engineering	PDF	-
2020	Intra- and inter-epoch temporal context network (IITNet) using sub-epoch features for automatic sleep scoring on raw single-channel EEG Biomedical Signal Processing and Control	PDF	-
2020	TinySleepNet: An Efficient Deep Learning Model for Sleep Stage Scoring based on Raw Single-Channel EEG International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC)	PDF	github
2020	Temporal dependency in automatic sleep scoring via deep learning based architectures: An empirical study International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC)	PDF	-
2019	End-to-end Sleep Staging with Raw Single Channel EEG using Deep Residual ConvNets IEEE EMBS International Conference on Biomedical & Health Informatics (BHI)	PDF	-
2018	A Structured Learning Approach with Neural Conditional Random Fields for Sleep Staging IEEE International Conference on Big Data	PDF	-
2018	Expert-level sleep scoring with deep neural networks Journal of the American Medical Informatics Association	PDF	-
2018	Deep Convolutional Network Method for Automatic Sleep Stage Classification Based on Neurophysiological Signals International Congress on Image and Signal Processing, BioMedical Engineering and Informatics (CISP-BMEI)	PDF	-
2018	Learning Sleep Stages from Radio Signals: A Conditional Adversarial Architecture International Conference on Machine Learning (ICML)	PDF	-
2017	A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series IEEE Transactions on Neural Systems and Rehabilitation Engineering	PDF	-
2017	DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG IEEE Transactions on Neural Systems and Rehabilitation Engineering	PDF	python2 python3

CNN with LSTM with attention

year	Paper	PDF	Code
2020	XSleepNet: Multi-View Sequential Model for Automatic Sleep Staging IEEE Transactions on Pattern Analysis and Machine Intelligence	PDF	-
2020	An Automatic Sleep Staging Model Combining Feature Learning and Sequence Learning International Conference on Advanced Computational Intelligence (ICACI)	PDF	-
2019	SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging IEEE Transactions on Neural Systems and Rehabilitation Engineering	PDF	github
2019	SleepEEGNet: Automated sleep stage scoring with sequence to sequence deep learning approach Plos One	PDF	github
2018	Automatic Sleep Stage Classification Using Single-Channel EEG: Learning Sequential Features with Attention-Based Recurrent Neural Networks IEEE Engineering in Medicine and Biology Society (EMBC)	PDF	-

Attention-based

year	Paper	PDF	Code
2021	An Attention-based Deep Learning Approach for Sleep Stage Classification with Single-Channel EEG IEEE Transactions on Neural Systems and Rehabilitation Engineering	PDF	github
2020	Convolution- and Attention-Based Neural Network for Automated Sleep Stage Classification International Journal of Environmental Research and Public Health	PDF	-
2020	A Residual Based Attention Model for EEG Based Sleep Staging IEEE Journal of Biomedical and Health Informatics	PDF	-
2018	Multivariate Sleep Stage Classification using Hybrid Self-Attentive Deep Learning Networks IEEE International Conference on Bioinformatics and Biomedicine (BIBM)	PDF	-

Transfer learning

year	Paper	PDF	Code
2021	RobustSleepNet: Transfer learning for automated sleep staging at scale IEEE Engineering in Medicine and Biology Society (EMBC)	PDF	github
2020	MetaSleepLearner: A Pilot Study on Fast Adaptation of Bio-signals-Based Sleep Stage Classifier to New Individual Subject Using Meta-Learning IEEE Journal of Biomedical and Health Informatics	PDF	github
2019	Towards More Accurate Automatic Sleep Staging via Deep Transfer Learning IEEE Transactions on Biomedical Engineering	PDF	github
2018	Multichannel Sleep Stage Classification and Transfer Learning using Convolutional Neural Networks IEEE Engineering in Medicine and Biology Society (EMBC)	PDF	-
2017	deep convolutional neural networks for interpretable analysis of eeg sleep stage scoring International Workshop on Machine Learning for Signal Processing (MLSP)	PDF	-

Personalized sleep staging

year	Paper	PDF	Code
2020	Personalized automatic sleep staging with single-night data: a pilot study with Kullback–Leibler divergence regularization IOP science	PDF
2018	Personalizing deep learning models for automatic sleep staging Arxiv	PDF

Domain Adaptation

year	Paper	PDF	Code
2022	ADAST: Attentive Cross-domain EEG-based Sleep Staging Framework with Iterative Self-Training IEEE Transactions on Emerging Topics in Computational Intelligence	PDF	github
2022	From unsupervised to semi-supervised adversarial domain adaptation in electroencephalography-based sleep staging Journal of Neural Engineering	PDF	-
2021	Transferring structured knowledge in unsupervised domain adaptation of a sleep staging network IEEE Journal of Biomedical and Health Informatics	PDF	-
2021	Unsupervised sleep staging system based on domain adaptation Biomedical Signal Processing and Control	PDF	-
2020	Attentive Adversarial Network for Large-Scale Sleep Staging Machine Learning for Healthcare (MLHC)	PDF	-

Self-Supervised Learning

year	Paper	PDF	Code
2021	Time-Series Representation Learning via Temporal and Contextual Contrasting IJCAI-21	PDF	github
2021	Self-supervised Electroencephalogram Representation Learning for Automatic Sleep Staging Arxiv	PDF	github
2021	SleepPriorCL: Contrastive Representation Learning with Prior Knowledge-based Positive Mining and Adaptive Temperature for Sleep Staging Arxiv	PDF	-
2021	Self-Supervised Learning for Sleep Stage Classification with Predictive and Discriminative Contrastive Coding IEEE International Conference on Acoustics, Speech and Signal Processing	PDF	-
2021	Self-supervised Contrastive Learning for EEG-based Sleep Staging International Joint Conference on Neural Networks (IJCNN)	PDF	github

Manual features + Deep learning

year	Paper	PDF	Code
2020	Automatic Sleep Stage Classification using Marginal Hilbert Spectrum Features and a Convolutional Neural Network International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC)	PDF	-
2018	Mixed Neural Network Approach for Temporal Sleep Stage Classification IEEE Transactions on Neural Systems and Rehabilitation Engineering	PDF	-
2017	SLEEPNET: Automated Sleep Staging System via Deep Learning Arxiv	PDF	-

Graph

year	Paper	PDF	Code
2022	An Attention-Guided Spatiotemporal Graph Convolutional Network for Sleep Stage Classification Life	PDF	-
2021	Multi-View Spatial-Temporal Graph Convolutional Networks with Domain Generalization for Sleep Stage Classification IEEE Transactions on Neural Systems and Rehabilitation Engineering	PDF	github
2020	A Graph-Temporal fused dual-input Convolutional Neural Network for Detecting Sleep Stages from EEG Signals IEEE Transactions on Circuits and Systems II: Express Briefs	PDF	-
2020	GraphSleepNet: Adaptive Spatial-Temporal Graph Convolutional Networks for Sleep Stage Classification IJCAI-20	PDF	github

Data Augmentations

year	Paper	PDF	Code
2020	EEG data augmentation: towards class imbalance problem in sleep staging tasks IOP science	PDF	-