Label-efficient sleep staging from multi-channel EEG with self-supervised contrastive learning and iterative self-distillation
Manual sleep stage classification from polysomnography (PSG) is labor-intensive and subject to expert variability, motivating automated and deployment-oriented solutions for clinical use. We present a multi-channel self-supervised learning (SSL) contrastive framework combined with iterative self-distillation for accurate and label-efficient sleep staging. The approach employs a dual-branch convolutional network that processes electroencephalogram (EEG) channels independently and integrates complementary information via a cross-attention fusion module. During pre-training, a contrastive objective leverages temporal adjacency to form positive pairs and maintains hard negatives dynamically to learn robust representations from unlabeled data. Subsequent fine-tuning with minimal labels is enhanced by iterative self-distillation through pseudo-label refinement. On the Sleep-EDF Expanded (SleepEDF-v2) dataset, the method achieves strong performance with only 1% labeled data (accuracy 76.31%, macro-F1 66.53%), competitive against existing SSL baselines. The resulting compact model and single-site training setup align with practical constraints in hospitals and wearable scenarios, reducing annotation burden and supporting secure, scalable clinical deployment. • Dual-channel SSL with cross-attention and self-distillation for sleep staging. • Label-efficient: 76.31% accuracy and 66.53% macro-F1 using only 1% labels. • Compact model and single-site setup support resource-constrained deployment.
Authors
- Fried-Michael Dahlweid (ORCID: https://orcid.org/0000-0002-5416-9915)
- Yiming Chen (ORCID: https://orcid.org/0000-0002-9569-1929)
- Peng Xiao (ORCID: https://orcid.org/0000-0003-1854-0585)
- Yuanwang Wei
- Zou Lai
- Jie Ouyang
- Jingxue Chen
Institutions
- University of Electronic Science and Technology of China (CN)
- Zhejiang University of Science and Technology (CN)
- Chengdu University of Information Technology (CN)
- Chengdu University of Technology (CN)
- Guangzhou University (CN)
- Zhejiang Lab (CN)
- Associazione Italiana Vulvodinia Onlus (IT)
- Zhejiang University (CN)
Publication Details
- Journal
- Array
- Published
- 2026-02-17
- DOI
- https://doi.org/10.1016/j.array.2026.100718
- Primary Topic
- EEG and Brain-Computer Interfaces
- Type
- article
- Field-Weighted Citation Impact
- 0.00