Reproduce “Application of Machine Learning to Sleep Stage Classification” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

Andrew Smith, Hardik Anand, Snezana Milosavljevic, Katherine M. Rentschler, Ana Pocivavsek, Homayoun Valafar

Researchers at the University of South Carolina developed an automated classifier for rodent vigilance states using a single cortical EEG channel and an EMG signal. The Random Forest model achieved 95.78% accuracy, demonstrating that machine learning can reliably classify sleep stages with reduced data, facilitating untethered telemetry in sleep research.

Key insights

  • Machine learning techniques can effectively classify rodent vigilance states with high accuracy, even when utilizing limited data from only a single EEG channel and an EMG signal.
  • The 50-second windowing approach for feature input significantly improved the temporal representation of sleep states, critical for accurate classification from reduced data.
  • Balancing the dataset by oversampling the underrepresented Paradoxical sleep class was crucial for improving model generalization and training stability.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.