Automatic Wake and Deep-Sleep Stage Classification Based on Wigner–Ville Distribution Using a Single Electroencephalogram Signal

Abstract

This research paper outlines a method for automatically classifying wakefulness and deep sleep stage (N3) based on the American Academy of Sleep Medicine (AASM) standards. The study employed a single-channel EEG signal, leveraging the Wigner–Ville Distribution (WVD) for time–frequency analysis to determine EEG energy per second in specific frequency bands (δ, θ, α, and entire band). Particle Swarm Optimization (PSO) was used to optimize thresholds for distinguishing between wakefulness and stage N3. This process aims to mimic a sleep technician’s visual scoring but in an automated fashion, with features and thresholds extracted to classify epochs into correct sleep stages. The study’s methodology was validated using overnight PSG recordings from 20 subjects, which were evaluated by a technician. The PSG setup followed the 10–20 standard system with varying sampling rates from different hospitals. Two baselines, T1 for the wake stage and T2 for the N3 stage, were calculated using PSO to ascertain the best thresholds, which were then used to classify EEG epochs. The results showed high sensitivity, accuracy, and kappa coefficient, indicating the effectiveness of the classification algorithm. They suggest that the proposed method can reliably determine sleep stages, being aligned closely with the AASM standards and offering an intuitive approach. The paper highlights the strengths of the proposed method over traditional classifiers and expresses the intentions to extend the algorithm to classify all sleep stages in the future.