Source-linked AI summary

SalientSleepNet: Multimodal Salient Wave Detection Network for Sleep Staging

Ziyu Jia, Youfang Lin, Jing Wang, Xuehui Wang, Peiyi Xie, Yingbin Zhang

arXiv:2105.13864v1cs.LGeess.SP

TL;DR

Sleep staging remains challenging because salient-wave extraction, multi-scale transition modeling, and stage-specific use of modalities are open issues. SalientSleepNet uses separate EEG and EOG streams with multi-scale extraction and multimodal attention. On two datasets it achieves the best overall performance against seven baselines, and the paper reports that it has the fewest parameters among existing deep learning models.

  • Problem

    Sleep staging still faces open challenges in extracting salient waves from multimodal data, modeling multi-scale transitions, and adapting to stage-specific modality contributions.

    Source: §1 · p. 2
  • Method

    SalientSleepNet uses separate EEG and EOG streams to capture salient features, a multi-scale extraction module for transition rules, and multimodal attention to emphasize useful modality information.

    Source: §1 · p. 2, §4.1 · p. 3, §4.2 · p. 4
  • Results

    On two publicly available datasets, SalientSleepNet achieves the best overall performance among seven baseline methods and is reported to have the fewest parameters among existing deep learning models.

    Source: §5.4 · p. 5, §6 · p. 6
  • Takeaways & Limitations

    The paper presents SalientSleepNet as a general framework for multimodal physiological time series, with experiments supporting state-of-the-art sleep-staging performance.

    Source: §6 · p. 6

    In this work, each sleep epoch is limited to EEG and EOG channels.

    Source: §3 · p. 3

Abstract

from arXiv · show

Sleep staging is fundamental for sleep assessment and disease diagnosis. Although previous attempts to classify sleep stages have achieved high classification performance, several challenges remain open: 1) How to effectively extract salient waves in multimodal sleep data; 2) How to capture the multi-scale transition rules among sleep stages; 3) How to adaptively seize the key role of specific modality for sleep staging. To address these challenges, we propose SalientSleepNet, a multimodal salient wave detection network for sleep staging. Specifically, SalientSleepNet is a temporal fully convolutional network based on the $\rm U^2$-Net architecture that is originally proposed for salient object detection in computer vision. It is mainly composed of two independent $\rm U^2$-like streams to extract the salient features from multimodal data, respectively. Meanwhile, the multi-scale extraction module is designed to capture multi-scale transition rules among sleep stages. Besides, the multimodal attention module is proposed to adaptively capture valuable information from multimodal data for the specific sleep stage. Experiments on the two datasets demonstrate that SalientSleepNet outperforms the state-of-the-art baselines. It is worth noting that this model has the least amount of parameters compared with the existing deep neural network models.

1 Introduction

The introduction motivates automatic sleep staging by identifying gaps in salient-wave extraction, multi-scale transition modeling, and stage-specific use of EEG and EOG. SalientSleepNet addresses these challenges with multimodal salient-feature extraction and multi-scale transition modeling, and is reported to achieve state-of-the-art performance with the fewest parameters among existing deep neural networks.

1 Introduction

  • Manual sleep staging is labor-intensive and time-consuming, motivating automatic methods despite deep learning's remaining challenges in wave extraction, transition modeling, and modality use.Experts classify 30-second epochs from polysomnography recordings collected using several sensors.Source: §1 · p. 1
  • Existing methods often capture salient waves indirectly through hand-extracted features, which require prior knowledge and may lose information.Stage-specific waves include N2 spindles and K-complexes and N3 delta waves.Source: §1 · pp. 1–2, Figure 1 · p. 1
  • Many existing approaches do not explicitly capture the multi-scale structure of sleep-stage transitions, and recurrent models can be difficult to tune and optimize.The AASM standard includes short-, middle-, and long-scale transition patterns.Source: §1 · p. 2, Figure 2 · p. 2
  • The useful modality varies by stage: EOG contributes more than EEG for distinguishing REM from N1, while EEG salient waves guide N2 and N3 classification.Source: §1 · p. 2
  • SalientSleepNet is proposed for EEG and EOG sleep staging, jointly capturing salient multimodal features and multi-scale transition rules from raw signals.Source: §1 · p. 2
  • SalientSleepNet is reported to achieve state-of-the-art performance and have the fewest parameters among existing deep neural network models for sleep staging.Source: §1 · p. 2

