AI Generalisation Gap In Comorbid Sleep Disorder Staging

View PDF HTML (experimental)

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 this https URL

Subjects:

Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

Cite as: arXiv:2603.23582 [cs.LG]

(or arXiv:2603.23582v2 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2603.23582

arXiv-issued DOI via DataCite

Submission history

From: Saswata Bose [view email] [v1] Tue, 24 Mar 2026 15:02:35 UTC (1,199 KB) [v2] Thu, 26 Mar 2026 18:13:16 UTC (1,199 KB)