Abstract:Background: Despite recent significant progress in the development of automatic sleep staging methods, building a good model still remains a big challenge for sleep studies with a small cohort due to the data-variability and data-inefficiency issues. This work presents a deep transfer learning approach to overcome these issues and enable transferring knowledge from a large dataset to a small cohort for automatic sleep staging. Methods: We start from a generic end-to-end deep learning framework for sequence-to-sequence sleep staging and derive two networks as the means for transfer learning. The networks are first trained in the source domain (i.e. the large database). The pretrained networks are then finetuned in the target domain (i.e. the small cohort) to complete knowledge transfer. We employ the Montreal Archive of Sleep Studies (MASS) database consisting of 200 subjects as the source domain and study deep transfer learning on three different target domains: the Sleep Cassette subset and the Sleep Telemetry subset of the Sleep-EDF Expanded database, and the Surrey-cEEGrid database. The target domains are purposely adopted to cover different degrees of data mismatch to the source domains. Results: Our experimental results show significant performance improvement on automatic sleep staging on the target domains achieved with the proposed deep transfer learning approach. Conclusions: These results suggest the efficacy of the proposed approach in addressing the above-mentioned data-variability and data-inefficiency issues. Significance: As a consequence, it would enable one to improve the quality of automatic sleep staging models when the amount of data is relatively small. The source code and the pretrained models are available at <a class="link-external link-http" href="http://github.com/pquochuy/sleep_transfer_learning" rel="external noopener nofollow">this http URL</a>.

Eeg-Based Sleep Quality Evaluation With Deep Transfer Learning

Automatic Classification of Sleep Stages Based on Raw Single-Channel EEG

A deep transfer learning approach for wearable sleep stage classification with photoplethysmography

Measuring Sleep Quality From Eeg With Machine Learning Approaches

Deep adaptation network for subject-specific sleep stage classification based on a single-lead ECG

Deep Learning for Sleep Stage Classification

Single-channel EEG sleep staging based on data augmentation and cross-subject discrepancy alleviation

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

Deep Learning in Automatic Sleep Staging With a Single Channel Electroencephalography

Automatic Sleep Stage Classification Based on Sparse Deep Belief Net and Combination of Multiple Classifiers

Automatic sleep-stage classification of heart rate and actigraphy data using deep and transfer learning approaches

EEGNet classification of sleep EEG for individual specialization based on data augmentation

Deep Learning Method For Sleep Stage Classification

Deep Convolutional Neural Networks for Interpretable Analysis of EEG Sleep Stage Scoring

Automatic Sleep Staging Method of EEG Signal Based on Transfer Learning and Fusion Network

Joint semi-supervised feature auto-weighting and classification model for eeg-based cross-subject sleep quality evaluation

Deep Convolutional Network Method For Automatic Sleep Stage Classification Based On Neurophysiological Signals

An End-to-End Deep Leaning Approach for Sleep-Wake Classification Using Single Channel EEG Signals

Towards More Accurate Automatic Sleep Staging via Deep Transfer Learning

Automatic Sleep Staging Based on Single-Channel EEG Signal Using Null Space Pursuit Decomposition Algorithm

A Deep Learning Approach with Conditional Random Field for Automatic Sleep Stage Scoring