Papers › NeuroNet: A Novel Hybrid Self-Supervised Learning Framework for Sleep Stage...

NeuroNet: A Novel Hybrid Self-Supervised Learning Framework for Sleep Stage Classification Using Single-Channel EEG

10 Apr 2024arXiv:2404.17585archive 2025-07-28

Cheol-Hui Lee, Hakseung Kim, Hyun-jee Han, Min-Kyung Jung, Byung C. Yoon, Dong-Joo Kim

The classification of sleep stages is a pivotal aspect of diagnosing sleep disorders and evaluating sleep quality. However, the conventional manual scoring process, conducted by clinicians, is time-consuming and prone to human bias. Recent advancements in deep learning have substantially propelled the automation of sleep stage classification. Nevertheless, challenges persist, including the need for large datasets with labels and the inherent biases in human-generated annotations. This paper introduces NeuroNet, a self-supervised learning (SSL) framework designed to effectively harness unlabeled single-channel sleep electroencephalogram (EEG) signals by integrating contrastive learning tasks and masked prediction tasks. NeuroNet demonstrates superior performance over existing SSL methodologies through extensive experimentation conducted across three polysomnography (PSG) datasets. Additionally, this study proposes a Mamba-based temporal context module to capture the relationships among diverse EEG epochs. Combining NeuroNet with the Mamba-based temporal context module has demonstrated the capability to achieve, or even surpass, the performance of the latest supervised learning methodologies, even with a limited amount of labeled data. This study is expected to establish a new benchmark in sleep stage classification, promising to guide future research and applications in the field of sleep analysis.

PaperPDFCode

Code

dlcjfgmlnasa/NeuroNet officialmentioned in papermentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Contrastive LearningEEGElectroencephalogram (EEG)MambaSelf-Supervised LearningSleep Stage Detection

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Sleep Stage Detection ISRUC-Sleep NeuroNet (C4-A1 only) Accuracy 77.05% #2 of 2 Archive leaderboard report
Sleep Stage Detection ISRUC-Sleep NeuroNet (C4-A1 only) Macro-F1 0.734 #2 of 2 Archive leaderboard report
Sleep Stage Detection ISRUC-Sleep (single-channel) NeuroNet (C4-A1 only) Accuracy 77.05% #1 of 1 Archive leaderboard report
Sleep Stage Detection ISRUC-Sleep (single-channel) NeuroNet (C4-A1 only) Macro-F1 0.734 #1 of 1 Archive leaderboard report
Sleep Stage Detection SHHS NeuroNet (C4-A1 only) Accuracy 86.88% #10 of 10 Archive leaderboard report
Sleep Stage Detection SHHS NeuroNet (C4-A1 only) Macro-F1 0.812 #10 of 10 Archive leaderboard report
Sleep Stage Detection SHHS (single-channel) NeuroNet (C4-A1 only) Accuracy 86.88% #5 of 5 Archive leaderboard report
Sleep Stage Detection SHHS (single-channel) NeuroNet (C4-A1 only) Macro-F1 0.812 #5 of 5 Archive leaderboard report
Sleep Stage Detection Sleep-EDFx NeuroNet (Fpz-Cz only) Accuracy 85.24% #1 of 3 Archive leaderboard report
Sleep Stage Detection Sleep-EDFx NeuroNet (Fpz-Cz only) Macro-F1 0.798 #1 of 3 Archive leaderboard report
Sleep Stage Detection Sleep-EDFx (single-channel) NeuroNet (Fpz-Cz only) Accuracy 85.24% #1 of 3 Archive leaderboard report
Sleep Stage Detection Sleep-EDFx (single-channel) NeuroNet (Fpz-Cz only) Macro-F1 0.798 #1 of 3 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Contrastive Learning

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections