A convolutional neural network for sleep stage scoring from raw single-channel EEG

Arnaud Sors (1) , Stéphane Bonnet (1) , Sebastien Mirek (2) , Laurent Vercueil (3) , Jean-François Payen (3)

1 CEA-LETI - Commissariat à l'énergie atomique et aux énergies alternatives - Laboratoire d'Electronique et de Technologie de l'Information
2 CHU Dijon - Centre Hospitalier Universitaire de Dijon - Hôpital François Mitterrand
3 CHU - Centre Hospitalier Universitaire [Grenoble]

We present a novel method for automatic sleep scoring based on single-channel EEG. We introduce the use of a deep convolutional neural network (CNN) on raw EEG samples for supervised learning of 5 class sleep stage prediction. The network has 14 layers, takes as input the 30-s epoch to be classified as well as two preceding epochs and one following epoch for temporal context, and requires no signal preprocessing or feature extraction phase. We train and evaluate our system using data from the Sleep Heart Health Study (SHHS), a large multi-center cohort study including expert-rated polysomnographic records. Performance metrics reach the state of the art, with accuracy of 0.87 and Cohen kappa of 0.81. The use of a large cohort with multiple expert raters guarantees good generalization. Finally, we present a method for visualizing class-wise patterns learned by the network.

Keywords: EEG; Single-channel; Sleep staging; Convolutional neural network; Classification; Sleep Heart Health Study

Domains: Engineering Sciences [physics]

Arnaud Sors, Stéphane Bonnet, Sebastien Mirek, Laurent Vercueil, Jean-François Payen. A convolutional neural network for sleep stage scoring from raw single-channel EEG. Biomedical Signal Processing and Control, 2018, 42, pp.107-114. ⟨10.1016/j.bspc.2017.12.001⟩. ⟨cea-02183891⟩