SlumberNet: Deep learning classification of sleep stages using residual neural networks

Abstract

Sleep research is fundamental to understanding health and well-being, as proper sleep is essential for maintaining optimal physiological function. Here we present SlumberNet, a novel deep learning model based on residual network (ResNet) architecture, designed to classify sleep states in mice using electroencephalogram (EEG) and electromyogram (EMG) signals. Our model was trained and tested on data from mice undergoing baseline sleep, sleep deprivation, and recovery sleep, enabling it to handle a wide range of sleep conditions. Employing k-fold cross-validation and data augmentation techniques, SlumberNet achieved high levels of accuracy ([~]98%) in predicting sleep stages and showed robust performance even with a small and diverse training dataset. Comparison of SlumberNets performance to manual sleep stage classification revealed a significant reduction in analysis time ([~]50x faster), without sacrificing accuracy. Our study showcases the potential of deep learning to facilitate sleep research by providing a more efficient, accurate, and scalable method for sleep stage classification. Our work with SlumberNet demonstrates the power of deep learning in sleep research, and looking forward, SlumberNet could be adapted to human EEG analysis and sleep stage classification. Thus, SlumberNet could be a valuable tool in understanding both sleep physiology and disorders in mammals.