Application of Machine Learning to Sleep Stage Classification

TL;DR

This study evaluates machine learning methods for automated sleep stage classification using single-channel EEG and EMG data in rodents, achieving high accuracy and reducing manual effort and variability.

Contribution

It introduces an open-access, automated classifier for sleep stages based on single EEG/EMG, demonstrating high accuracy with multiple machine learning techniques.

Findings

Random Forest achieved 95.78% accuracy.

Artificial Neural Network achieved 93.31% accuracy.

Various classifiers can reliably predict sleep stages from minimal data.

Abstract

Sleep studies are imperative to recapitulate phenotypes associated with sleep loss and uncover mechanisms contributing to psychopathology. Most often, investigators manually classify the polysomnography into vigilance states, which is time-consuming, requires extensive training, and is prone to inter-scorer variability. While many works have successfully developed automated vigilance state classifiers based on multiple EEG channels, we aim to produce an automated and open-access classifier that can reliably predict vigilance state based on a single cortical electroencephalogram (EEG) from rodents to minimize the disadvantages that accompany tethering small animals via wires to computer programs. Approximately 427 hours of continuously monitored EEG, electromyogram (EMG), and activity were labeled by a domain expert out of 571 hours of total data. Here we evaluate the performance of various machine learning techniques on classifying 10-second epochs into one of three discrete classes: paradoxical, slow-wave, or wake. Our investigations include Decision Trees, Random Forests, Naive Bayes Classifiers, Logistic Regression Classifiers, and Artificial Neural Networks. These methodologies have achieved accuracies ranging from approximately 74% to approximately 96%. Most notably, the Random Forest and the ANN achieved remarkable accuracies of 95.78% and 93.31%, respectively. Here we have shown the potential of various machine learning classifiers to automatically, accurately, and reliably classify vigilance states based on a single EEG reading and a single EMG reading.