Abstract:Despite the progress recently made towards automatic sleep staging for adults, children have complicated sleep structures that require attention to the pediatric sleep staging. Semi-supervised learning, i.e., training networks with both labeled and unlabeled data, greatly reduces the burden of epoch-by-epoch annotation for physicians. However, the inherent class-imbalance problem in sleep staging task undermines the effectiveness of semi-supervised methods such as pseudo-labeling. In this paper, we propose a Bi-Stream Adversarial Learning network (BiSALnet) to generate pseudo-labels with higher confidence for network optimization. Adversarial learning strategy is adopted in Student and Teacher branches of the two-stream networks. The similarity measurement function minimizes the divergence between the outputs of the Student and Teacher branches, and the discriminator continuously enhances its discriminative ability. In addition, we employ a powerful symmetric positive definite (SPD) manifold structure in the Student branch to capture the desired feature distribution properties. The joint discriminative power of convolutional features and nonlinear complex information aggregated by SPD matrices is combined by the attention feature fusion module to improve the sleep stage classification performance. The BiSALnet is tested on pediatric dataset collected from local hospital. Experimental results show that our method yields the overall classification accuracy of 0.80, kappa of 0.73 and F1-score of 0.76. We also examine the generality of our method on a well-known public dataset Sleep-EDF. Our BiSALnet exhibits noticeable performance with accuracy of 0.91, kappa of 0.85 and F1-score of 0.77. Remarkably, we have obtained comparable performance with state-of-the-art supervised approaches with fairly limited labeled data.

A Semi-supervised Multi-scale Arbitrary Dilated Convolution Neural Network for Pediatric Sleep Staging

CareSleepNet: A Hybrid Deep Learning Network for Automatic Sleep Staging

Adversarial learning for semi-supervised pediatric sleep staging with single-EEG channel

AFSleepNet: Attention-Based Multi-View Feature Fusion Framework for Pediatric Sleep Staging

MAGSleepNet: Adaptively multi-scale temporal focused sleep staging model for multi-age groups

A Sequential End-to-End Neonatal Sleep Staging Model with Squeeze and Excitation Blocks and Sequential Multi-scale Convolution Neural Networks

Sleep Staging Based on Multi Scale Dual Attention Network

MtCLSS: Multi-Task Contrastive Learning for Semi-Supervised Pediatric Sleep Staging

A Novel Sleep Staging Network Based on Data Adaptation and Multimodal Fusion.

Single Channel Sleep Staging Based on Unsupervised Feature Learning

Automatic Sleep Staging Based on Curriculum Learning Approach

A Novel Sleep Staging Network Based on Multi-Scale Dual Attention

SHNN: A single-channel EEG sleep staging model based on semi-supervised learning

A Hybrid Neonatal Sleep Staging Method Based on Convolutional Neural Networks and Graph Neural Networks

Extracting Multi-Scale and Salient Features by MSE Based U-Structure and CBAM for Sleep Staging

Multi-Modal Physiological Signals Based Squeeze-and-Excitation Network With Domain Adversarial Learning for Sleep Staging

Dual-modal and Multi-Scale Deep Neural Networks for Sleep Staging Using EEG and ECG Signals.

SalientSleepNet: Multimodal Salient Wave Detection Network for Sleep Staging

Pediatric Automatic Sleep Staging: A comparative study of state-of-the-art deep learning methods

Deep Learning for Pediatric Sleep Staging from Photoplethysmography: A Transfer Learning Approach from Adults to Children