BTCRSleep: a boundary temporal context refinement-based fully convolutional network for sleep staging with single-channel EEG

This study attempts to capture the boundary context, which contains the characteristics of brain waves during sleep stage transition, to improve the performance of sleep staging. Approach. In this paper we propose a fully convolutional network with boundary temporal context refinement, called BTCRSleep (Boundary Temporal Context Refinement Sleep). The boundary temporal context refinement module refines the boundary information on sleep stages on the basis of extracting multi-scale temporal dependences between epochs and enhances the abstract capability of the boundary temporal context. In addition, we design a class-aware data augmentation method to effectively learn the boundary temporal context between the minority class and other sleep stages. Main results. We evaluate the performance of our proposed network using four public datasets: the 2013 version of Sleep-EDF Expanded (SEDF), the 2018 version of Sleep-EDF Expanded (SEDFX), the Sleep Heart Health Study (SHHS) and CAP Sleep Database (CAP). The evaluation results on the four datasets showed that our model obtains the best total accuracy and kappa score compared with state-of-the-art methods. On average, accuracies of 84.9% in SEDF, 82.9% in SEDFX, 85.2% in SHHS and 76.9% in CAP are obtained under subject-independent cross-validation. We demonstrate that the boundary temporal context contributes to the improvement in capturing the temporal dependences across different epochs.
Source: Physiological Measurement - Category: Physiology Authors: Source Type: research