Sound-Based Sleep Staging By Exploiting Real-World Unlabeled Data

JongMok Kim, Daewoo Kim, Eunsung Cho, Hai Hong Tran, Joonki Hong, Dongheon Lee, JungKyung Hong, In-Young Yoon, Jeong-Whun Kim, Hyeryung Jang, Nojun Kwak

Keywords: sound sleep staging, real-world unlabeled data, semi-supervised learning, consistency loss, contrastive loss

TL;DR: We propose a novel semi-supervised learning method with the real world sleep sound data, and achieve the significant improvement in various distribution of test dataset.

Abstract: With a growing interest in sleep monitoring at home, sound-based sleep staging with deep learning has emerged as a potential solution. However, collecting labeled data is restrictive in the home environments due to the inconvenience of installing medical equipment at home. To handle this, we propose novel training approaches using accessible real-world sleep sound data. Our key contributions include a new semi-supervised learning technique called sequential consistency loss that considers the time-series nature of sleep sound and a semi-supervised contrastive learning method which handles out-of-distribution data in unlabeled home recordings. Our model was evaluated on various datasets including a labeled home sleep sound dataset and the public PSG-Audio dataset, demonstrating the robustness and generalizability of our model across real-world scenarios.

Anonymity: We agree to keep the submission (including any supplements and/or code) anonymous.

Formatting: We confirm that we read and complied with the author's instructions.