IDEAS home Printed from https://ideas.repec.org/a/gam/jijerp/v19y2022i10p6322-d821711.html
   My bibliography  Save this article

A Deep Learning Method Approach for Sleep Stage Classification with EEG Spectrogram

Author

Listed:
  • Chengfan Li

    (School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China)

  • Yueyu Qi

    (School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China)

  • Xuehai Ding

    (School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China)

  • Junjuan Zhao

    (School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China)

  • Tian Sang

    (School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China)

  • Matthew Lee

    (12th Grade, The Bishop’s School, La Jolla, CA 92037, USA)

Abstract

The classification of sleep stages is an important process. However, this process is time-consuming, subjective, and error-prone. Many automated classification methods use electroencephalogram (EEG) signals for classification. These methods do not classify well enough and perform poorly in the N1 due to unbalanced data. In this paper, we propose a sleep stage classification method using EEG spectrogram. We have designed a deep learning model called EEGSNet based on multi-layer convolutional neural networks (CNNs) to extract time and frequency features from the EEG spectrogram, and two-layer bi-directional long short-term memory networks (Bi-LSTMs) to learn the transition rules between features from adjacent epochs and to perform the classification of sleep stages. In addition, to improve the generalization ability of the model, we have used Gaussian error linear units (GELUs) as the activation function of CNN. The proposed method was evaluated by four public databases, the Sleep-EDFX-8, Sleep-EDFX-20, Sleep-EDFX-78, and SHHS. The accuracy of the method is 94.17%, 86.82%, 83.02% and 85.12%, respectively, for the four datasets, the MF1 is 87.78%, 81.57%, 77.26% and 78.54%, respectively, and the Kappa is 0.91, 0.82, 0.77 and 0.79, respectively. In addition, our proposed method achieved better classification results on N1, with an F1-score of 70.16%, 52.41%, 50.03% and 47.26% for the four datasets.

Suggested Citation

  • Chengfan Li & Yueyu Qi & Xuehai Ding & Junjuan Zhao & Tian Sang & Matthew Lee, 2022. "A Deep Learning Method Approach for Sleep Stage Classification with EEG Spectrogram," IJERPH, MDPI, vol. 19(10), pages 1-17, May.
    • Handle: RePEc:gam:jijerp:v:19:y:2022:i:10:p:6322-:d:821711
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/1660-4601/19/10/6322/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/1660-4601/19/10/6322/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Tianqi Zhu & Wei Luo & Feng Yu, 2020. "Convolution- and Attention-Based Neural Network for Automated Sleep Stage Classification," IJERPH, MDPI, vol. 17(11), pages 1-13, June.
    2. Sajad Mousavi & Fatemeh Afghah & U Rajendra Acharya, 2019. "SleepEEGNet: Automated sleep stage scoring with sequence to sequence deep learning approach," PLOS ONE, Public Library of Science, vol. 14(5), pages 1-15, May.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Tingting Li & Bofeng Zhang & Hehe Lv & Shengxiang Hu & Zhikang Xu & Yierxiati Tuergong, 2022. "CAttSleepNet: Automatic End-to-End Sleep Staging Using Attention-Based Deep Neural Networks on Single-Channel EEG," IJERPH, MDPI, vol. 19(9), pages 1-15, April.
    2. Sajad Mousavi & Atiyeh Fotoohinasab & Fatemeh Afghah, 2020. "Single-modal and multi-modal false arrhythmia alarm reduction using attention-based convolutional and recurrent neural networks," PLOS ONE, Public Library of Science, vol. 15(1), pages 1-15, January.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jijerp:v:19:y:2022:i:10:p:6322-:d:821711. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.