Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments

Sleep stage classification plays an important role in the diagnosis of sleep-related diseases. However, traditional automatic sleep stage classification is quite challenging because of the complexity associated with the establishment of mathematical models and the extraction of handcrafted features....

Full description

Saved in:
Bibliographic Details
Main Authors: Zhihong Cui, Xiangwei Zheng, Xuexiao Shao, Lizhen Cui
Format: Article
Language:English
Published: Wiley 2018-01-01
Series:Complexity
Online Access:http://dx.doi.org/10.1155/2018/9248410
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1832550271659540480
author Zhihong Cui
Xiangwei Zheng
Xuexiao Shao
Lizhen Cui
author_facet Zhihong Cui
Xiangwei Zheng
Xuexiao Shao
Lizhen Cui
author_sort Zhihong Cui
collection DOAJ
description Sleep stage classification plays an important role in the diagnosis of sleep-related diseases. However, traditional automatic sleep stage classification is quite challenging because of the complexity associated with the establishment of mathematical models and the extraction of handcrafted features. In addition, the rapid fluctuations between sleep stages often result in blurry feature extraction, which might lead to an inaccurate assessment of electroencephalography (EEG) sleep stages. Hence, we propose an automatic sleep stage classification method based on a convolutional neural network (CNN) combined with the fine-grained segment in multiscale entropy. First, we define every 30 seconds of the multichannel EEG signal as a segment. Then, we construct an input time series based on the fine-grained segments, which means that the posterior and current segments are reorganized as an input containing several segments and the size of the time series is decided based on the scale chosen depending on the fine-grained segments. Next, each segment in this series is individually put into the designed CNN and feature maps are obtained after two blocks of convolution and max-pooling as well as a full-connected operation. Finally, the results from the full-connected layer of each segment in the input time sequence are put into the softmax classifier together to get a single most likely sleep stage. On a public dataset called ISRUC-Sleep, the average accuracy of our proposed method is 92.2%. Moreover, it yields an accuracy of 90%, 86%, 93%, 97%, and 90% for stage W, stage N1, stage N2, stage N3, and stage REM, respectively. Comparative analysis of performance suggests that the proposed method is better, as opposed to that of several state-of-the-art ones. The sleep stage classification methods based on CNN and the fine-grained segments really improve a key step in the study of sleep disorders and expedite sleep research.
format Article
id doaj-art-4c967b29537248a087dd8e01c1f49f25
institution Kabale University
issn 1076-2787
1099-0526
language English
publishDate 2018-01-01
publisher Wiley
record_format Article
series Complexity
spelling doaj-art-4c967b29537248a087dd8e01c1f49f252025-02-03T06:07:11ZengWileyComplexity1076-27871099-05262018-01-01201810.1155/2018/92484109248410Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained SegmentsZhihong Cui0Xiangwei Zheng1Xuexiao Shao2Lizhen Cui3School of Information Science and Engineering, Shandong Normal University, Ji’nan 250014, ChinaSchool of Information Science and Engineering, Shandong Normal University, Ji’nan 250014, ChinaSchool of Information Science and Engineering, Shandong Normal University, Ji’nan 250014, ChinaSchool of Software, Shandong University, Ji’nan 250101, ChinaSleep stage classification plays an important role in the diagnosis of sleep-related diseases. However, traditional automatic sleep stage classification is quite challenging because of the complexity associated with the establishment of mathematical models and the extraction of handcrafted features. In addition, the rapid fluctuations between sleep stages often result in blurry feature extraction, which might lead to an inaccurate assessment of electroencephalography (EEG) sleep stages. Hence, we propose an automatic sleep stage classification method based on a convolutional neural network (CNN) combined with the fine-grained segment in multiscale entropy. First, we define every 30 seconds of the multichannel EEG signal as a segment. Then, we construct an input time series based on the fine-grained segments, which means that the posterior and current segments are reorganized as an input containing several segments and the size of the time series is decided based on the scale chosen depending on the fine-grained segments. Next, each segment in this series is individually put into the designed CNN and feature maps are obtained after two blocks of convolution and max-pooling as well as a full-connected operation. Finally, the results from the full-connected layer of each segment in the input time sequence are put into the softmax classifier together to get a single most likely sleep stage. On a public dataset called ISRUC-Sleep, the average accuracy of our proposed method is 92.2%. Moreover, it yields an accuracy of 90%, 86%, 93%, 97%, and 90% for stage W, stage N1, stage N2, stage N3, and stage REM, respectively. Comparative analysis of performance suggests that the proposed method is better, as opposed to that of several state-of-the-art ones. The sleep stage classification methods based on CNN and the fine-grained segments really improve a key step in the study of sleep disorders and expedite sleep research.http://dx.doi.org/10.1155/2018/9248410
spellingShingle Zhihong Cui
Xiangwei Zheng
Xuexiao Shao
Lizhen Cui
Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments
Complexity
title Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments
title_full Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments
title_fullStr Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments
title_full_unstemmed Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments
title_short Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments
title_sort automatic sleep stage classification based on convolutional neural network and fine grained segments
url http://dx.doi.org/10.1155/2018/9248410
work_keys_str_mv AT zhihongcui automaticsleepstageclassificationbasedonconvolutionalneuralnetworkandfinegrainedsegments
AT xiangweizheng automaticsleepstageclassificationbasedonconvolutionalneuralnetworkandfinegrainedsegments
AT xuexiaoshao automaticsleepstageclassificationbasedonconvolutionalneuralnetworkandfinegrainedsegments
AT lizhencui automaticsleepstageclassificationbasedonconvolutionalneuralnetworkandfinegrainedsegments