Automatic sleep stage classification using time–frequency images of CWT and transfer learning using convolution neural network

Publication date: Available online 31 January 2020Source: Biocybernetics and Biomedical EngineeringAuthor(s): Pankaj Jadhav, Gaurav Rajguru, Debabrata Datta, Siddhartha MukhopadhyayAbstractFor automatic sleep stage classification, the existing methods mostly rely on hand-crafted features selected from polysomnographic records. In this paper, the goal is to develop a deep learning-based method by using single channel electroencephalogram (EEG) that automatically exploits the time–frequency spectrum of EEG signal, removing the need for manual feature extraction. The time–frequency RGB color images for EEG signal are extracted using continuous wavelet transform (CWT). The transfer learning of a pre-trained convolution neural network, squeezenet is employed to classify these CWT images into its sleep stages. The proposed method is evaluated using a publicly available Physionet sleep EDFx dataset using single-channel EEG Fpz-Cz channel. Evaluation results show that this method can achieve near state of the art accuracy even using single channel EEG signal.

Source: Biocybernetics and Biomedical Engineering - February 1, 2020 Category: Biomedical Engineering Source Type: research