Multi-source unsupervised domain-adaptation for automatic sleep staging.

Yangxin Zhu,Shikui Tu,Lin Zhang,Lei Xu
DOI: https://doi.org/10.1109/BIBM58861.2023.10385825
2023-01-01
Abstract:Sleep staging using electroencephalogram (EEG) is of great significance for diagnosing sleep disorders. Recently, due to the high cost of manually annotating EEG signals and the problem of domain shift across different datasets, many unsupervised domain adaptation methods have been applied to sleep staging. However, all these methods are single-source unsupervised domain adaptation (SUDA) methods. When applied to the multi-source unsupervised domain adaptation (MUDA) problem, most SUDA methods treat all source domains equally, without considering the domain shift between them. As a result, achiving feature alignment across all domains becomes a challenge for these SUDA methods. In our proposed model, we address the challenge of domain shift in multi-source unsupervised domain adaptation (MUDA) for automatic sleep staging. Specially, we create a domain-specific branch for each pair of source and target domains, which focuses on aligning the features extracted from the respective domains using an adversarial-based domain adaptation approach. Additionally, we incorporate a domain-invariant branch into our model for learning the domain-invariant representation. We also introduce an adaptive-based mixing strategy to assigns weights to each branches, which can dynamically adjust the importance of each branch depending on their relevance to the specific prediction task. The experiments conducted on three public datasets show the superior performance of our model. Compared to other state-of-the-art MUDA models, our model’s average classification accuracy improves by 2% to 9.9%. The source code is available at https://github.com/CMACH508/MUDAEEG.
What problem does this paper attempt to address?