TinySleepNet: An Efficient Deep Learning Model for Sleep Stage Scoring based on Raw Single-Channel EEG

Akara Supratak*, Yi-Ke GUO

*Corresponding author for this work

    Research output: Chapter in book/report/conference proceedingConference contributionpeer-review

    54 Citations (Scopus)

    Abstract

    Deep learning has become popular for automatic sleep stage scoring due to its capability to extract useful features from raw signals. Most of the existing models, however, have been overengineered to consist of many layers or have introduced additional steps in the processing pipeline, such as converting signals to spectrogram-based images. They require to be trained on a large dataset to prevent the overfitting problem (but most of the sleep datasets contain a limited amount of class-imbalanced data) and are difficult to be applied (as there are many hyperparameters to be configured in the pipeline). In this paper, we propose an efficient deep learning model, named TinySleepNet, and a novel technique to effectively train the model end-to-end for automatic sleep stage scoring based on raw single-channel EEG. Our model consists of a less number of model parameters to be trained compared to the existing ones, requiring a less amount of training data and computational resources. Our training technique incorporates data augmentation that can make our model be more robust the shift along the time axis, and can prevent the model from remembering the sequence of sleep stages. We evaluated our model on seven public sleep datasets that have different characteristics in terms of scoring criteria and recording channels and environments. The results show that, with the same model architecture and the training parameters, our method achieves a similar (or better) performance compared to the state-of-the-art methods on all datasets. This demonstrates that our method can generalize well to the largest number of different datasets.

    Original languageEnglish
    Title of host publication42nd Annual International Conferences of the IEEE Engineering in Medicine and Biology Society
    Subtitle of host publicationEnabling Innovative Technologies for Global Healthcare, EMBC 2020
    PublisherIEEE
    Pages641-644
    Number of pages4
    ISBN (Electronic)9781728119908
    DOIs
    Publication statusPublished - Jul 2020
    Event42nd Annual International Conferences of the IEEE Engineering in Medicine and Biology Society, EMBC 2020 - Montreal, Canada
    Duration: 20 Jul 202024 Jul 2020

    Publication series

    NameProceedings of the Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS
    Volume2020-July
    ISSN (Print)1557-170X

    Conference

    Conference42nd Annual International Conferences of the IEEE Engineering in Medicine and Biology Society, EMBC 2020
    Country/TerritoryCanada
    CityMontreal
    Period20/07/2024/07/20

    Scopus Subject Areas

    • Signal Processing
    • Biomedical Engineering
    • Computer Vision and Pattern Recognition
    • Health Informatics

    Fingerprint

    Dive into the research topics of 'TinySleepNet: An Efficient Deep Learning Model for Sleep Stage Scoring based on Raw Single-Channel EEG'. Together they form a unique fingerprint.

    Cite this