A robust two-stage sleep spindle detection approach using single-channel EEG

Abstract

Objective. Sleep spindles in the electroencephalogram (EEG) are significant in sleep analysis related to cognitive functions and neurological diseases, and thus are of great clinical interests. An automatic sleep spindle detection algorithm could help decrease the workload of visual inspection by sleep clinicians. Approach. We propose a robust two-stage approach for sleep spindle detection using single-channel EEG. In the pre-detection stage, a stable number of sleep spindle candidates are discovered using the Teager energy operator with adaptive parameters, where the number of true sleep spindles are ensured as many as possible to maximize the detection sensitivity. In the refinement stage, representative features are designed and a bagging classifier is exploited to further recognize the true spindles from all candidates, in order to remove the false detection in the first stage. Main results. Using the union of all experts’ annotations as the ground truth, its performance outperforms state-of-the-art works in terms of F1-score (F1) on two public databases (F1: 0.814 for Montreal archive of sleep studies dataset and 0.690 for DREAMS dataset). The annotation consistency between the proposed method and certain selected expert as the trainer could exceed the consistency between two human experts. Significance. The proposed sleep spindle detection method is based on single-channel EEG thus introduces as less interference to the subjects as possible. It is robust to subject variations between databases and is capable of learning certain annotation rules, which is expected to help facilitate the manual labeling of certain experts. In addition, this method is fast enough for real-time applications.