2 Related Work

Prior sleep-staging research includes CNN-, RNN-, and feed-forward approaches that extract signal features and model temporal context. The paper identifies a remaining need to detect and fuse salient waves in raw multimodal signals and explicitly capture multi-scale transitions.

2 Related Work

  • Earlier machine-learning methods relied on hand-engineered features requiring substantial prior knowledge, preceding the shift toward deep learning for automatic sleep staging.Source: §2 · p. 2
  • Sleep-staging methods include CNN or RNN models, a sequence-to-sequence approach combining raw signals and time-frequency images, and a feed-forward time-series segmentation model.Examples include DeepSleepNet, SleepEEGNet, XSleepNet, TinySleepNet, and SleepUtime.Source: §2 · p. 2
  • Existing deep learning methods are described as unable to effectively detect and fuse salient waves in raw multimodal signals or explicitly capture multi-scale sleep-stage transitions.Source: §2 · p. 2

3 Preliminaries

The preliminaries formulate sleep staging as mapping a sequence of sleep epochs to a corresponding sequence of stage labels. In this work, each epoch has EEG and EOG channels, and the architecture overview depicts separate modality streams and downstream processing modules.

3 Preliminaries

  • The architecture overview shows independent EEG and EOG U2-structures, a multi-scale feature extraction module, multimodal attention, and a segment-wise classifier.Each U2-structure contains nested U-units of depth l = 4.Source: Figure 3 · p. 3
  • Each sleep epoch is represented as x ∈ R^(n×C), with n sampling points and C channels; this work uses an EEG channel and an EOG channel.Source: §3 · p. 3
  • The task maps a sequence of sleep epochs to predicted labels corresponding to W, N1, N2, N3, and REM.Source: §3 · p. 3

4 Multimodal Salient Wave Detection Network for Sleep Staging

SalientSleepNet combines separate EEG and EOG feature streams with multiscale transition extraction and adaptive modality attention, then predicts sleep-epoch labels from signal features.

4 Multimodal Salient Wave Detection Network for Sleep Staging

  • The architecture combines two modality-specific U2 streams, multiscale transition extraction, adaptive attention, segment-wise prediction, and a bottleneck to reduce computational cost.Source: §4 · p. 3

4.1 Two-Stream U2-Structure

  • Independent encoder-decoder U2 structures process EEG and EOG separately to learn distinctive features directly from raw signals.Each U2-structure has five encoder U-units and four decoder U-units.Source: §4.1 · pp. 3–4

4.2 Multi-scale Extraction Module

  • Four dilated convolutions at rates 1–4 extract different receptive-field scales, whose feature maps are concatenated before a bottleneck reduces channels and computational cost.Source: §4.2 · p. 4, Figure 4 · p. 4

4.3 Multimodal Attention Module

  • The MMA fuses EEG and EOG feature maps, then uses channel-wise attention to strengthen features important for a particular sleep stage.Source: §4.3 · p. 4, Figure 5 · p. 4

4.4 Segment-wise Classification

  • Because pixel-wise salient-object models do not directly fit physiological segment-wise classification, the classifier pools features to epoch count and predicts a label sequence.Source: §4.4 · p. 5

5 Experiments

Experiments on two public sleep datasets compare SalientSleepNet with seven baselines and evaluate its detected waves and component ablations. The reported results favor the full model overall and show how its modules and bottleneck relate to performance and model size.

