Personalized Sleep Staging Leveraging Source-free Unsupervised Domain Adaptation
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.
Authors
- Benyan Luo (ORCID: https://orcid.org/0000-0002-9892-5778)
- Sha Zhao (ORCID: https://orcid.org/0000-0003-4628-5198)
- Gang Pan (ORCID: https://orcid.org/0000-0002-4049-6181)
- Tao Li (ORCID: https://orcid.org/0000-0003-3831-901X)
- Yangxuan Zhou (ORCID: https://orcid.org/0009-0002-6808-9569)
- Jiquan Wang (ORCID: https://orcid.org/0000-0002-1621-1836)
- Shijian Li (ORCID: https://orcid.org/0000-0001-5846-3065)
- Haiteng Jiang
Institutions
- Allen Institute for Brain Science (US)
- Zhejiang University of Technology (CN)
- Zhejiang University (CN)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i13.33592
- Citations
- 1
- Primary Topic
- Context-Aware Activity Recognition Systems
- Type
- article
- Field-Weighted Citation Impact
- 0.43
Funders
- National Natural Science Foundation of China
- Fundamental Research Funds for the Central Universities
- Natural Science Foundation of Zhejiang Province