DREAM: Domain Invariant and Contrastive Representation for Sleep Dynamics
DREAM: Domain Invariant and Contrastive Representation for Sleep Dynamics
复制标题
DOI:
10.1109/icdm54844.2022.00126
复制
发表时间:
2022-11
期刊:
影响因子:
--
通讯作者:
Seungyeon Lee;Thai-Hoang Pham;Ping Zhang
中科院分区:
文献类型:
--
作者:
Seungyeon Lee;Thai-Hoang Pham;Ping Zhang
Sleep staging is a key challenge in diagnosing and treating sleep-related diseases due to its labor-intensive, time-consuming, costly, and error-prone. With the availability of large-scale sleep signal data, many deep learning methods are proposed for automatic sleep staging. However, these existing methods face several challenges including the heterogeneity of patients’ underlying health conditions and the difficulty modeling complex interactions between sleep stages. In this paper, we propose a neural network architecture named DREAM to tackle these issues for automatic sleep staging. DREAM consists of (i) a feature representation network that generates robust representations for sleep signals via the variational auto-encoder framework and contrastive learning and (ii) a sleep stage classification network that explicitly models the interactions between sleep stages in the sequential context at both feature representation and label classification levels via Transformer and conditional random field architectures. Our experimental results indicate that DREAM significantly outperforms existing methods for automatic sleep staging on three sleep signal datasets.