SwinSleep: A Deep Learning Framework Advancing Overnight Sleep Staging Toward Clinical Practice

Di Zhang,Yichong She,Jinbo Sun,Xuejuan Yang,Xiao Zeng,Wei Qin
DOI: https://doi.org/10.1109/tim.2024.3470049
IF: 5.6
2024-10-16
IEEE Transactions on Instrumentation and Measurement
Abstract:Sleep staging is fundamental to sleep research and essential in diagnosing sleep disorders. Computers have made significant advancements in sleep staging, matching, or exceeding human scorers on certain evaluation metrics. However, the translation of automated methods into clinical practice still faces significant challenges. Variations in decision flavors between automated algorithms and human experts pose limitations on the acceptance of automated staging by healthcare professionals. To reduce these disparities, we demonstrate a deep learning framework named SwinSleep for polysomnography (PSG) feature extraction and sleep scoring. SwinSleep is a 1-D adaptation of Swin Transformer, tailored to the sleep physiological signals. It takes whole PSG data as input and produces overnight hypnograms in a single pass, enabling the capture of human scoring tendencies. We also introduced hypnogram-based sleep metrics for model evaluations, capturing the clinical significance and, to some extent, mirroring human scorer tendencies. SwinSleep has been rigorously developed and evaluated on a dataset comprising 22380 PSG recordings from multiple cohorts, which outperforms other similar studies in the mixed-cohort evaluation. On hold-out evaluations, SwinSleep surpasses two other publicly available methods in classification metrics by 1.7%–7.4% or more, also surpasses human scorers by 7.2%. In addition to classification performance, SwinSleep significantly enhances the prediction of overnight sleep metrics, outperforming established benchmark algorithms. It instills greater confidence in sleep physicians in the automated staging process, providing a valuable tool to improve efficiency and consistency in professional sleep centers.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?