U-Time: A Fully Convolutional Network for Time Series Segmentation Applied to Sleep Staging

Abstract

Neural networks are becoming more and more popular for the analysis of physiological time-series. The most successful deep learning systems in this domain combine convolutional and recurrent layers to extract useful features to model temporal relations. Unfortunately, these recurrent models are difficult to tune and optimize. In our experience, they often require task-specific modifications, which makes them challenging to use for non-experts. We propose U-Time, a fully feed-forward deep learning approach to physiological time series segmentation developed for the analysis of sleep data. U-Time is a temporal fully convolutional network based on the U-Net architecture that was originally proposed for image segmentation. U-Time maps sequential inputs of arbitrary length to sequences of class labels on a freely chosen temporal scale. This is done by implicitly classifying every individual time-point of the input signal and aggregating these classifications over fixed intervals to form the final predictions. We evaluated U-Time for sleep stage classification on a large collection of sleep electroencephalography (EEG) datasets. In all cases, we found that U-Time reaches or outperforms current state-of-the-art deep learning models while being much more robust in the training process and without requiring architecture or hyperparameter adaptation across tasks.

To appear in Advances in Neural Information Processing Systems (NeurIPS), 2019

Cited by in corpus (10)

• SleepTransformer: Automatic Sleep Staging with Interpretability and Uncertainty Quantification
• XSleepNet: Multi-View Sequential Model for Automatic Sleep Staging
• Towards More Accurate Automatic Sleep Staging via Deep Transfer Learning
• SalientSleepNet: Multimodal Salient Wave Detection Network for Sleep Staging
• Automatic Detection of Interplanetary Coronal Mass Ejections in Solar Wind In Situ Data
• aSAGA: Automatic Sleep Analysis with Gray Areas
• TRIER: Template-Guided Neural Networks for Robust and Interpretable Sleep Stage Identification from EEG Recordings
• Sleep Staging Based on Multi Scale Dual Attention Network
• Self-supervised Contrastive Learning for EEG-based Sleep Staging
• MRNet: a Multi-scale Residual Network for EEG-based Sleep Staging