Personalized Sleep Staging Leveraging Source-free Unsupervised Domain Adaptation

作     者：Zhou, Yangxuan Zhao, Sha Wang, Jiquan Jiang, Haiteng Li, Shijian Luo, Benyan Li, Tao Pan, Gang 

作者机构：State Key Laboratory of Brain-machine Intelligence Zhejiang University Hangzhou China College of Computer Science and Technology Zhejiang University Hangzhou China Department of Neurobiology Affiliated Mental Health Center Hangzhou Seventh People’s Hospital Zhejiang University School of Medicine Hangzhou China MOE Frontier Science Center for Brain Science and Brain-machine Integration Zhejiang University Hangzhou China The First Affiliated Hospital College of Medicine Zhejiang University Hangzhou China 

出 版 物：《arXiv》 (arXiv)

年 卷 期：2024年

核心收录：

主　　题：Sleep research 

摘      要：Sleep staging is important for monitoring sleep quality and diagnosing sleep-related disorders. Recently, numerous deep learning-based models have been proposed for automatic sleep staging using polysomnography recordings. Most of them are trained and tested on the same labeled datasets which results in poor generalization to unseen target domains. However, they regard the subjects in the target domains as a whole and overlook the individual discrepancies, which limits the model’s generalization ability to new patients (i.e., unseen subjects) and plug-and-play applicability in clinics. To address this, we propose a novel Source-Free Unsupervised Individual Domain Adaptation (SF-UIDA) framework for sleep staging, leveraging sequential cross-view contrasting and pseudo-label based fine-tuning. It is actually a two-step subject-specific adaptation scheme, which enables the source model to effectively adapt to newly appeared unlabeled individual without access to the source data. It meets the practical needs in real-world scenarios, where the personalized customization can be plug-and-play applied to new ones. Our framework is applied to three classic sleep staging models and evaluated on three public sleep datasets, achieving the state-of-the-art performance. Copyright © 2024, The Authors. All rights reserved.