Convolutional Neural Networks for Deep Sleep Detection Based on Data Augmentation

Abstract

Sleep is a necessary process that individuals undergo daily for physical recovery, and the proportion of deep sleep in the sleep stages is a critical aspect of the recovery process. Convolutional Neural Networks (CNNs) have shown remarkable success in automatically identifying deep sleep stages through the analysis of electroencephalogram (EEG) signals. This article introduces three data augmentation techniques, including time shifting, amplitude scaling and noise addition, to enhance the diversity and features of the data. These techniques aim to enable machine learning models to extract features from various aspects of sleep EEG data, thus improving the model’s accuracy. Three deep learning models are introduced, namely DeepConvNet, ShallowConvNet and EEGNet, for the identification of deep sleep. To evaluate the proposed methods, the Sleep-EDF public dataset was utilized. Experimental results demonstrate that the enhanced dataset formed by applying the three data augmentation techniques achieved higher accuracy in all deep learning models compared to the original dataset. This highlights the feasibility and effectiveness of these methods in deep sleep detection.