5.1 Datasets and Preprocessing

  • The study evaluates SalientSleepNet on Sleep-EDF-39 and Sleep-EDF-153, using Fpz-Cz EEG and ROC-LOC EOG at 100 Hz after merging N3 and N4 and retaining in-bed epochs.Sleep-EDF-39 has 42,308 epochs from 20 subjects, and Sleep-EDF-153 has 195,479 epochs from 78 subjects.Source: §5.1 · p. 5, §5 · p. 5

5.2 Baseline Methods

  • The comparison includes seven baselines, ranging from SVM and Random Forest to deep models that combine convolutional feature extraction with recurrent sequence modeling.The listed deep baselines include DeepSleepNet, SeqSleepNet, SleepEEGNet, SleepUtime, and TinySleepNet.Source: §5.2 · p. 5

5.3 Experiment Settings

  • SalientSleepNet is trained with Adam for 60 epochs using batches of 8 and input sequences of 20 sleep epochs, and evaluated with 20-fold cross-validation.The learning rate is 10^-3.Source: §5.3 · p. 5

5.4 Experiment Results

  • SalientSleepNet achieves the best overall performance among the compared baselines and improves N1 F1-score, while using the fewest parameters among the deep learning models.Table 1 compares methods on Sleep-EDF-39 and Sleep-EDF-153; the text gives no metric values here.Source: §5.4 · p. 5, Table 1 · p. 6
  • Figure 6 illustrates detected EEG waves in N2 and N3 and EOG waves in N1 and REM, including spindle and K-complex waves in N2 and delta waves in N3.The text also identifies slow eye movement in N1 and rapid eye movement in REM; the authors describe the detections as interpretable to a certain extent.Source: §5.4 · pp. 5–6, Figure 6 · p. 6

5.5 Ablation Experiment

  • Adding nested U-units, multi-scale extraction, and multimodal attention successively improves the basic two-stream model by capturing diverse salient waves, multiscale features, and stage-relevant modality information.The ablation variants add these components in that order.Source: §5.5 · p. 6
  • The bottleneck layer reduces SalientSleepNet's parameter count by 16%, from 1,128,865 to 948,795, while the model retains the reported smallest size and best performance among baselines.The comparison removes the bottleneck layer from the multi-scale extraction module.Source: §5.5 · p. 6

6 Conclusion

The conclusion presents SalientSleepNet as a multimodal sleep-staging network that detects and fuses salient waves while extracting multiscale stage transitions. The authors report state-of-the-art performance and the fewest parameters among existing deep learning models.

6 Conclusion

  • SalientSleepNet detects and fuses salient waves in multimodal data and extracts multi-scale transition rules among sleep stages.The conclusion frames it as a multimodal salient wave detection network for sleep staging.Source: §6 · p. 6
  • The authors report state-of-the-art sleep-staging performance and the fewest parameters among existing deep learning models.Source: §6 · p. 6
  • The authors describe SalientSleepNet as a general framework for multimodal physiological time series.Source: §6 · p. 6

Appendix

The appendix supplies sleep-stage descriptions, dataset counts, and supplementary implementation and model information. It also outlines the segment-wise classifier that maps intermediate representations to label sequences.

  • The appendix describes N1, N2, N3, REM, and Wake using characteristic EEG or eye-movement features and their typical share of a night’s sleep.For example, N2 is associated with K-complexes and sleep spindles, while N3 has high-amplitude, slow EEG signals.Source: Appendix · p. 8
  • Sleep-EDF-39 contains 42,308 epochs and Sleep-EDF-153 contains 195,479 epochs.The appendix provides a table describing sample numbers and proportions for the two datasets.Source: Appendix · p. 8
  • The segment-wise classifier pools the intermediate representation to the number of sleep epochs, then applies a Softmax convolution to predict the label sequence.The appendix identifies this classifier as a trainable link between the encoder-decoder representation and the label space.Source: Appendix · p. 9
  • Supplementary material lists model and training hyperparameters, including Adam optimization, 60 training epochs, batch size 8, and a window size of 20.It also includes a confusion-matrix figure for Sleep-EDF-39 and Sleep-EDF-153.Source: Appendix · p. 10
Loading 2105.13864v1…