A hybrid deep learning scheme for multi-channel sleep stage classification

[thumbnail of TSP_CMC_21830.pdf]
TSP_CMC_21830.pdf - Published Version (669kB) | Preview
Available under license: Creative Commons Attribution

Pei, Wei, Li, Yan, Siuly, Siuly ORCID: 0000-0003-2491-0546 and Wen, Peng (2021) A hybrid deep learning scheme for multi-channel sleep stage classification. Computers, Materials and Continua, 71 (1). pp. 889-905. ISSN 1546-2218


Sleep stage classification plays a significant role in the accurate diagnosis and treatment of sleep-related diseases. This study aims to develop an efficient deep learning based scheme for correctly identifying sleep stages using multi-biological signals such as electroencephalography (EEG), electrocardiogram (ECG), electromyogram (EMG), and electrooculogram (EOG). Most of the prior studies in sleep stage classification focus on hand-crafted feature extraction methods. Traditional hand-crafted feature extraction methods choose features manually from raw data, which is tedious, and these features are limited in their ability to balance efficiency and accuracy. Moreover, most of the existing works on sleep staging are either single channel (a single-lead EEG may not contain enough information) or only EEG signal based which can not reveal more complicated physical features for reliable classification of various sleep stages. This study proposes an approach to combine Convolutional Neural Networks (CNNs) and Gated Recurrent Units (GRUs) that can discover hidden features frommulti-biological signal data to recognize the different sleep stages efficiently. In the proposed scheme, the CNN is designed to extract concealed features from the multi-biological signals, and the GRU is employed to automatically learn the transition rules among different sleep stages. After that, the softmax layers are used to classify various sleep stages. The proposed method was tested on two publicly available databases: Sleep Heart Health Study (SHHS) and St. Vincent's University Hospital/University College Dublin Sleep Apnoea (UCDDB). The experimental results reveal that the proposed model yields better performance compared to state-of-the-art works. Our proposed scheme will assist in building a new system to deal with multi-channel or multi-modal signal processing tasks in various applications.

Dimensions Badge

Altmetric Badge

Item type Article
URI https://vuir.vu.edu.au/id/eprint/44979
DOI 10.32604/cmc.2022.021830
Official URL https://www.techscience.com/cmc/v71n1/45428
Subjects Current > FOR (2020) Classification > 4006 Communications engineering
Current > FOR (2020) Classification > 4602 Artificial intelligence
Current > FOR (2020) Classification > 4605 Data management and data science
Current > Division/Research > Institute for Sustainable Industries and Liveable Cities
Keywords deep learning, sleep stage classification, multi-biological signals
Citations in Scopus 5 - View on Scopus
Download/View statistics View download statistics for this item

Search Google Scholar

Repository staff login