Sleep Track: Automated Detection and Classification of Sleep Stages

Abstract

Sleep is vital for our body’s physical restoration, but sleep disorders can cause various problems. Determining sleep stages is essential for diagnosing and curing such disorders. Polysomnography (PSG) signals are recordings of brain activity, eye movements, muscle activity and other physiological signals that are collected during a sleep study. Insomnia, Sleep Apnea, and Restless Legs Syndrome are some of the sleep problems that can be identified using these signals. However, analysing PSG signals manually can be time-consuming and prone to errors. Deep Learning Models such as Convolutional Neural Networks (CNN), can be used to automate the analysis of PSG signals. CNN is followed by Long-Short Term Memory (LSTM) and CNN are used as a stack ensemble method to recognize patterns in the signals that correspond to different sleep stages and events. By training these models on large datasets of PSG signals, they can detect the disorders. The dataset is collected from PhysioNet Sleep-EDF dataset that consists of PSG signals. The accuracies obtained using different training and testing data using CNN and CNN-LSTM are 95.15% and 83.9% respectively, and using metadata classifier the overall accuracy is increased by 1%. The future enhancement of the paper can be done by considering Heart rate, EEG Pz-oz signals and EEG Pz-oz along with EEG Fpz-cz.