WaveSleepNet: An Interpretable Network for Expert-like Sleep Staging
- URL: http://arxiv.org/abs/2404.15342v1
- Date: Thu, 11 Apr 2024 03:47:58 GMT
- Title: WaveSleepNet: An Interpretable Network for Expert-like Sleep Staging
- Authors: Yan Pei, Wei Luo,
- Abstract summary: WaveSleepNet is an interpretable neural network for sleep staging.
WaveSleepNet uses latent space representations to identify characteristic wave prototypes corresponding to different sleep stages.
The efficacy of WaveSleepNet is validated across three public datasets.
- Score: 4.4697567606459545
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Although deep learning algorithms have proven their efficiency in automatic sleep staging, the widespread skepticism about their "black-box" nature has limited its clinical acceptance. In this study, we propose WaveSleepNet, an interpretable neural network for sleep staging that reasons in a similar way to sleep experts. In this network, we utilize the latent space representations generated during training to identify characteristic wave prototypes corresponding to different sleep stages. The feature representation of an input signal is segmented into patches within the latent space, each of which is compared against the learned wave prototypes. The proximity between these patches and the wave prototypes is quantified through scores, indicating the prototypes' presence and relative proportion within the signal. The scores are served as the decision-making criteria for final sleep staging. During training, an ensemble of loss functions is employed for the prototypes' diversity and robustness. Furthermore, the learned wave prototypes are visualized by analysing occlusion sensitivity. The efficacy of WaveSleepNet is validated across three public datasets, achieving sleep staging performance that are on par with the state-of-the-art models when several WaveSleepNets are combine into a larger network. A detailed case study examined the decision-making process of the WaveSleepNet which aligns closely with American Academy of Sleep Medicine (AASM) manual guidelines. Another case study systematically explained the misidentified reason behind each sleep stage. WaveSleepNet's transparent process provides specialists with direct access to the physiological significance of its criteria, allowing for future adaptation or enrichment by sleep experts.
Related papers
- ST-USleepNet: A Spatial-Temporal Coupling Prominence Network for Multi-Channel Sleep Staging [9.83413257745779]
Sleep staging is critical for assessing sleep quality and diagnosing disorders.
Recent advancements in artificial intelligence have driven the development of automated sleep staging models.
We propose a novel framework named ST-USleepNet, comprising a spatial-temporal graph construction module and a U-shaped sleep network.
arXiv Detail & Related papers (2024-08-21T14:57:44Z) - aSAGA: Automatic Sleep Analysis with Gray Areas [2.47298967960367]
State-of-the-art automatic sleep staging methods have already demonstrated comparable reliability and superior time efficiency to manual sleep staging.
We propose a human-in-the-loop concept for sleep analysis, presenting an automatic sleep staging model (aSAGA) that performs effectively with both clinical polysomnographic recordings and home sleep studies.
arXiv Detail & Related papers (2023-10-03T13:17:38Z) - Sleep Activity Recognition and Characterization from Multi-Source
Passively Sensed Data [67.60224656603823]
Sleep Activity Recognition methods can provide indicators to assess, monitor, and characterize subjects' sleep-wake cycles and detect behavioral changes.
We propose a general method that continuously operates on passively sensed data from smartphones to characterize sleep and identify significant sleep episodes.
Thanks to their ubiquity, these devices constitute an excellent alternative data source to profile subjects' biorhythms in a continuous, objective, and non-invasive manner.
arXiv Detail & Related papers (2023-01-17T15:18:45Z) - ProductGraphSleepNet: Sleep Staging using Product Spatio-Temporal Graph
Learning with Attentive Temporal Aggregation [4.014524824655106]
This work proposes an adaptive product graph learning-based graph convolutional network, named ProductGraphSleepNet, for learning joint-temporal graphs.
The proposed network makes it possible for clinicians to comprehend and interpret the learned connectivity graphs for sleep stages.
arXiv Detail & Related papers (2022-12-09T14:34:58Z) - TransSleep: Transitioning-aware Attention-based Deep Neural Network for
Sleep Staging [2.105172041656126]
We propose a novel deep neural network structure, TransSleep, that captures distinctive local temporal patterns.
Results show that TransSleep achieves promising performance in automatic sleep staging.
arXiv Detail & Related papers (2022-03-22T08:55:32Z) - Preliminary study on using vector quantization latent spaces for TTS/VC
systems with consistent performance [55.10864476206503]
We investigate the use of quantized vectors to model the latent linguistic embedding.
By enforcing different policies over the latent spaces in the training, we are able to obtain a latent linguistic embedding.
Our experiments show that the voice cloning system built with vector quantization has only a small degradation in terms of perceptive evaluations.
arXiv Detail & Related papers (2021-06-25T07:51:35Z) - SalientSleepNet: Multimodal Salient Wave Detection Network for Sleep
Staging [10.269152939137854]
We propose SalientSleepNet, a salient wave detection network for sleep staging.
It is composed of two independent $rm U2$-like streams to extract salient features from multimodal data.
Experiments on the two datasets demonstrate that SalientSleepNet outperforms the state-of-the-art baselines.
arXiv Detail & Related papers (2021-05-24T16:32:09Z) - Convolutional Neural Networks for Sleep Stage Scoring on a Two-Channel
EEG Signal [63.18666008322476]
Sleep problems are one of the major diseases all over the world.
Basic tool used by specialists is the Polysomnogram, which is a collection of different signals recorded during sleep.
Specialists have to score the different signals according to one of the standard guidelines.
arXiv Detail & Related papers (2021-03-30T09:59:56Z) - Fast accuracy estimation of deep learning based multi-class musical
source separation [79.10962538141445]
We propose a method to evaluate the separability of instruments in any dataset without training and tuning a neural network.
Based on the oracle principle with an ideal ratio mask, our approach is an excellent proxy to estimate the separation performances of state-of-the-art deep learning approaches.
arXiv Detail & Related papers (2020-10-19T13:05:08Z) - Training Binary Neural Networks through Learning with Noisy Supervision [76.26677550127656]
This paper formalizes the binarization operations over neural networks from a learning perspective.
Experimental results on benchmark datasets indicate that the proposed binarization technique attains consistent improvements over baselines.
arXiv Detail & Related papers (2020-10-10T01:59:39Z) - Automatic detection of microsleep episodes with deep learning [55.41644538483948]
Brief fragments of sleep shorter than 15 s are defined as microsleep episodes (MSEs)
maintenance of wakefulness test (MWT) is often used in a clinical setting to assess vigilance.
MSEs are mostly not considered in the absence of established scoring criteria defining MSEs.
We aimed for automatic detection of MSEs with machine learning based on raw EEG and EOG data as input.
arXiv Detail & Related papers (2020-09-07T11:38:40Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.