Single-channel EEG sleep stage classification using synchrosqueezed transform and sequential representation learning
- 1Department of Electrical and Computer Engineering, Tarbiat Modares University
- 2Department of Computer Science and Engineering, Michigan State University
Abstract
Accurate classification of sleep stages is essential for both sleep research and clinical practice, yet remains challenging due to the complexity of EEG signals and feature overlap between stages. This study introduces a novel three-stage approach using single-channel EEG signals from the Sleep-EDF dataset. We focused on the Fpz-Cz channel, applying the SynchroSqueezed Transform (SST) to convert signals into time-frequency representations (TFRs). To address the complexity of classifying long, 3000-sample signals, we first divided each 30-second epoch into smaller, 256-sample segments. In Stage 1, a Convolutional Neural Network (CNN) was trained on these segments to initialize the model’s weights. In Stage 2, contrastive learning further refined the encoder, improving feature separability between sleep stages. This two-step pretraining established a robust foundation for feature extraction. In Stage 3, we returned to the full 3000-sample signals and used the pretrained encoder in a time-distributed manner, extracting features from 256-sample windows. This enabled the encoder to process each segment, with the extracted features then passed to two stacked Gated Recurrent Units (GRUs) to capture long-term dependencies for final classification. This model achieved an accuracy of 83%, highlighting the efficacy of our approach for sleep stage classification.
Related
- Machine learning-based typing of Salmonella enterica O-serogroups by the Fourier-Transform Infrared (FTIR) Spectroscopy-based IR Biotyper system
- High-dimensional hippocampus representation facilitates learning of low-dimensional abstract representation and supports generalization
- Electroencephalography-based Alzheimer's disease detection using multi-scale information retention transform: a subject-level comparison of machine learning classifiers
- Tri-Modality Representation Learning for Molecular Property Prediction
- CUBE: Multimodal Representation Learning Reveals Biological Structure Across Histomorphology, Spatial Protein Phenotypes, and Transcriptome-Associated Signals
BibTeX
@article{aram2026singlechannel,
title = {{Single-channel EEG sleep stage classification using synchrosqueezed transform and sequential representation learning}},
author = {Sadaf Aram and Mohammad M. Ghassemi and Babak Mohammadzadeh Asl},
journal = {Scientific Reports},
year = {2026},
month = jul,
doi = {10.1038/s41598-026-62832-5},
url = {https://doi.org/10.1038/s41598-026-62832-5},
note = {CC BY},
}
RIS
TY - JOUR
TI - Single-channel EEG sleep stage classification using synchrosqueezed transform and sequential representation learning
AU - Sadaf Aram
AU - Mohammad M. Ghassemi
AU - Babak Mohammadzadeh Asl
PY - 2026
DA - 2026/07/01
JO - Scientific Reports
DO - 10.1038/s41598-026-62832-5
UR - https://doi.org/10.1038/s41598-026-62832-5
AB - Accurate classification of sleep stages is essential for both sleep research and clinical practice, yet remains challenging due to the complexity of EEG signals and feature overlap between stages. This study introduces a novel three-stage approach using single-channel EEG signals from the Sleep-EDF dataset. We focused on the Fpz-Cz channel, applying the SynchroSqueezed Transform (SST) to convert signals into time-frequency representations (TFRs). To address the complexity of classifying long, 3000-sample signals, we first divided each 30-second epoch into smaller, 256-sample segments. In Stage 1, a Convolutional Neural Network (CNN) was trained on these segments to initialize the model’s weights. In Stage 2, contrastive learning further refined the encoder, improving feature separability between sleep stages. This two-step pretraining established a robust foundation for feature extraction. In Stage 3, we returned to the full 3000-sample signals and used the pretrained encoder in a time-distributed manner, extracting features from 256-sample windows. This enabled the encoder to process each segment, with the extracted features then passed to two stacked Gated Recurrent Units (GRUs) to capture long-term dependencies for final classification. This model achieved an accuracy of 83%, highlighting the efficacy of our approach for sleep stage classification.
SN - 2045-2322
ER -