FlexibleSleepNet:A Model for Automatic Sleep Stage Classification Based on Multi-Channel Polysomnography
Ze Ren, Jin Ma, Ying Ding
Abstract
In the task of automatic sleep stage classification, deep learning models often face the challenge of balancing temporal-spatial feature extraction with computational complexity. To address this issue, this study introduces FlexibleSleepNet, a lightweight convolutional neural network architecture designed around the Adaptive Feature Extraction (AFE) Module and Scale-Varying Compression (SVC) Module. Through multi-channel polysomnography data input and preprocessing, FlexibleSleepNet utilizes the AFE Module to capture intra-channel features and employs the SVC Module for channel feature compression and dimension expansion. The collaborative work of these modules enables the network to effectively capture temporal-spatial dependencies between channels. Additionally, the network extracts feature maps through four distinct stages, each from different receptive field scales, culminating in precise sleep stage classification via a classification module. This study conducted k-fold cross-validation on three different databases: SleepEDF-20, SleepEDF-78, and SHHS. Experimental results show that FlexibleSleepNet demonstrates superior classification performance, achieving classification accuracies of 86.9% and 87.6% on the SleepEDF-20 and SHHS datasets, respectively. It performs particularly well on the SleepEDF-78 dataset, where it reaches a classification accuracy of 87.0%. Additionally, it has significantly enhanced computational efficiency while maintaining low computational complexity.