Keondo Park, Joopyo Hong, Wooseok Lee, Hyun-Woo Shin, Hyung-Sin Kim
{"title":"DistillSleep: Real-Time, On-Device, Interpretable Sleep Staging from Single-Channel EEG.","authors":"Keondo Park, Joopyo Hong, Wooseok Lee, Hyun-Woo Shin, Hyung-Sin Kim","doi":"10.1093/sleep/zsaf240","DOIUrl":null,"url":null,"abstract":"<p><strong>Study objectives: </strong>Polysomnography (PSG) is the current gold standard for sleep staging but requires laboratory equipment, multiple sensors, and labor-intensive manual scoring. We developed DistillSleep, a single-channel electroencephalogram (EEG) framework that delivers accurate, real-time, and interpretable sleep staging on resource-constrained devices.</p><p><strong>Methods: </strong>DistillSleep consists of (1) a high-capacity teacher model and (2) a 109 k-parameter student model designed for edge deployment. Both incorporate a Multi-Wavelength Pyramid module and Transformer-based architecture to capture intra- and inter-epoch features. Feature- and prediction-level knowledge distillation transfers the teacher's expertise to the student. Training and evaluation used >10 000 overnight recordings from six cohorts (SHHS1, PhysioNet 2018, DCSM, KISS, SleepEDF-78, ISRUC), following AASM guidelines. Performance was assessed with Macro-F1.</p><p><strong>Results: </strong>The teacher achieved state-of-the-art Macro-F1 scores (SHHS1 81.1%, PhysioNet 78.9%, DCSM 81.2%, KISS 80.0%) and provided frequency-resolved saliency maps, inter-epoch context and well-calibrated confidence (ECE 0.07). The student maintained competitive accuracy (up to 79.7% Macro-F1) while executing <10 ms per 30-second epoch on three embedded platforms (Raspberry Pi 4B, Jetson orin nano, Coral dev board), reducing computational load 115-fold versus the best prior method (SleePyCo). Interpretability was transferred intact to the student, offering clinicians frequency-band importance and inter-epoch context visualizations, and calibration was further improved by 2.7$\\times$.</p><p><strong>Conclusions: </strong>DistillSleep combines expert-level accuracy, millisecond-scale latency, and transparent decision logic in a single-channel EEG form factor. These capabilities pave the way for point-of-care diagnostics, same-night therapy titration, and large-scale home monitoring, expanding the reach of sleep medicine while retaining clinical trust.</p>","PeriodicalId":22018,"journal":{"name":"Sleep","volume":" ","pages":""},"PeriodicalIF":4.9000,"publicationDate":"2025-08-22","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"Sleep","FirstCategoryId":"3","ListUrlMain":"https://doi.org/10.1093/sleep/zsaf240","RegionNum":2,"RegionCategory":"医学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"Medicine","Score":null,"Total":0}
引用次数: 0
Abstract
Study objectives: Polysomnography (PSG) is the current gold standard for sleep staging but requires laboratory equipment, multiple sensors, and labor-intensive manual scoring. We developed DistillSleep, a single-channel electroencephalogram (EEG) framework that delivers accurate, real-time, and interpretable sleep staging on resource-constrained devices.
Methods: DistillSleep consists of (1) a high-capacity teacher model and (2) a 109 k-parameter student model designed for edge deployment. Both incorporate a Multi-Wavelength Pyramid module and Transformer-based architecture to capture intra- and inter-epoch features. Feature- and prediction-level knowledge distillation transfers the teacher's expertise to the student. Training and evaluation used >10 000 overnight recordings from six cohorts (SHHS1, PhysioNet 2018, DCSM, KISS, SleepEDF-78, ISRUC), following AASM guidelines. Performance was assessed with Macro-F1.
Results: The teacher achieved state-of-the-art Macro-F1 scores (SHHS1 81.1%, PhysioNet 78.9%, DCSM 81.2%, KISS 80.0%) and provided frequency-resolved saliency maps, inter-epoch context and well-calibrated confidence (ECE 0.07). The student maintained competitive accuracy (up to 79.7% Macro-F1) while executing <10 ms per 30-second epoch on three embedded platforms (Raspberry Pi 4B, Jetson orin nano, Coral dev board), reducing computational load 115-fold versus the best prior method (SleePyCo). Interpretability was transferred intact to the student, offering clinicians frequency-band importance and inter-epoch context visualizations, and calibration was further improved by 2.7$\times$.
Conclusions: DistillSleep combines expert-level accuracy, millisecond-scale latency, and transparent decision logic in a single-channel EEG form factor. These capabilities pave the way for point-of-care diagnostics, same-night therapy titration, and large-scale home monitoring, expanding the reach of sleep medicine while retaining clinical trust.
期刊介绍:
SLEEP® publishes findings from studies conducted at any level of analysis, including:
Genes
Molecules
Cells
Physiology
Neural systems and circuits
Behavior and cognition
Self-report
SLEEP® publishes articles that use a wide variety of scientific approaches and address a broad range of topics. These may include, but are not limited to:
Basic and neuroscience studies of sleep and circadian mechanisms
In vitro and animal models of sleep, circadian rhythms, and human disorders
Pre-clinical human investigations, including the measurement and manipulation of sleep and circadian rhythms
Studies in clinical or population samples. These may address factors influencing sleep and circadian rhythms (e.g., development and aging, and social and environmental influences) and relationships between sleep, circadian rhythms, health, and disease
Clinical trials, epidemiology studies, implementation, and dissemination research.