Advancing Sleep Disorder Diagnostics: A Transformer-based EEG Model for Sleep Stage Classification and OSA Prediction

Cheng Wan, Micky C. Nnamdi, Wenqi Shi, Benjamin Smith, Chad Purnell, May Dongmei Wang

Keywords: sleep stage classification, obstructive sleep apnea, transformer, clinical decision support

Abstract:

Sleep disorders, particularly Obstructive Sleep Apnea (OSA), have a considerable effect on an individual's health and quality of life. Accurate sleep stage classification and prediction of OSA are crucial for timely diagnosis and effective management of sleep disorders.
In this study, we propose a sequential network that enhances sleep stage classification by incorporating self-attention mechanisms and Conditional Random Fields (CRF) into a deep learning model comprising multi-kernel Convolutional Neural Networks (CNNs) and Transformer-based encoders. The self-attention mechanism enables the model to focus on the most discriminative features extracted from single-channel electroencephalography (EEG) recordings, while the CRF module captures the temporal dependencies between sleep stages, improving the model's ability to learn more plausible sleep stage sequences.
Moreover, we explore the relationship between sleep stages and OSA severity by utilizing the predicted sleep stage features to train various regression models for Apnea-Hypopnea Index (AHI) prediction.
Our experiments demonstrate an improved sleep stage classification performance of 78.7\%, particularly on datasets with diverse AHI values, and highlight the potential of leveraging sleep stage information for monitoring OSA.
By employing advanced deep learning techniques, we thoroughly explore the intricate relationship between sleep stages and sleep apnea, laying the foundation for more precise and automated diagnostics of sleep disorders.

Track: 10. Digital health

Registration Id: 3ZN22WR742N

Submission Number: 271