SpindleFlexNet: Flexible sleep spindles detection for EEG signals based on an adaptive one-dimensional RetinaNet-based framework
Sleep spindle is a physiologically significant biomedical signal in electroencephalographic (EEG) waveforms, which is typically a low-amplitude event in sleep. Due to the small signal ratio in the overall EEG, previous detection methods have limited capability to capture its start and end points and lack flexibility in handling multi-spindle scenarios. To address the gap, we address the problem from a new perspective and introduce SpindleFlexNet, the first framework in this field to apply deep learning-based one-dimensional object detection, leveraging an adapted one-dimensional RetinaNet architecture. The framework employs one-dimensional anchor generation, matching, and regression, along with a customized one-dimensional loss function. Analyses were conducted on two public datasets: the Montreal Archive of Sleep Studies and DREAMS, from which a total of 11,061 and 335 segments were obtained, respectively. When trained on these datasets, SpindleFlexNet achieved an average recall, precision, and F1-score of 0.61, 0.76, 0.67, and 0.58, 0.80, 0.67 in five-fold cross-validation. The model demonstrates stable detection performance and good generalization, making it a practical tool for sleep research. Potential applications include automated spindle labeling in clinical settings and as a reference for studies combining EEG with simultaneous functional magnetic resonance imaging.