SpindleU-Net: An Adaptive U-Net Framework for Sleep Spindle Detection in Single-Channel EEG.

Abstract

The sleep spindles in EEG have become one type of biomarker used to assess cognitive abilities and related disorders, and thus their detection is crucial for clinical research. This task, traditionally performed by sleep experts, is time-consuming. Many methods have been proposed to automate this process, yet an increase in performance is still expected. Inspired by the application in image segmentation, we propose a point-wise spindle detection method based on the U-Net framework with an attention module (SpindleU-Net). It maps the sequences of arbitrary-length EEG inputs to those of dense labels of spindle or non-spindle on freely chosen intervals. The attention module that focuses on the salient spindle region allows better performance, and a task-specific loss function is defined to alleviate the problem of imbalanced classification. As a deep learning method, SpindleU-Net outperforms state-of-the-art methods on the widely used benchmark dataset of MASS as well as the DREAMS dataset with a small number of samples. On MASS dataset it achieves average F1 scores of 0.854 and 0.803 according to its consistency with the annotations by two sleep experts respectively. On DREAMS dataset, it shows the average F1 score of 0.739. Its cross-dataset performance is also better compared to other methods, showing the good generalization ability for cross-dataset applications.