Automated Vigilance State Classification in Rodents Using Machine Learning and Feature Engineering

TL;DR

This paper introduces an automated machine learning framework that classifies rodent sleep states from EEG data with high accuracy, enhancing throughput and reproducibility in preclinical sleep research.

Contribution

It presents a novel feature engineering and machine learning approach, specifically using XGBoost, for accurate automated vigilance state classification in rodents.

Findings

Achieved 91.5% overall accuracy in classification

Outperformed baseline methods in sleep state detection

Validated during a major data science competition

Abstract

Preclinical sleep research remains constrained by labor intensive, manual vigilance state classification and inter rater variability, limiting throughput and reproducibility. This study presents an automated framework developed by Team Neural Prognosticators to classify electroencephalogram (EEG) recordings of small rodents into three critical vigilance states paradoxical sleep (REM), slow wave sleep (SWS), and wakefulness. The system integrates advanced signal processing with machine learning, leveraging engineered features from both time and frequency domains, including spectral power across canonical EEG bands (delta to gamma), temporal dynamics via Maximum-Minimum Distance, and cross-frequency coupling metrics. These features capture distinct neurophysiological signatures such as high frequency desynchronization during wakefulness, delta oscillations in SWS, and REM specific bursts. Validated during the 2024 Big Data Health Science Case Competition (University of South Carolina Big Data Health Science Center, 2024), our XGBoost model achieved 91.5% overall accuracy, 86.8% precision, 81.2% recall, and an F1 score of 83.5%, outperforming all baseline methods. Our approach represents a critical advancement in automated sleep state classification and a valuable tool for accelerating discoveries in sleep science and the development of targeted interventions for chronic sleep disorders. As a publicly available code (BDHSC) resource is set to contribute significantly to advancements.