Phan Huy, Lorenzen Kristian P, Heremans Elisabeth, Chen Oliver Y, Tran Minh C, Koch Philipp, Mertins Alfred, Baumert Mathias, Mikkelsen Kaare B, De Vos Maarten
IEEE J Biomed Health Inform. 2023 Oct;27(10):4748-4757. doi: 10.1109/JBHI.2023.3303197. Epub 2023 Oct 5.
Human sleep is cyclical with a period of approximately 90 minutes, implying long temporal dependency in the sleep data. Yet, exploring this long-term dependency when developing sleep staging models has remained untouched. In this work, we show that while encoding the logic of a whole sleep cycle is crucial to improve sleep staging performance, the sequential modelling approach in existing state-of-the-art deep learning models are inefficient for that purpose. We thus introduce a method for efficient long sequence modelling and propose a new deep learning model, L-SeqSleepNet, which takes into account whole-cycle sleep information for sleep staging. Evaluating L-SeqSleepNet on four distinct databases of various sizes, we demonstrate state-of-the-art performance obtained by the model over three different EEG setups, including scalp EEG in conventional Polysomnography (PSG), in-ear EEG, and around-the-ear EEG (cEEGrid), even with a single EEG channel input. Our analyses also show that L-SeqSleepNet is able to alleviate the predominance of N2 sleep (the major class in terms of classification) to bring down errors in other sleep stages. Moreover the network becomes much more robust, meaning that for all subjects where the baseline method had exceptionally poor performance, their performance are improved significantly. Finally, the computation time only grows at a sub-linear rate when the sequence length increases.
人类睡眠具有周期性,周期约为90分钟,这意味着睡眠数据存在长期的时间依赖性。然而,在开发睡眠分期模型时探索这种长期依赖性仍未得到关注。在这项工作中,我们表明,虽然编码整个睡眠周期的逻辑对于提高睡眠分期性能至关重要,但现有最先进的深度学习模型中的序列建模方法在实现这一目的时效率低下。因此,我们引入了一种高效的长序列建模方法,并提出了一种新的深度学习模型L-SeqSleepNet,该模型在睡眠分期时考虑了全周期睡眠信息。在四个不同规模的数据库上对L-SeqSleepNet进行评估,我们证明了该模型在三种不同的脑电图设置下取得了领先的性能,包括传统多导睡眠图(PSG)中的头皮脑电图、入耳式脑电图和耳周脑电图(cEEGrid),即使是单通道脑电图输入。我们的分析还表明,L-SeqSleepNet能够减轻N2睡眠(分类中的主要类别)的主导地位,以减少其他睡眠阶段的错误。此外,该网络变得更加稳健,这意味着对于所有基线方法表现异常差的受试者,他们的表现都有显著改善。最后,当序列长度增加时,计算时间仅以亚线性速率增长。