English

AI Generalisation Gap In Comorbid Sleep Disorder Staging

Machine Learning 2026-03-30 v2 Artificial Intelligence

Abstract

Accurate sleep staging is essential for diagnosing OSA and hypopnea in stroke patients. Although PSG is reliable, it is costly, labor-intensive, and manually scored. While deep learning enables automated EEG-based sleep staging in healthy subjects, our analysis shows poor generalization to clinical populations with disrupted sleep. Using Grad-CAM interpretations, we systematically demonstrate this limitation. We introduce iSLEEPS, a newly clinically annotated ischemic stroke dataset (to be publicly released), and evaluate a SE-ResNet plus bidirectional LSTM model for single-channel EEG sleep staging. As expected, cross-domain performance between healthy and diseased subjects is poor. Attention visualizations, supported by clinical expert feedback, show the model focuses on physiologically uninformative EEG regions in patient data. Statistical and computational analyses further confirm significant sleep architecture differences between healthy and ischemic stroke cohorts, highlighting the need for subject-aware or disease-specific models with clinical validation before deployment. A summary of the paper and the code is available at https://himalayansaswatabose.github.io/iSLEEPS_Explainability.github.io/

Keywords

Cite

@article{arxiv.2603.23582,
  title  = {AI Generalisation Gap In Comorbid Sleep Disorder Staging},
  author = {Saswata Bose and Suvadeep Maiti and Shivam Kumar Sharma and Mythirayee S and Tapabrata Chakraborti and Srijitesh Rajendran and Raju S. Bapi},
  journal= {arXiv preprint arXiv:2603.23582},
  year   = {2026}
}
R2 v1 2026-07-01T11:36:06.124Z