An EEG-based Machine Learning framework for diagnosing Acute Sleep Deprivation

Daya Kumar^1,2*Apurva Narayan^1,2Saptharishi Lalgudi Ganesan^3,4,5,6

• ¹The University of Western Ontario, Ontario, Canada
• ²International Center for Applied Systems Science for Sustainable Development (ICASSSD), Cambridge, Canada
• ³London Health Sciences Centre Children's Hospital Pediatric Critical Care Unit, London, Canada
• ⁴Western University, Western Institute for Neurosciences, London, Canada
• ⁵Children's Health Research Institute, London, Canada
• ⁶Western University Schulich School of Medicine & Dentistry, London, Canada

Study Objective: Acute sleep deprivation significantly impacts cognitive function, contributes to accidents, and increases the risk of chronic illnesses, underscoring the need for reliable and objective diagnosis. Our work aims to develop a machine learning-based approach to discriminate between EEG recordings from acutely sleep-deprived individuals and those that are well-rested, facilitating the objective detection of acute sleep deprivation and enabling timely intervention to mitigate its adverse effects. Methods: Sixty-one-channel eyes-open resting-state electroencephalography (EEG) data from a publicly available dataset of 71 participants were analyzed. Following preprocessing, EEG recordings were segmented into contiguous, non-overlapping 20-second epochs. For each epoch, a comprehensive set of features was extracted, including statistical descriptors, spectral measures, functional connectivity indices, and graph-theoretic metrics. Four machine learning classifiers—Light Gradient Boosting Machine (LightGBM), eXtreme Gradient Boosting (XGBoost), Random Forest (RF), and Support Vector Classifier (SVC)—were trained on these features using nested stratified cross-validation to ensure unbiased performance evaluation. In parallel, three deep learning models—a Convolutional Neural Network (CNN), Long Short-Term Memory network (LSTM), and Transformer—were trained directly on the raw multi-channel EEG time-series data. All models were evaluated under two conditions: (i) without subject-level separation, allowing the same participant to contribute to both training and test sets, and (ii) with subject-level separation, where models were tested exclusively on entirely unseen participants. Model performance was assessed using accuracy, F1-score, and area under the receiver operating characteristic curve (AUC). Results: Without subject-level separation, CNN achieved the highest accuracy (95.72%), followed closely by XGBoost (95.42%), LightGBM (94.83%), and RF (94.53%), with SVC at 85.25%. LSTM (66.75%) and Transformer (77.39%) performed substantially lower. Under subject-level separation, RF achieved the highest accuracy (68.23%), followed by XGBoost (66.36%), LightGBM (66.21%), CNN (65.35%), and SVC (65.08%), with LSTM (61.70%) and Transformer (63.35%) lowest. Conclusions: This study demonstrates the strong potential of EEG-based machine learning for acute sleep deprivation detection, while also underscoring the challenges of achieving robust subject-level generalization. Despite reduced accuracy under cross-subject evaluation, the findings support the feasibility of developing scalable, non-invasive tools for sleep deprivation detection using EEG and advanced ML techniques.